1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
|
# Copyright 2004,2005 Pierre Martineau <pmartino@users.sourceforge.net>
# This file is part of Bibus, a bibliographic database that can
# work together with OpenOffice.org to generate bibliographic indexes.
#
# Bibus is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# Bibus is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Bibus; if not, write to the Free Software
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
#
import wx
import BIB,cPickle
class keyDropTarget(wx.PyDropTarget):
def __init__(self,window):
wx.PyDropTarget.__init__(self,bibDataObject())
self.window = window
self.keytree = window.keytree
self.currentitem = None
self.db = window.db
# data object to store DnD object
#self.mydata = bibDataObject()
#self.SetDataObject(self.mydata)
def OnDragOver(self,x,y,dragReturn):
pos = wx.Point(x,y)
item,flags = self.keytree.HitTest(pos)
if (flags & wx.TREE_HITTEST_ONITEMLABEL) and (self.window.allowed[self.keytree.GetPyData(item)[2]] & (BIB.REF_DROP_OK|BIB.DROP_OK) ):
#self.keytree.SelectItem(item)
if self.currentitem != None:
self.keytree.SetItemBold(self.currentitem,False)
#self.keytree.SetItemBackgroundColour(self.currentitem,wx.NullColour)
self.keytree.SetItemBold(item,True)
#self.keytree.SetItemBackgroundColour(item,wx.LIGHT_GREY)
self.currentitem = item
elif flags & wx.TREE_HITTEST_ONITEMBUTTON and not self.keytree.IsExpanded(item):
self.keytree.Expand(item)
dragReturn = wx.DragNone
else:
dragReturn = wx.DragNone
return dragReturn
def OnDrop(self,x,y):
pos = wx.Point(x,y)
item,flags = self.keytree.HitTest(pos)
if (flags & wx.TREE_HITTEST_ONITEMLABEL) and (self.window.allowed[self.keytree.GetPyData(item)[2]] & (BIB.REF_DROP_OK|BIB.DROP_OK) ):
return True
else:
return False
def OnData(self,x,y,result):
pos = wx.Point(x,y)
item,flags = self.keytree.HitTest(pos)
newparentID,user,id = self.keytree.GetPyData(item)
if self.GetData():
mydrag = self.GetDataObject() # strange. A DataObject instead of a bibDataObject
if not mydrag.IsSupported(BIB.DnD_FORMAT):
return
refdragged = cPickle.loads(mydrag.GetDataHere(BIB.DnD_FORMAT))
if refdragged.format == BIB.DnD_REF:
if refdragged.dbInfo == self.db.getDbInfo():
#""" The dropped data comes from the same database, table => just update the links"""
for ref in refdragged:
self.db.writeLink((newparentID,ref[0]))
else:
#"""We can just copy and not move"""
result = wx.DragCopy
for ref in refdragged:
if ref[BIB.BIBLIOGRAPHIC_FIELDS['Identifier']]:
Identifier = ref[BIB.BIBLIOGRAPHIC_FIELDS['Identifier']]
else:
Identifier = ref[BIB.BIBLIOGRAPHIC_FIELDS['Author']].split(u',')[0].replace(u' ',u'_')+ ref[BIB.BIBLIOGRAPHIC_FIELDS['Year']]
ref = (None,Identifier)+ref[2:]
ref_id = self.db.writeRef(ref)
if ref_id != None:
self.db.writeLink((newparentID,ref_id))
elif refdragged.format == BIB.DnD_KEY and refdragged.dbInfo == self.db.getDbInfo():
# DnD of key in the same database (does not make sense if the db are differents)
# we move the key
result = wx.DragMove
oldkey_id = refdragged[0]
allowed=self.window.allowed[id]
#
key = newparentID
drop = True
rootID = self.db.getRoot(self.keytree.user)[0][0]
while key != rootID: # check if newparentID is a child (of a child of a child ...) of oldkey_id
if key == oldkey_id:
drop = False
break
else:
key = self.db.getKeyParent(self.keytree.user,key)
#
oldparentId = self.db.getKeyParent(self.keytree.user,oldkey_id)
if (allowed & BIB.DROP_OK) and (newparentID != oldparentId) and drop:
if not self.db.keyExist(self.keytree.user,newparentID,self.db.getKeyName(self.keytree.user,oldkey_id)): # The new key does not exist
self.db.modifyKeyParent(self.keytree.user,newparentID,oldkey_id)
newitem = self.keytree.AppendItem(item,self.db.getKeyName(self.keytree.user,oldkey_id))
self.keytree.SetPyData(newitem,(oldkey_id,newparentID,BIB.ID_CHILDREN[id]))
self.keytree.EnsureVisible(newitem)
if self.db.getKeyChildren(self.keytree.user,oldkey_id):
self.keytree.SetItemHasChildren(newitem,True)
else:
self.keytree.SetItemHasChildren(newitem,False)
else:
# the Key exists -> refuse drag&drop
wx.LogError(_("Sorry but the key you are moving already exist in this folder"))
result = wx.DragNone
else:
result = wx.DragNone
self.keytree.SetItemBold(item,False)
return result
def OnLeave(self):
if self.currentitem != None:
self.keytree.SetItemBold(self.currentitem,False)
#self.keytree.SetItemBackgroundColour(self.currentitem,wx.NullColour)
#self.keytree.Refresh()
class bibDragData(list):
def __init__(self,info,format):
self.dbInfo = info # info is a tuple/ For MySQL = (host,db,table) ; for SQLite = (file,table)
self.format = format # format == BIB.DnD_REF or BIB.DnD_KEY depending if we drag a reference or a key
class bibDataObject(wx.CustomDataObject):
"""DataObject for DnD"""
def __init__(self):
wx.CustomDataObject.__init__(self,BIB.DnD_FORMAT)
self.data = ""
def setObject(self,data):
return self.SetData(cPickle.dumps(data,True))
def getObject(self):
return cPickle.loads(self.GetDataHere())
class HTMLDataObject(wx.CustomDataObject):
"""DataObject for DnD"""
def __init__(self,text):
wx.CustomDataObject.__init__(self,wx.CustomDataFormat('text/html'))
self.SetData(text)
def SetData(self,data):
if type(data) == unicode:
wx.CustomDataObject.SetData(self,data.encode('utf-8'))
elif type(data) == str:
wx.CustomDataObject.SetData(self,data.decode(BIB.ENCODING).encode('utf-8'))
else:
wx.CustomDataObject.SetData(self,data)
class keyDropSource(wx.DropSource):
def __init__(self,win=None,**args):
wx.DropSource.__init__(self,win=win,**args)
self.keytree = win.keytree # the keytree that initiated the drag event
self.currentitem = None
def GiveFeedback(self, effect):
where = wx.GetMousePosition() # screen coordinates
if self.keytree == wx.FindWindowAtPoint(where):
pt = self.keytree.ScreenToClient(where) # window coordinates
item, flags = self.keytree.HitTest(pt)
if (flags & wx.TREE_HITTEST_ONITEMLABEL) and (self.window.allowed[self.keytree.GetPyData(item)[2]] & BIB.DROP_OK ):
if self.currentitem != None:
self.keytree.SetItemBold(self.currentitem,False)
self.keytree.SetItemBold(item,True)
self.currentitem = item
elif flags & wx.TREE_HITTEST_ONITEMBUTTON and not self.keytree.IsExpanded(item):
self.keytree.Expand(item)
return True
|
