dbm.py 136 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180318131823183318431853186318731883189319031913192319331943195319631973198319932003201320232033204320532063207320832093210321132123213321432153216321732183219322032213222322332243225322632273228322932303231323232333234323532363237323832393240324132423243324432453246324732483249325032513252325332543255325632573258325932603261326232633264326532663267326832693270327132723273327432753276327732783279328032813282328332843285328632873288328932903291329232933294329532963297329832993300330133023303330433053306330733083309331033113312331333143315331633173318331933203321332233233324332533263327332833293330333133323333333433353336333733383339334033413342334333443345334633473348334933503351335233533354335533563357335833593360336133623363336433653366336733683369337033713372337333743375337633773378337933803381338233833384338533863387338833893390339133923393339433953396
  1. """
  2. @package dbm.py
  3. @brief GRASS Attribute Table Manager
  4. This program is based on FileHunter, published in 'The wxPython Linux
  5. Tutorial' on wxPython WIKI pages.
  6. It also uses some functions at
  7. http://aspn.activestate.com/ASPN/Cookbook/Python/Recipe/426407
  8. @code
  9. python dbm.py vector@mapset
  10. @endcode
  11. List of classes:
  12. - Log
  13. - VirtualAttributeList
  14. - AttributeManager
  15. - DisplayAttributesDialog
  16. - VectorDBInfo
  17. - ModifyTableRecord
  18. (C) 2007-2008 by the GRASS Development Team
  19. This program is free software under the GNU General Public
  20. License (>=v2). Read the file COPYING that comes with GRASS
  21. for details.
  22. @author Jachym Cepicky <jachym.cepicky gmail.com>
  23. @author Martin Landa <landa.martin gmail.com>
  24. """
  25. import sys
  26. import os
  27. import locale
  28. import tempfile
  29. import copy
  30. ### i18N
  31. import gettext
  32. gettext.install('grasswxpy', os.path.join(os.getenv("GISBASE"), 'locale'), unicode=True)
  33. import globalvar
  34. globalvar.CheckForWx()
  35. import wx
  36. import wx.lib.mixins.listctrl as listmix
  37. import wx.lib.flatnotebook as FN
  38. import wx.lib.scrolledpanel as scrolled
  39. import sqlbuilder
  40. import grassenv
  41. import gcmd
  42. import utils
  43. import gdialogs
  44. import gselect
  45. from debug import Debug as Debug
  46. from preferences import globalSettings as UserSettings
  47. class Log:
  48. """
  49. The log output is redirected to the status bar of the containing frame.
  50. """
  51. def __init__(self, parent):
  52. self.parent = parent
  53. def write(self, text_string):
  54. """Update status bar"""
  55. self.parent.SetStatusText(text_string.strip())
  56. class VirtualAttributeList(wx.ListCtrl,
  57. listmix.ListCtrlAutoWidthMixin,
  58. listmix.ColumnSorterMixin):
  59. """
  60. Support virtual list class
  61. """
  62. def __init__(self, parent, log, mapDBInfo, layer):
  63. wx.ListCtrl.__init__(self, parent=parent, id=wx.ID_ANY,
  64. style=wx.LC_REPORT | wx.LC_HRULES |
  65. wx.LC_VRULES | wx.LC_VIRTUAL | wx.LC_SORT_ASCENDING)
  66. #
  67. # initialize variables
  68. #
  69. self.log = log
  70. self.mapDBInfo = mapDBInfo
  71. self.layer = layer
  72. self.columns = {} # <- LoadData()
  73. # self.selectedCats = []
  74. # self.lastTurnSelectedCats = [] # just temporary, for comparation
  75. #
  76. # add some attributes (colourful background for each item rows)
  77. #
  78. self.attr1 = wx.ListItemAttr()
  79. # self.attr1.SetBackgroundColour("light blue")
  80. self.attr1.SetBackgroundColour(wx.Colour(238,238,238))
  81. self.attr2 = wx.ListItemAttr()
  82. self.attr2.SetBackgroundColour("white")
  83. self.il = wx.ImageList(16, 16)
  84. self.sm_up = self.il.Add(wx.ArtProvider_GetBitmap(wx.ART_GO_UP, wx.ART_TOOLBAR,
  85. (16,16)))
  86. self.sm_dn = self.il.Add(wx.ArtProvider_GetBitmap(wx.ART_GO_DOWN, wx.ART_TOOLBAR,
  87. (16,16)))
  88. self.SetImageList(self.il, wx.IMAGE_LIST_SMALL)
  89. keyColumn = self.LoadData(layer)
  90. # setup mixins
  91. listmix.ListCtrlAutoWidthMixin.__init__(self)
  92. listmix.ColumnSorterMixin.__init__(self, len(self.columns))
  93. # sort item by category (id)
  94. if keyColumn > -1:
  95. self.SortListItems(col=keyColumn, ascending=True)
  96. else:
  97. self.SortListItems(col=0, ascending=True)
  98. # events
  99. self.Bind(wx.EVT_LIST_ITEM_SELECTED, self.OnItemSelected)
  100. self.Bind(wx.EVT_LIST_ITEM_DESELECTED, self.OnItemDeselected)
  101. self.Bind(wx.EVT_LIST_COL_CLICK, self.OnColumnClick) # sorting
  102. # self.Bind(wx.EVT_LIST_DELETE_ITEM, self.OnItemDelete, self.list)
  103. # self.Bind(wx.EVT_LIST_COL_RIGHT_CLICK, self.OnColRightClick, self.list)
  104. # self.Bind(wx.EVT_LIST_COL_BEGIN_DRAG, self.OnColBeginDrag, self.list)
  105. # self.Bind(wx.EVT_LIST_COL_DRAGGING, self.OnColDragging, self.list)
  106. # self.Bind(wx.EVT_LIST_COL_END_DRAG, self.OnColEndDrag, self.list)
  107. # self.Bind(wx.EVT_LIST_BEGIN_LABEL_EDIT, self.OnBeginEdit, self.list)
  108. # self.list.Bind(wx.EVT_LEFT_DCLICK, self.OnDoubleClick)
  109. # self.list.Bind(wx.EVT_RIGHT_DOWN, self.OnRightDown)
  110. def Update(self, mapDBInfo):
  111. """Update list according new mapDBInfo description"""
  112. self.mapDBInfo = mapDBInfo
  113. self.LoadData(self.layer)
  114. def LoadData(self, layer, columns=None, where=None):
  115. """Load data into list
  116. @param layer layer number
  117. @param columns list of columns for output
  118. @param where where statement
  119. @return id of key column
  120. @return -1 if key column is not displayed
  121. """
  122. self.log.write(_("Loading data..."))
  123. tableName = self.mapDBInfo.layers[layer]['table']
  124. keyColumn = self.mapDBInfo.layers[layer]['key']
  125. self.columns = self.mapDBInfo.tables[tableName]
  126. cmd = ["v.db.select",
  127. "-c", "--q",
  128. "map=%s" % self.mapDBInfo.map,
  129. "layer=%d" % layer]
  130. if not columns:
  131. columns = self.mapDBInfo.GetColumns(tableName)
  132. else:
  133. all = self.mapDBInfo.GetColumns(tableName)
  134. cmd.append("columns=%s" % ','.join(columns))
  135. for col in columns:
  136. if col not in all:
  137. wx.MessageBox(parent=self,
  138. message=_("Column <%(column)s> not found in "
  139. "in the table <%(table)s>.") % \
  140. { 'column' : col, 'table' : tableName },
  141. caption=_("Error"), style=wx.OK | wx.ICON_ERROR | wx.CENTRE)
  142. return
  143. if where:
  144. cmd.append("where=%s" % where)
  145. try:
  146. # for maps connected via v.external
  147. keyId = columns.index(keyColumn)
  148. except:
  149. keyId = -1
  150. #
  151. # read data
  152. #
  153. # FIXME: Max. number of rows, while the GUI is still usable
  154. # stdout can be very large, do not use PIPE, redirect to temp file
  155. # TODO: more effective way should be implemented...
  156. outFile = tempfile.NamedTemporaryFile(mode='w+b')
  157. selectCommand = gcmd.Command(cmd,
  158. stdout=outFile)
  159. # These two should probably be passed to init more cleanly
  160. # setting the numbers of items = number of elements in the dictionary
  161. self.itemDataMap = {}
  162. self.itemIndexMap = []
  163. self.itemCatsMap = {}
  164. self.DeleteAllItems()
  165. # self.ClearAll()
  166. for i in range(self.GetColumnCount()):
  167. self.DeleteColumn(0)
  168. i = 0
  169. info = wx.ListItem()
  170. info.m_mask = wx.LIST_MASK_TEXT | wx.LIST_MASK_IMAGE | wx.LIST_MASK_FORMAT
  171. info.m_image = -1
  172. info.m_format = 0
  173. for column in columns:
  174. info.m_text = column
  175. self.InsertColumnInfo(i, info)
  176. i += 1
  177. if i >= 256:
  178. self.log.write(_("Can display only 256 columns."))
  179. i = 0
  180. outFile.seek(0)
  181. while True:
  182. # os.linesep doesn't work here (MSYS)
  183. record = outFile.readline().replace('\n', '')
  184. if not record:
  185. break
  186. self.itemDataMap[i] = []
  187. j = 0
  188. for value in record.split('|'):
  189. # casting ...
  190. try:
  191. self.itemDataMap[i].append(self.columns[columns[j]]['ctype'] (value))
  192. except:
  193. self.itemDataMap[i].append('')
  194. if keyId > -1 and keyId == j:
  195. try:
  196. cat = self.columns[columns[j]]['ctype'] (value)
  197. except ValueError, e:
  198. cat = -1
  199. wx.MessageBox(parent=self,
  200. message=_("Error loading attribute data. "
  201. "Record number: %(rec)d. Unable to cast value '%(val)s' in "
  202. "key column (%(key)s) to integer.\n\n"
  203. "Details: %(detail)s") % \
  204. { 'rec' : i + 1, 'val' : value,
  205. 'key' : keyColumn, 'detail' : e},
  206. caption=_("Error"),
  207. style=wx.OK | wx.ICON_ERROR | wx.CENTRE)
  208. j += 1
  209. # insert to table
  210. # index = self.InsertStringItem(index=sys.maxint, label=str(self.itemDataMap[i][0]))
  211. # for j in range(len(self.itemDataMap[i][1:])):
  212. # self.SetStringItem(index=index, col=j+1, label=str(self.itemDataMap[i][j+1]))
  213. # self.SetItemData(item=index, data=i)
  214. self.itemIndexMap.append(i)
  215. if keyId > -1: # load cats only when LoadData() is called first time
  216. self.itemCatsMap[i] = cat
  217. i += 1
  218. if i >= 100000:
  219. self.log.write(_("Limit 100000 records."))
  220. break
  221. self.SetItemCount(i)
  222. i = 0
  223. for col in columns:
  224. width = self.columns[col]['length'] * 6 # FIXME
  225. if width < 60:
  226. width = 60
  227. if width > 300:
  228. width = 300
  229. self.SetColumnWidth(col=i, width=width)
  230. i += 1
  231. self.SendSizeEvent()
  232. return keyId
  233. def OnItemSelected(self, event):
  234. """Item selected. Add item to selected cats..."""
  235. # cat = int(self.GetItemText(event.m_itemIndex))
  236. # if cat not in self.selectedCats:
  237. # self.selectedCats.append(cat)
  238. # self.selectedCats.sort()
  239. event.Skip()
  240. def OnItemDeselected(self, event):
  241. """Item deselected. Remove item from selected cats..."""
  242. # cat = int(self.GetItemText(event.m_itemIndex))
  243. # if cat in self.selectedCats:
  244. # self.selectedCats.remove(cat)
  245. # self.selectedCats.sort()
  246. event.Skip()
  247. def GetSelectedItems(self):
  248. """Return list of selected items (category numbers)"""
  249. cats = []
  250. item = self.GetFirstSelected()
  251. while item != -1:
  252. cats.append(self.GetItemText(item))
  253. item = self.GetNextSelected(item)
  254. return cats
  255. def GetColumnText(self, index, col):
  256. """Return column text"""
  257. item = self.GetItem(index, col)
  258. return item.GetText()
  259. def GetListCtrl(self):
  260. """Returt list"""
  261. return self
  262. def OnGetItemText(self, item, col):
  263. """Get item text"""
  264. index = self.itemIndexMap[item]
  265. s = str(self.itemDataMap[index][col])
  266. return s
  267. def OnGetItemAttr(self, item):
  268. """Get item attributes"""
  269. index = self.itemIndexMap[item]
  270. if ( index % 2) == 0:
  271. return self.attr2
  272. else:
  273. return self.attr1
  274. def OnColumnClick(self, event):
  275. """Column heading clicked -> sorting"""
  276. self._col = event.GetColumn()
  277. # remove duplicated arrow symbol from column header
  278. # FIXME: should be done automatically
  279. info = wx.ListItem()
  280. info.m_mask = wx.LIST_MASK_TEXT | wx.LIST_MASK_IMAGE
  281. info.m_image = -1
  282. for column in range(self.GetColumnCount()):
  283. info.m_text = self.GetColumn(column).GetText()
  284. self.SetColumn(column, info)
  285. event.Skip()
  286. def SortItems(self, sorter=cmp):
  287. """Sort items"""
  288. items = list(self.itemDataMap.keys())
  289. items.sort(self.Sorter)
  290. self.itemIndexMap = items
  291. # redraw the list
  292. self.Refresh()
  293. def Sorter(self, key1, key2):
  294. colName = self.GetColumn(self._col).GetText()
  295. ascending = self._colSortFlag[self._col]
  296. try:
  297. item1 = self.columns[colName]["ctype"](self.itemDataMap[key1][self._col])
  298. item2 = self.columns[colName]["ctype"](self.itemDataMap[key2][self._col])
  299. except ValueError:
  300. item1 = self.itemDataMap[key1][self._col]
  301. item2 = self.itemDataMap[key2][self._col]
  302. if type(item1) == type('') or type(item2) == type(''):
  303. cmpVal = locale.strcoll(str(item1), str(item2))
  304. else:
  305. cmpVal = cmp(item1, item2)
  306. # If the items are equal then pick something else to make the sort value unique
  307. if cmpVal == 0:
  308. cmpVal = apply(cmp, self.GetSecondarySortValues(self._col, key1, key2))
  309. if ascending:
  310. return cmpVal
  311. else:
  312. return -cmpVal
  313. def GetSortImages(self):
  314. """Used by the ColumnSorterMixin, see wx/lib/mixins/listctrl.py"""
  315. return (self.sm_dn, self.sm_up)
  316. class AttributeManager(wx.Frame):
  317. """
  318. GRASS Attribute manager main window
  319. """
  320. def __init__(self, parent, id, title, vectmap,
  321. size = wx.DefaultSize, style = wx.DEFAULT_FRAME_STYLE,
  322. pointdata=None, log=None):
  323. self.vectmap = vectmap
  324. self.pointdata = pointdata
  325. self.parent = parent # GMFrame
  326. self.cmdLog = log # self.parent.goutput
  327. wx.Frame.__init__(self, parent, id, title, style=style)
  328. # icon
  329. self.SetIcon(wx.Icon(os.path.join(globalvar.ETCICONDIR, 'grass_sql.ico'), wx.BITMAP_TYPE_ICO))
  330. self.panel = wx.Panel(parent=self, id=wx.ID_ANY)
  331. try:
  332. self.map = self.parent.curr_page.maptree.Map
  333. self.mapdisplay = self.parent.curr_page.maptree.mapdisplay
  334. except:
  335. self.map = self.mapdisplay = None
  336. if pointdata:
  337. self.icon = pointdata[0]
  338. self.pointsize = pointdata[1]
  339. else:
  340. self.icon = None
  341. self.pointsize = None
  342. # status bar log class
  343. self.log = Log(self) # -> statusbar
  344. # query map layer (if parent (GMFrame) is given)
  345. self.qlayer = None
  346. # -> layers / tables description
  347. self.mapDBInfo = VectorDBInfo(self.vectmap)
  348. if len(self.mapDBInfo.layers.keys()) == 0:
  349. wx.MessageBox(parent=self.parent,
  350. message=_("Database connection for vector map <%s> "
  351. "is not defined in DB file. "
  352. "You can define new connection in "
  353. "'Manage layers' tab.") % self.vectmap,
  354. caption=_("Attribute Table Manager"),
  355. style=wx.OK | wx.ICON_INFORMATION | wx.CENTRE)
  356. #
  357. # list of command/SQL statements to be performed
  358. #
  359. self.listOfCommands = []
  360. self.listOfSQLStatements = []
  361. self.CreateStatusBar(number=1)
  362. # set up virtual lists (each layer)
  363. ### {layer: list, widgets...}
  364. self.layerPage = {}
  365. # auinotebook (browse, manage, settings)
  366. #self.notebook = wx.aui.AuiNotebook(parent=self, id=wx.ID_ANY,
  367. # style=wx.aui.AUI_NB_BOTTOM)
  368. # really needed (ML)
  369. # self.notebook.SetFont(wx.Font(10, wx.FONTFAMILY_MODERN, wx.NORMAL, wx.NORMAL, 0, ''))
  370. self.notebook = FN.FlatNotebook(parent=self.panel, id=wx.ID_ANY,
  371. style=FN.FNB_BOTTOM |
  372. FN.FNB_NO_NAV_BUTTONS |
  373. FN.FNB_FANCY_TABS)
  374. dbmStyle = globalvar.FNPageStyle
  375. self.browsePage = FN.FlatNotebook(self.panel, id=wx.ID_ANY,
  376. style=dbmStyle)
  377. # self.notebook.AddPage(self.browsePage, caption=_("Browse data"))
  378. self.notebook.AddPage(self.browsePage, text=_("Browse data")) # FN
  379. self.browsePage.SetTabAreaColour(globalvar.FNPageColor)
  380. self.manageTablePage = FN.FlatNotebook(self.panel, id=wx.ID_ANY,
  381. style=dbmStyle)
  382. #self.notebook.AddPage(self.manageTablePage, caption=_("Manage tables"))
  383. self.notebook.AddPage(self.manageTablePage, text=_("Manage tables")) # FN
  384. self.manageTablePage.SetTabAreaColour(globalvar.FNPageColor)
  385. self.manageLayerPage = FN.FlatNotebook(self.panel, id=wx.ID_ANY,
  386. style=dbmStyle)
  387. #self.notebook.AddPage(self.manageLayerPage, caption=_("Manage layers"))
  388. self.notebook.AddPage(self.manageLayerPage, text=_("Manage layers")) # FN
  389. self.manageLayerPage.SetTabAreaColour(globalvar.FNPageColor)
  390. self.infoCollapseLabelExp = _("Click here to show database connection information")
  391. self.infoCollapseLabelCol = _("Click here to hide database connection information")
  392. self.__createBrowsePage()
  393. self.__createManageTablePage()
  394. self.__createManageLayerPage()
  395. self.notebook.SetSelection(0) # select browse tab
  396. #
  397. # buttons
  398. #
  399. self.btnQuit = wx.Button(parent=self.panel, id=wx.ID_EXIT)
  400. # events
  401. self.btnQuit.Bind(wx.EVT_BUTTON, self.OnCloseWindow)
  402. self.Bind(FN.EVT_FLATNOTEBOOK_PAGE_CHANGED, self.OnLayerPageChanged, self.browsePage)
  403. self.Bind(FN.EVT_FLATNOTEBOOK_PAGE_CHANGED, self.OnLayerPageChanged, self.manageTablePage)
  404. # do layout
  405. self.__layout()
  406. # self.SetMinSize(self.GetBestSize())
  407. self.SetSize((680, 550)) # FIXME hard-coded size
  408. self.SetMinSize(self.GetSize())
  409. def __createBrowsePage(self, onlyLayer=-1):
  410. """Create browse tab page"""
  411. for layer in self.mapDBInfo.layers.keys():
  412. if onlyLayer > 0 and layer != onlyLayer:
  413. continue
  414. panel = wx.Panel(parent=self.browsePage, id=wx.ID_ANY)
  415. self.layerPage[layer] = {'browsePage': panel.GetId()}
  416. self.browsePage.AddPage(page=panel, text=" %d / %s %s" % \
  417. (layer, _("Table"), self.mapDBInfo.layers[layer]['table']))
  418. pageSizer = wx.BoxSizer(wx.VERTICAL)
  419. # attribute data
  420. listBox = wx.StaticBox(parent=panel, id=wx.ID_ANY,
  421. label=" %s " % _("Attribute data - right-click to edit/manage records"))
  422. listSizer = wx.StaticBoxSizer(listBox, wx.VERTICAL)
  423. sqlBox = wx.StaticBox(parent=panel, id=wx.ID_ANY,
  424. label=" %s " % _("SQL Query"))
  425. sqlSizer = wx.StaticBoxSizer(sqlBox, wx.VERTICAL)
  426. win = VirtualAttributeList(panel, self.log,
  427. self.mapDBInfo, layer)
  428. win.Bind(wx.EVT_LIST_ITEM_ACTIVATED, self.OnDataItemActivated)
  429. win.Bind(wx.EVT_COMMAND_RIGHT_CLICK, self.OnDataRightUp) #wxMSW
  430. win.Bind(wx.EVT_RIGHT_UP, self.OnDataRightUp) #wxGTK
  431. if UserSettings.Get(group='atm', key='leftDbClick', subkey='selection') == 0:
  432. win.Bind(wx.EVT_LEFT_DCLICK, self.OnDataItemEdit)
  433. win.Bind(wx.EVT_COMMAND_LEFT_DCLICK, self.OnDataItemEdit)
  434. else:
  435. win.Bind(wx.EVT_LEFT_DCLICK, self.OnDataDrawSelected)
  436. win.Bind(wx.EVT_COMMAND_LEFT_DCLICK, self.OnDataDrawSelected)
  437. listSizer.Add(item=win, proportion=1,
  438. flag=wx.EXPAND | wx.ALL,
  439. border=3)
  440. # sql statement box
  441. btnApply = wx.Button(parent=panel, id=wx.ID_APPLY)
  442. btnApply.SetToolTipString(_("Apply SELECT statement and reload data records"))
  443. btnApply.Bind(wx.EVT_BUTTON, self.OnApplySqlStatement)
  444. btnSqlBuilder = wx.Button(parent=panel, id=wx.ID_ANY, label=_("SQL Builder"))
  445. btnSqlBuilder.Bind(wx.EVT_BUTTON, self.OnBuilder)
  446. sqlSimple = wx.RadioButton(parent=panel, id=wx.ID_ANY,
  447. label=_("Simple"))
  448. sqlSimple.SetValue(True)
  449. sqlAdvanced = wx.RadioButton(parent=panel, id=wx.ID_ANY,
  450. label=_("Advanced"))
  451. sqlSimple.Bind(wx.EVT_RADIOBUTTON, self.OnChangeSql)
  452. sqlAdvanced.Bind(wx.EVT_RADIOBUTTON, self.OnChangeSql)
  453. sqlWhere = wx.TextCtrl(parent=panel, id=wx.ID_ANY, value="",
  454. style=wx.TE_PROCESS_ENTER)
  455. sqlStatement = wx.TextCtrl(parent=panel, id=wx.ID_ANY,
  456. value="SELECT * FROM %s" % \
  457. self.mapDBInfo.layers[layer]['table'],
  458. style=wx.TE_PROCESS_ENTER)
  459. sqlWhere.Bind(wx.EVT_TEXT_ENTER, self.OnApplySqlStatement)
  460. sqlStatement.Bind(wx.EVT_TEXT_ENTER, self.OnApplySqlStatement)
  461. sqlLabel = wx.StaticText(parent=panel, id=wx.ID_ANY,
  462. label="SELECT * FROM %s WHERE " % \
  463. self.mapDBInfo.layers[layer]['table'])
  464. label_query = wx.StaticText(parent=panel, id=wx.ID_ANY,
  465. label="")
  466. sqlFlexSizer = wx.FlexGridSizer (cols=3, hgap=5, vgap=5)
  467. sqlFlexSizer.AddGrowableCol(1)
  468. sqlFlexSizer.Add(item=sqlSimple,
  469. flag=wx.ALIGN_CENTER_VERTICAL)
  470. sqlSimpleSizer = wx.BoxSizer(wx.HORIZONTAL)
  471. sqlSimpleSizer.Add(item=sqlLabel,
  472. flag=wx.ALIGN_CENTER_VERTICAL)
  473. sqlSimpleSizer.Add(item=sqlWhere, proportion=1,
  474. flag=wx.EXPAND | wx.ALIGN_CENTER_VERTICAL)
  475. sqlFlexSizer.Add(item=sqlSimpleSizer,
  476. flag=wx.ALIGN_CENTER_VERTICAL | wx.EXPAND)
  477. sqlFlexSizer.Add(item=btnApply,
  478. flag=wx.ALIGN_RIGHT)
  479. sqlFlexSizer.Add(item=sqlAdvanced,
  480. flag=wx.ALIGN_CENTER_VERTICAL)
  481. sqlFlexSizer.Add(item=sqlStatement,
  482. flag=wx.EXPAND)
  483. sqlFlexSizer.Add(item=btnSqlBuilder,
  484. flag=wx.ALIGN_RIGHT)
  485. sqlSizer.Add(item=sqlFlexSizer,
  486. flag=wx.ALL | wx.EXPAND,
  487. border=3)
  488. pageSizer.Add(item=listSizer,
  489. proportion=1,
  490. flag=wx.ALL | wx.EXPAND,
  491. border=5)
  492. pageSizer.Add(item=sqlSizer,
  493. proportion=0,
  494. flag=wx.BOTTOM | wx.LEFT | wx.RIGHT | wx.EXPAND,
  495. border=5)
  496. panel.SetSizer(pageSizer)
  497. self.layerPage[layer]['data'] = win.GetId()
  498. self.layerPage[layer]['simple'] = sqlSimple.GetId()
  499. self.layerPage[layer]['advanced'] = sqlAdvanced.GetId()
  500. self.layerPage[layer]['where'] = sqlWhere.GetId()
  501. self.layerPage[layer]['builder'] = btnSqlBuilder.GetId()
  502. self.layerPage[layer]['statement'] = sqlStatement.GetId()
  503. self.browsePage.SetSelection(0) # select first layer
  504. try:
  505. self.layer = self.mapDBInfo.layers.keys()[0]
  506. self.OnChangeSql(None)
  507. self.log.write(_("Number of loaded records: %d") % \
  508. self.FindWindowById(self.layerPage[self.layer]['data']).GetItemCount())
  509. except IndexError:
  510. self.layer = None
  511. def __createManageTablePage(self, onlyLayer=-1):
  512. """Create manage page (create/link and alter tables)"""
  513. for layer in self.mapDBInfo.layers.keys():
  514. if onlyLayer > 0 and layer != onlyLayer:
  515. continue
  516. panel = wx.Panel(parent=self.manageTablePage, id=wx.ID_ANY)
  517. self.layerPage[layer]['tablePage'] = panel.GetId()
  518. self.manageTablePage.AddPage(page=panel, text=" %d / %s %s" % (layer, _("Table"), self.mapDBInfo.layers[layer]['table']))
  519. pageSizer = wx.BoxSizer(wx.VERTICAL)
  520. #
  521. # dbInfo
  522. #
  523. infoCollapse = wx.CollapsiblePane(parent=panel,
  524. label=self.infoCollapseLabelExp,
  525. style=wx.CP_DEFAULT_STYLE |
  526. wx.CP_NO_TLW_RESIZE | wx.EXPAND)
  527. self.MakeInfoPaneContent(layer, infoCollapse.GetPane())
  528. infoCollapse.Collapse(False)
  529. self.Bind(wx.EVT_COLLAPSIBLEPANE_CHANGED, self.OnInfoPaneChanged, infoCollapse)
  530. #
  531. # table description
  532. #
  533. table = self.mapDBInfo.layers[layer]['table']
  534. tableBox = wx.StaticBox(parent=panel, id=wx.ID_ANY,
  535. label=" %s " % _("Table <%s> - right-click to delete column(s)") % table)
  536. tableSizer = wx.StaticBoxSizer(tableBox, wx.VERTICAL)
  537. list = self.__createTableDesc(panel, table)
  538. list.Bind(wx.EVT_COMMAND_RIGHT_CLICK, self.OnTableRightUp) #wxMSW
  539. list.Bind(wx.EVT_RIGHT_UP, self.OnTableRightUp) #wxGTK
  540. self.layerPage[layer]['tableData'] = list.GetId()
  541. #
  542. # add column
  543. #
  544. columnBox = wx.StaticBox(parent=panel, id=wx.ID_ANY,
  545. label=" %s " % _("Manage columns"))
  546. columnSizer = wx.StaticBoxSizer(columnBox, wx.VERTICAL)
  547. addSizer = wx.FlexGridSizer (cols=5, hgap=3, vgap=3)
  548. addSizer.AddGrowableCol(3)
  549. label = wx.StaticText(parent=panel, id=wx.ID_ANY, label=_("Column name"))
  550. column = wx.TextCtrl(parent=panel, id=wx.ID_ANY, value='',
  551. size=(150, -1), style=wx.TE_PROCESS_ENTER)
  552. column.Bind(wx.EVT_TEXT, self.OnTableAddColumnName)
  553. column.Bind(wx.EVT_TEXT_ENTER, self.OnTableItemAdd)
  554. self.layerPage[layer]['addColName'] = column.GetId()
  555. addSizer.Add(item=label,
  556. flag=wx.ALIGN_CENTER_VERTICAL)
  557. addSizer.Add(item=column,
  558. flag=wx.ALIGN_CENTER_VERTICAL)
  559. # data type
  560. label = wx.StaticText(parent=panel, id=wx.ID_ANY, label=_("Data type"))
  561. addSizer.Add(item=label,
  562. flag=wx.ALIGN_CENTER_VERTICAL)
  563. subSizer = wx.BoxSizer(wx.HORIZONTAL)
  564. type = wx.Choice (parent=panel, id=wx.ID_ANY,
  565. choices = ["integer",
  566. "double",
  567. "varchar",
  568. "date"]) # FIXME
  569. type.SetSelection(0)
  570. type.Bind(wx.EVT_CHOICE, self.OnTableChangeType)
  571. self.layerPage[layer]['addColType'] = type.GetId()
  572. subSizer.Add(item=type,
  573. flag=wx.ALIGN_CENTER_VERTICAL | wx.LEFT | wx.RIGHT,
  574. border=3)
  575. # length
  576. label = wx.StaticText(parent=panel, id=wx.ID_ANY, label=_("Data length"))
  577. length = wx.SpinCtrl(parent=panel, id=wx.ID_ANY, size=(65, -1),
  578. initial=250,
  579. min=1, max=1e6)
  580. length.Enable(False)
  581. self.layerPage[layer]['addColLength'] = length.GetId()
  582. subSizer.Add(item=label,
  583. flag=wx.ALIGN_CENTER_VERTICAL | wx.LEFT | wx.RIGHT,
  584. border=3)
  585. subSizer.Add(item=length,
  586. flag=wx.ALIGN_CENTER_VERTICAL | wx.LEFT | wx.RIGHT,
  587. border=3)
  588. addSizer.Add(item=subSizer,
  589. flag=wx.ALIGN_CENTER_VERTICAL | wx.LEFT | wx.RIGHT,
  590. border=3)
  591. btnAddCol = wx.Button(parent=panel, id=wx.ID_ANY, label=_("Add"))
  592. btnAddCol.Bind(wx.EVT_BUTTON, self.OnTableItemAdd)
  593. btnAddCol.Enable(False)
  594. self.layerPage[layer]['addColButton'] = btnAddCol.GetId()
  595. addSizer.Add(item=btnAddCol,
  596. proportion=0,
  597. flag=wx.EXPAND | wx.ALIGN_RIGHT | wx.FIXED_MINSIZE |
  598. wx.ALIGN_CENTER_VERTICAL )
  599. # rename col
  600. label = wx.StaticText(parent=panel, id=wx.ID_ANY, label=_("Rename column"))
  601. column = wx.ComboBox(parent=panel, id=wx.ID_ANY, size=(150, -1),
  602. style=wx.CB_SIMPLE | wx.CB_READONLY,
  603. choices=self.mapDBInfo.GetColumns(table))
  604. column.SetSelection(0)
  605. self.layerPage[layer]['renameCol'] = column.GetId()
  606. addSizer.Add(item=label,
  607. flag=wx.ALIGN_CENTER_VERTICAL)
  608. addSizer.Add(item=column,
  609. flag=wx.ALIGN_CENTER_VERTICAL)
  610. label = wx.StaticText(parent=panel, id=wx.ID_ANY, label=_("To"))
  611. columnTo = wx.TextCtrl(parent=panel, id=wx.ID_ANY, value='',
  612. size=(150, -1), style=wx.TE_PROCESS_ENTER)
  613. columnTo.Bind(wx.EVT_TEXT, self.OnTableRenameColumnName)
  614. columnTo.Bind(wx.EVT_TEXT_ENTER, self.OnTableItemChange)
  615. self.layerPage[layer]['renameColTo'] = columnTo.GetId()
  616. addSizer.Add(item=label,
  617. flag=wx.ALIGN_CENTER_VERTICAL | wx.ALIGN_CENTER)
  618. addSizer.Add(item=columnTo,
  619. flag=wx.ALIGN_CENTER_VERTICAL)
  620. btnRenameCol = wx.Button(parent=panel, id=wx.ID_ANY, label=_("&Rename"))
  621. btnRenameCol.Bind(wx.EVT_BUTTON, self.OnTableItemChange)
  622. btnRenameCol.Enable(False)
  623. self.layerPage[layer]['renameColButton'] = btnRenameCol.GetId()
  624. addSizer.Add(item=btnRenameCol,
  625. proportion=0,
  626. flag=wx.EXPAND | wx.ALIGN_RIGHT | wx.FIXED_MINSIZE |
  627. wx.ALIGN_CENTER_VERTICAL)
  628. columnSizer.Add(item=addSizer, proportion=1,
  629. flag=wx.ALL | wx.EXPAND, border=3)
  630. tableSizer.Add(item=list,
  631. flag=wx.ALL | wx.EXPAND,
  632. proportion=1,
  633. border=3)
  634. pageSizer.Add(item=infoCollapse,
  635. flag=wx.ALL | wx.EXPAND,
  636. proportion=0,
  637. border=3)
  638. pageSizer.Add(item=tableSizer,
  639. flag=wx.LEFT | wx.RIGHT | wx.BOTTOM | wx.EXPAND,
  640. proportion=1,
  641. border=5)
  642. pageSizer.Add(item=columnSizer,
  643. flag=wx.LEFT | wx.RIGHT | wx.BOTTOM | wx.EXPAND,
  644. proportion=0,
  645. border=5)
  646. panel.SetSizer(pageSizer)
  647. self.layerPage[layer]['dbinfo'] = infoCollapse.GetId()
  648. self.manageTablePage.SetSelection(0) # select first layer
  649. try:
  650. self.layer = self.mapDBInfo.layers.keys()[0]
  651. except IndexError:
  652. self.layer = None
  653. def __createTableDesc(self, parent, table):
  654. """Create list with table description"""
  655. list = TableListCtrl(parent=parent, id=wx.ID_ANY,
  656. table=self.mapDBInfo.tables[table],
  657. columns=self.mapDBInfo.GetColumns(table))
  658. list.Populate()
  659. # sorter
  660. # itemDataMap = list.Populate()
  661. # listmix.ColumnSorterMixin.__init__(self, 2)
  662. return list
  663. def __createManageLayerPage(self):
  664. """Create manage page"""
  665. splitterWin = wx.SplitterWindow(parent=self.manageLayerPage, id=wx.ID_ANY)
  666. self.manageLayerPage.AddPage(page=splitterWin,
  667. text=_("Layers of vector map")) # dummy page
  668. #
  669. # list of layers
  670. #
  671. panelList = wx.Panel(parent=splitterWin, id=wx.ID_ANY)
  672. panelListSizer = wx.BoxSizer(wx.VERTICAL)
  673. layerBox = wx.StaticBox(parent=panelList, id=wx.ID_ANY,
  674. label=" %s " % _("List of layers"))
  675. layerSizer = wx.StaticBoxSizer(layerBox, wx.VERTICAL)
  676. self.layerList = self.__createLayerDesc(panelList)
  677. self.layerList.Bind(wx.EVT_COMMAND_RIGHT_CLICK, self.OnLayerRightUp) #wxMSW
  678. self.layerList.Bind(wx.EVT_RIGHT_UP, self.OnLayerRightUp) #wxGTK
  679. layerSizer.Add(item=self.layerList,
  680. flag=wx.ALL | wx.EXPAND,
  681. proportion=1,
  682. border=3)
  683. panelListSizer.Add(item=layerSizer,
  684. flag=wx.ALL | wx.EXPAND,
  685. proportion=1,
  686. border=3)
  687. panelList.SetSizer(panelListSizer)
  688. #
  689. # manage part
  690. #
  691. panelManage = wx.Panel(parent=splitterWin, id=wx.ID_ANY)
  692. manageSizer = wx.BoxSizer(wx.VERTICAL)
  693. self.manageLayerBook = LayerBook(parent=panelManage, id=wx.ID_ANY,
  694. parentDialog=self)
  695. manageSizer.Add(item=self.manageLayerBook,
  696. proportion=1,
  697. flag=wx.LEFT | wx.RIGHT | wx.BOTTOM | wx.EXPAND,
  698. border=5)
  699. panelManage.SetSizer(manageSizer)
  700. splitterWin.SplitHorizontally(panelList, panelManage, 100)
  701. splitterWin.Fit()
  702. def __createLayerDesc(self, parent):
  703. """Create list of linked layers"""
  704. list = LayerListCtrl(parent=parent, id=wx.ID_ANY,
  705. layers=self.mapDBInfo.layers)
  706. list.Populate()
  707. # sorter
  708. # itemDataMap = list.Populate()
  709. # listmix.ColumnSorterMixin.__init__(self, 2)
  710. return list
  711. def __layout(self):
  712. """Do layout"""
  713. # frame body
  714. mainSizer = wx.BoxSizer(wx.VERTICAL)
  715. # buttons
  716. btnSizer = wx.BoxSizer(wx.HORIZONTAL)
  717. btnSizer.Add(item=self.btnQuit, proportion=1,
  718. flag=wx.ALL | wx.ALIGN_RIGHT | wx.SHAPED, border=5)
  719. # btnSizer.Add(item=self.btnApply,
  720. # flag=wx.ALL, border=5)
  721. mainSizer.Add(item=self.notebook, proportion=1, flag=wx.EXPAND)
  722. mainSizer.Add(item=btnSizer, proportion=0, flag=wx.EXPAND | wx.ALL, border=5)
  723. self.panel.SetAutoLayout(True)
  724. self.panel.SetSizer(mainSizer)
  725. mainSizer.Fit(self.panel)
  726. self.Layout()
  727. def OnDataRightUp(self, event):
  728. """Table description area, context menu"""
  729. if not hasattr(self, "popupDataID1"):
  730. self.popupDataID1 = wx.NewId()
  731. self.popupDataID2 = wx.NewId()
  732. self.popupDataID3 = wx.NewId()
  733. self.popupDataID4 = wx.NewId()
  734. self.popupDataID5 = wx.NewId()
  735. self.popupDataID6 = wx.NewId()
  736. self.popupDataID7 = wx.NewId()
  737. self.popupDataID8 = wx.NewId()
  738. self.popupDataID9 = wx.NewId()
  739. self.popupDataID10 = wx.NewId()
  740. self.Bind(wx.EVT_MENU, self.OnDataItemEdit, id=self.popupDataID1)
  741. self.Bind(wx.EVT_MENU, self.OnDataItemAdd, id=self.popupDataID2)
  742. self.Bind(wx.EVT_MENU, self.OnDataItemDelete, id=self.popupDataID3)
  743. self.Bind(wx.EVT_MENU, self.OnDataItemDeleteAll, id=self.popupDataID4)
  744. self.Bind(wx.EVT_MENU, self.OnDataSelectAll, id=self.popupDataID5)
  745. self.Bind(wx.EVT_MENU, self.OnDataSelectNone, id=self.popupDataID6)
  746. self.Bind(wx.EVT_MENU, self.OnDataDrawSelected, id=self.popupDataID7)
  747. self.Bind(wx.EVT_MENU, self.OnDataDrawSelectedZoom, id=self.popupDataID8)
  748. self.Bind(wx.EVT_MENU, self.OnExtractSelected, id=self.popupDataID9)
  749. self.Bind(wx.EVT_MENU, self.OnDataReload, id=self.popupDataID10)
  750. list = self.FindWindowById(self.layerPage[self.layer]['data'])
  751. # generate popup-menu
  752. menu = wx.Menu()
  753. menu.Append(self.popupDataID1, _("Edit selected record"))
  754. selected = list.GetFirstSelected()
  755. if selected == -1 or list.GetNextSelected(selected) != -1:
  756. menu.Enable(self.popupDataID1, False)
  757. menu.Append(self.popupDataID2, _("Insert new record"))
  758. menu.Append(self.popupDataID3, _("Delete selected record(s)"))
  759. menu.Append(self.popupDataID4, _("Delete all records"))
  760. menu.AppendSeparator()
  761. menu.Append(self.popupDataID5, _("Select all"))
  762. menu.Append(self.popupDataID6, _("Deselect all"))
  763. menu.AppendSeparator()
  764. menu.Append(self.popupDataID7, _("Highlight selected"))
  765. menu.Append(self.popupDataID8, _("Highlight selected and zoom"))
  766. if not self.map or len(list.GetSelectedItems()) == 0:
  767. menu.Enable(self.popupDataID7, False)
  768. menu.Enable(self.popupDataID8, False)
  769. menu.Append(self.popupDataID9, _("Extract selected"))
  770. if list.GetFirstSelected() == -1:
  771. menu.Enable(self.popupDataID3, False)
  772. menu.Enable(self.popupDataID9, False)
  773. menu.AppendSeparator()
  774. menu.Append(self.popupDataID10, _("Reload"))
  775. self.PopupMenu(menu)
  776. menu.Destroy()
  777. # update statusbar
  778. self.log.write(_("Number of loaded records: %d") % \
  779. list.GetItemCount())
  780. def OnDataItemDelete(self, event):
  781. """Delete selected item(s) from the list (layer/category pair)"""
  782. list = self.FindWindowById(self.layerPage[self.layer]['data'])
  783. item = list.GetFirstSelected()
  784. table = self.mapDBInfo.layers[self.layer]["table"]
  785. key = self.mapDBInfo.layers[self.layer]["key"]
  786. indeces = []
  787. # collect SQL statements
  788. while item != -1:
  789. index = list.itemIndexMap[item]
  790. indeces.append(index)
  791. cat = list.itemCatsMap[index]
  792. self.listOfSQLStatements.append('DELETE FROM %s WHERE %s=%d' % \
  793. (table, key, cat))
  794. item = list.GetNextSelected(item)
  795. if UserSettings.Get(group='atm', key='askOnDeleteRec', subkey='enabled'):
  796. deleteDialog = wx.MessageBox(parent=self,
  797. message=_("Selected data records (%d) will permanently deleted "
  798. "from table. Do you want to delete them?") % \
  799. (len(self.listOfSQLStatements)),
  800. caption=_("Delete records"),
  801. style=wx.YES_NO | wx.CENTRE)
  802. if deleteDialog != wx.YES:
  803. self.listOfSQLStatements = []
  804. return
  805. # restore maps
  806. i = 0
  807. indexTemp = copy.copy(list.itemIndexMap)
  808. list.itemIndexMap = []
  809. dataTemp = copy.deepcopy(list.itemDataMap)
  810. list.itemDataMap = {}
  811. catsTemp = copy.deepcopy(list.itemCatsMap)
  812. list.itemCatsMap = {}
  813. i = 0
  814. for index in indexTemp:
  815. if index in indeces:
  816. continue
  817. list.itemIndexMap.append(i)
  818. list.itemDataMap[i] = dataTemp[index]
  819. list.itemCatsMap[i] = catsTemp[index]
  820. i += 1
  821. list.SetItemCount(len(list.itemIndexMap))
  822. # deselect items
  823. item = list.GetFirstSelected()
  824. while item != -1:
  825. list.SetItemState(item, 0, wx.LIST_STATE_SELECTED | wx.LIST_STATE_FOCUSED)
  826. item = list.GetNextSelected(item)
  827. # submit SQL statements
  828. self.ApplyCommands()
  829. event.Skip()
  830. def OnDataItemDeleteAll(self, event):
  831. """Delete all items from the list"""
  832. list = self.FindWindowById(self.layerPage[self.layer]['data'])
  833. if UserSettings.Get(group='atm', key='askOnDeleteRec', subkey='enabled'):
  834. deleteDialog = wx.MessageBox(parent=self,
  835. message=_("All data records (%d) will permanently deleted "
  836. "from table. Do you want to delete them?") % \
  837. (len(list.itemIndexMap)),
  838. caption=_("Delete records"),
  839. style=wx.YES_NO | wx.CENTRE)
  840. if deleteDialog != wx.YES:
  841. return
  842. list.DeleteAllItems()
  843. list.itemDataMap = {}
  844. list.itemIndexMap = []
  845. list.SetItemCount(0)
  846. table = self.mapDBInfo.layers[self.layer]["table"]
  847. self.listOfSQLStatements.append('DELETE FROM %s' % table)
  848. self.ApplyCommands()
  849. event.Skip()
  850. def _drawSelected(self, zoom):
  851. """Highlight selected features"""
  852. if not self.map or not self.mapdisplay:
  853. return
  854. list = self.FindWindowById(self.layerPage[self.layer]['data'])
  855. cats = map(int, list.GetSelectedItems())
  856. digitToolbar = self.mapdisplay.toolbars['vdigit']
  857. if digitToolbar and \
  858. digitToolbar.GetLayer().GetName() == self.vectmap:
  859. self.mapdisplay.digit.driver.SetSelected(cats, cats=True)
  860. if zoom:
  861. n, s, w, e = self.mapdisplay.digit.driver.GetRegionSelected()
  862. self.mapdisplay.Map.GetRegion(n=n, s=s, w=w, e=e,
  863. update=True)
  864. else:
  865. # add map layer with higlighted vector features
  866. self.AddQueryMapLayer()
  867. if zoom:
  868. keyColumn = self.mapDBInfo.layers[self.layer]['key']
  869. where = ''
  870. for range in utils.ListOfCatsToRange(cats).split(','):
  871. if '-' in range:
  872. min, max = range.split('-')
  873. where += '%s >= %d and %s <= %d or ' % \
  874. (keyColumn, int(min),
  875. keyColumn, int(max))
  876. else:
  877. where += '%s = %d or ' % (keyColumn, int(range))
  878. where = where.rstrip('or ')
  879. cmd = gcmd.Command(["v.db.select",
  880. "-r", "--q",
  881. "map=%s" % self.mapDBInfo.map,
  882. "layer=%d" % self.layer,
  883. "where=%s" % where])
  884. region = {}
  885. for line in cmd.ReadStdOutput():
  886. key, value = line.split('=')
  887. region[key] = float(value)
  888. self.mapdisplay.Map.GetRegion(n=region['n'], s=region['s'],
  889. w=region['w'], e=region['e'],
  890. update=True)
  891. if zoom:
  892. self.mapdisplay.Map.AdjustRegion() # adjust resolution
  893. self.mapdisplay.Map.AlignExtentFromDisplay() # adjust extent
  894. self.mapdisplay.MapWindow.UpdateMap(render=True, renderVector=True)
  895. else:
  896. self.mapdisplay.MapWindow.UpdateMap(render=False, renderVector=True)
  897. def OnDataDrawSelected(self, event):
  898. """Reload table description"""
  899. self._drawSelected(zoom=False)
  900. event.Skip()
  901. def OnDataDrawSelectedZoom(self, event):
  902. self._drawSelected(zoom=True)
  903. event.Skip()
  904. def OnDataItemAdd(self, event):
  905. """Add new record to the attribute table"""
  906. list = self.FindWindowById(self.layerPage[self.layer]['data'])
  907. table = self.mapDBInfo.layers[self.layer]['table']
  908. keyColumn = self.mapDBInfo.layers[self.layer]['key']
  909. # (column name, value)
  910. data = []
  911. # collect names of all visible columns
  912. columnName = []
  913. for i in range(list.GetColumnCount()):
  914. columnName.append(list.GetColumn(i).GetText())
  915. # maximal category number
  916. maxCat = max(list.itemCatsMap.values())
  917. # key column must be always presented
  918. if keyColumn not in columnName:
  919. columnName.insert(0, keyColumn) # insert key column on first position
  920. data.append((keyColumn, str(maxCat + 1)))
  921. missingKey = True
  922. else:
  923. missingKey = False
  924. # add other visible columns
  925. colIdx = 0
  926. keyId = -1
  927. for col in columnName:
  928. if col == keyColumn: # key
  929. if missingKey is False:
  930. data.append((col, str(maxCat + 1)))
  931. keyId = colIdx
  932. else:
  933. data.append((col, ''))
  934. colIdx += 1
  935. dlg = ModifyTableRecord(parent=self, id=wx.ID_ANY,
  936. title=_("Insert new record"),
  937. data=data, keyEditable=(keyId, True))
  938. if dlg.ShowModal() == wx.ID_OK:
  939. try: # get category number
  940. cat = int(dlg.GetValues(columns=[keyColumn])[0])
  941. except:
  942. cat = -1
  943. try:
  944. if cat in list.itemCatsMap.values():
  945. raise ValueError(_("Record with category number %d "
  946. "already exists in the table.") % cat)
  947. values = dlg.GetValues() # values (need to be casted)
  948. columnsString = ''
  949. valuesString = ''
  950. for i in range(len(values)):
  951. if len(values[i]) == 0: # NULL
  952. if columnName[i] == keyColumn:
  953. raise ValueError(_("Category number (column %s)"
  954. " is missing.") % keyColumn)
  955. else:
  956. continue
  957. try:
  958. values[i] = list.columns[columnName[i]]['ctype'] (values[i])
  959. except:
  960. raise ValueError(_("Casting value '%(value)s' to %(type)s failed.") %
  961. {'value' : str(values[i]),
  962. 'type' : list.columns[columnName[i]]['type']})
  963. columnsString += '%s,' % columnName[i]
  964. if list.columns[columnName[i]]['ctype'] == str:
  965. valuesString += "'%s'," % values[i]
  966. else:
  967. valuesString += "%s," % values[i]
  968. except ValueError, err:
  969. wx.MessageBox(parent=self,
  970. message="%s%s%s" % (_("Unable to insert new record."),
  971. os.linesep, err),
  972. caption=_("Error"), style=wx.OK | wx.ICON_ERROR | wx.CENTRE)
  973. return
  974. # remove category if need
  975. if missingKey is True:
  976. del values[0]
  977. # add new item to the list
  978. index = max(list.itemIndexMap) + 1
  979. list.itemIndexMap.append(index)
  980. list.itemDataMap[index] = values
  981. list.itemCatsMap[index] = cat
  982. list.SetItemCount(list.GetItemCount() + 1)
  983. self.listOfSQLStatements.append('INSERT INTO %s (%s) VALUES(%s)' % \
  984. (table,
  985. columnsString.strip(','),
  986. valuesString.strip(',')))
  987. self.ApplyCommands()
  988. def OnDataItemEdit(self, event):
  989. """Edit selected record of the attribute table"""
  990. list = self.FindWindowById(self.layerPage[self.layer]['data'])
  991. item = list.GetFirstSelected()
  992. if item == -1:
  993. return
  994. table = self.mapDBInfo.layers[self.layer]['table']
  995. keyColumn = self.mapDBInfo.layers[self.layer]['key']
  996. cat = list.itemCatsMap[list.itemIndexMap[item]]
  997. # (column name, value)
  998. data = []
  999. # collect names of all visible columns
  1000. columnName = []
  1001. for i in range(list.GetColumnCount()):
  1002. columnName.append(list.GetColumn(i).GetText())
  1003. # key column must be always presented
  1004. if keyColumn not in columnName:
  1005. columnName.insert(0, keyColumn) # insert key column on first position
  1006. data.append((keyColumn, str(cat)))
  1007. keyId = 0
  1008. missingKey = True
  1009. else:
  1010. missingKey = False
  1011. # add other visible columns
  1012. for i in range(len(columnName)):
  1013. if columnName[i] == keyColumn: # key
  1014. if missingKey is False:
  1015. data.append((columnName[i], str(cat)))
  1016. keyId = i
  1017. else:
  1018. if missingKey is True:
  1019. value = list.GetItem(item, i-1).GetText()
  1020. else:
  1021. value = list.GetItem(item, i).GetText()
  1022. data.append((columnName[i], value))
  1023. dlg = ModifyTableRecord(parent=self, id=wx.ID_ANY,
  1024. title=_("Update existing record"),
  1025. data=data, keyEditable=(keyId, False))
  1026. if dlg.ShowModal() == wx.ID_OK:
  1027. values = dlg.GetValues() # string
  1028. updateString = ''
  1029. try:
  1030. for i in range(len(values)):
  1031. if i == keyId: # skip key column
  1032. continue
  1033. if str(list.GetItem(item, i).GetText()) != str(values[i]):
  1034. if len(values[i]) > 0:
  1035. try:
  1036. if missingKey is True:
  1037. idx = i - 1
  1038. else:
  1039. idx = i
  1040. list.itemDataMap[item][idx] = \
  1041. list.columns[columnName[i]]['ctype'] (values[i])
  1042. except:
  1043. raise ValueError(_("Casting value '%(value)s' to %(type)s failed.") % \
  1044. {'value' : str(values[i]),
  1045. 'type' : list.columns[columnName[i]]['type']})
  1046. if list.columns[columnName[i]]['ctype'] == str:
  1047. updateString += "%s='%s'," % (columnName[i], values[i])
  1048. else:
  1049. updateString += "%s=%s," % (columnName[i], values[i])
  1050. else: # NULL
  1051. updateString += "%s=NULL," % (columnName[i])
  1052. except ValueError, err:
  1053. wx.MessageBox(parent=self,
  1054. message="%s%s%s" % (_("Unable to update existing record."),
  1055. os.linesep, err),
  1056. caption=_("Error"), style=wx.OK | wx.ICON_ERROR | wx.CENTRE)
  1057. return
  1058. if len(updateString) > 0:
  1059. self.listOfSQLStatements.append('UPDATE %s SET %s WHERE %s=%d' % \
  1060. (table, updateString.strip(','),
  1061. keyColumn, cat))
  1062. self.ApplyCommands()
  1063. def OnDataReload(self, event):
  1064. """Reload list of records"""
  1065. self.OnApplySqlStatement(None)
  1066. self.listOfSQLStatements = []
  1067. def OnDataSelectAll(self, event):
  1068. """Select all items"""
  1069. list = self.FindWindowById(self.layerPage[self.layer]['data'])
  1070. item = -1
  1071. while True:
  1072. item = list.GetNextItem(item)
  1073. if item == -1:
  1074. break
  1075. list.SetItemState(item, wx.LIST_STATE_SELECTED, wx.LIST_STATE_SELECTED)
  1076. event.Skip()
  1077. def OnDataSelectNone(self, event):
  1078. """Deselect items"""
  1079. list = self.FindWindowById(self.layerPage[self.layer]['data'])
  1080. item = -1
  1081. while True:
  1082. item = list.GetNextItem(item, wx.LIST_STATE_SELECTED)
  1083. if item == -1:
  1084. break
  1085. list.SetItemState(item, 0, wx.LIST_STATE_SELECTED | wx.LIST_STATE_FOCUSED)
  1086. event.Skip()
  1087. def OnTableChangeType(self, event):
  1088. """Data type for new column changed. Enable or disable
  1089. data length widget"""
  1090. win = self.FindWindowById(self.layerPage[self.layer]['addColLength'])
  1091. if event.GetString() == "varchar":
  1092. win.Enable(True)
  1093. else:
  1094. win.Enable(False)
  1095. def OnTableRenameColumnName(self, event):
  1096. """Editing column name to be added to the table"""
  1097. btn = self.FindWindowById(self.layerPage[self.layer]['renameColButton'])
  1098. col = self.FindWindowById(self.layerPage[self.layer]['renameCol'])
  1099. colTo = self.FindWindowById(self.layerPage[self.layer]['renameColTo'])
  1100. if len(col.GetValue()) > 0 and len(colTo.GetValue()) > 0:
  1101. btn.Enable(True)
  1102. else:
  1103. btn.Enable(False)
  1104. event.Skip()
  1105. def OnTableAddColumnName(self, event):
  1106. """Editing column name to be added to the table"""
  1107. btn = self.FindWindowById(self.layerPage[self.layer]['addColButton'])
  1108. if len(event.GetString()) > 0:
  1109. btn.Enable(True)
  1110. else:
  1111. btn.Enable(False)
  1112. event.Skip()
  1113. def OnTableItemChange(self, event):
  1114. """Rename column in the table"""
  1115. list = self.FindWindowById(self.layerPage[self.layer]['tableData'])
  1116. name = self.FindWindowById(self.layerPage[self.layer]['renameCol']).GetValue()
  1117. nameTo = self.FindWindowById(self.layerPage[self.layer]['renameColTo']).GetValue()
  1118. table = self.mapDBInfo.layers[self.layer]["table"]
  1119. if not name or not nameTo:
  1120. wx.MessageBox(self=self,
  1121. message=_("Unable to rename column. "
  1122. "No column name defined."),
  1123. caption=_("Error"), style=wx.OK | wx.ICON_ERROR | wx.CENTRE)
  1124. return
  1125. else:
  1126. item = list.FindItem(start=-1, str=name)
  1127. if item > -1:
  1128. if list.FindItem(start=-1, str=nameTo) > -1:
  1129. wx.MessageBox(parent=self,
  1130. message=_("Unable to rename column <%(column)s> to "
  1131. "<%(columnTo)s>. Column already exists "
  1132. "in the table <%(table)s>.") % \
  1133. {'column' : name, 'columnTo' : nameTo,
  1134. 'table' : table},
  1135. caption=_("Error"), style=wx.OK | wx.ICON_ERROR | wx.CENTRE)
  1136. return
  1137. else:
  1138. list.SetItemText(item, nameTo)
  1139. self.listOfCommands.append(['v.db.renamecol',
  1140. 'map=%s' % self.vectmap,
  1141. 'layer=%d' % self.layer,
  1142. 'column=%s,%s' % (name, nameTo)])
  1143. else:
  1144. wx.MessageBox(parent=self,
  1145. message=_("Unable to rename column. "
  1146. "Column <%(column)s> doesn't exist in the table <%(table)s>.") %
  1147. {'column' : name, 'table' : table},
  1148. caption=_("Error"), style=wx.OK | wx.ICON_ERROR | wx.CENTRE)
  1149. return
  1150. # apply changes
  1151. self.ApplyCommands()
  1152. # update widgets
  1153. self.FindWindowById(self.layerPage[self.layer]['renameCol']).SetItems(self.mapDBInfo.GetColumns(table))
  1154. self.FindWindowById(self.layerPage[self.layer]['renameCol']).SetSelection(0)
  1155. self.FindWindowById(self.layerPage[self.layer]['renameColTo']).SetValue('')
  1156. event.Skip()
  1157. def OnTableRightUp(self, event):
  1158. """Table description area, context menu"""
  1159. if not hasattr(self, "popupTableID"):
  1160. self.popupTableID1 = wx.NewId()
  1161. self.popupTableID2 = wx.NewId()
  1162. self.popupTableID3 = wx.NewId()
  1163. self.Bind(wx.EVT_MENU, self.OnTableItemDelete, id=self.popupTableID1)
  1164. self.Bind(wx.EVT_MENU, self.OnTableItemDeleteAll, id=self.popupTableID2)
  1165. self.Bind(wx.EVT_MENU, self.OnTableReload, id=self.popupTableID3)
  1166. # generate popup-menu
  1167. menu = wx.Menu()
  1168. menu.Append(self.popupTableID1, _("Drop selected column"))
  1169. if self.FindWindowById(self.layerPage[self.layer]['tableData']).GetFirstSelected() == -1:
  1170. menu.Enable(self.popupTableID1, False)
  1171. menu.Append(self.popupTableID2, _("Drop all columns"))
  1172. menu.AppendSeparator()
  1173. menu.Append(self.popupTableID3, _("Reload"))
  1174. self.PopupMenu(menu)
  1175. menu.Destroy()
  1176. def OnTableItemDelete(self, event):
  1177. """Delete selected item(s) from the list"""
  1178. list = self.FindWindowById(self.layerPage[self.layer]['tableData'])
  1179. item = list.GetFirstSelected()
  1180. while item != -1:
  1181. self.listOfCommands.append(['v.db.dropcol',
  1182. 'map=%s' % self.vectmap,
  1183. 'layer=%d' % self.layer,
  1184. 'column=%s' % list.GetItemText(item)])
  1185. list.DeleteItem(item)
  1186. item = list.GetFirstSelected()
  1187. # apply changes
  1188. self.ApplyCommands()
  1189. # update widgets
  1190. table = self.mapDBInfo.layers[self.layer]['table']
  1191. self.FindWindowById(self.layerPage[self.layer]['renameCol']).SetItems(self.mapDBInfo.GetColumns(table))
  1192. self.FindWindowById(self.layerPage[self.layer]['renameCol']).SetSelection(0)
  1193. event.Skip()
  1194. def OnTableItemDeleteAll(self, event):
  1195. """Delete all items from the list"""
  1196. table = self.mapDBInfo.layers[self.layer]['table']
  1197. cols = self.mapDBInfo.GetColumns(table)
  1198. for col in cols:
  1199. self.listOfCommands = [['v.db.dropcol',
  1200. 'map=%s' % self.vectmap,
  1201. 'layer=%d' % self.layer,
  1202. 'column=%s' % col]]
  1203. self.FindWindowById(self.layerPage[self.layer]['tableData']).DeleteAllItems()
  1204. # apply changes
  1205. self.ApplyCommands()
  1206. # update widgets
  1207. table = self.mapDBInfo.layers[self.layer]['table']
  1208. self.FindWindowById(self.layerPage[self.layer]['renameCol']).SetItems(self.mapDBInfo.GetColumns(table))
  1209. self.FindWindowById(self.layerPage[self.layer]['renameCol']).SetSelection(0)
  1210. event.Skip()
  1211. def OnTableReload(self, event=None):
  1212. """Reload table description"""
  1213. self.FindWindowById(self.layerPage[self.layer]['tableData']).Populate(update=True)
  1214. self.listOfCommands = []
  1215. def OnTableItemAdd(self, event):
  1216. """Add new column to the table"""
  1217. table = self.mapDBInfo.layers[self.layer]['table']
  1218. name = self.FindWindowById(self.layerPage[self.layer]['addColName']).GetValue()
  1219. if not name:
  1220. wx.MessageBox(parent=self,
  1221. message=_("Unable to add column to the table. "
  1222. "No column name defined."),
  1223. caption=_("Error"), style=wx.OK | wx.ICON_ERROR | wx.CENTRE)
  1224. return
  1225. type = self.FindWindowById(self.layerPage[self.layer]['addColType']). \
  1226. GetStringSelection()
  1227. # cast type if needed
  1228. if type == 'double':
  1229. type = 'double precision'
  1230. if type == 'varchar':
  1231. length = int(self.FindWindowById(self.layerPage[self.layer]['addColLength']). \
  1232. GetValue())
  1233. else:
  1234. length = '' # FIXME
  1235. # add item to the list of table columns
  1236. list = self.FindWindowById(self.layerPage[self.layer]['tableData'])
  1237. # check for duplicate items
  1238. if list.FindItem(start=-1, str=name) > -1:
  1239. wx.MessageBox(parent=self,
  1240. message=_("Column <%(column)s> already exists in table <%(table)s>.") % \
  1241. {'column' : name, 'table' : self.mapDBInfo.layers[self.layer]["table"]},
  1242. caption=_("Error"), style=wx.OK | wx.ICON_ERROR | wx.CENTRE)
  1243. return
  1244. index = list.InsertStringItem(sys.maxint, str(name))
  1245. list.SetStringItem(index, 0, str(name))
  1246. list.SetStringItem(index, 1, str(type))
  1247. list.SetStringItem(index, 2, str(length))
  1248. # add v.db.addcol command to the list
  1249. if type == 'varchar':
  1250. type += ' (%d)' % length
  1251. self.listOfCommands.append(['v.db.addcol',
  1252. 'map=%s' % self.vectmap,
  1253. 'layer=%d' % self.layer,
  1254. 'columns=%s %s' % (name, type)])
  1255. # apply changes
  1256. self.ApplyCommands()
  1257. # update widgets
  1258. self.FindWindowById(self.layerPage[self.layer]['addColName']).SetValue('')
  1259. self.FindWindowById(self.layerPage[self.layer]['renameCol']).SetItems(self.mapDBInfo.GetColumns(table))
  1260. self.FindWindowById(self.layerPage[self.layer]['renameCol']).SetSelection(0)
  1261. event.Skip()
  1262. def OnLayerPageChanged(self, event):
  1263. """Layer tab changed"""
  1264. pageNum = event.GetSelection()
  1265. self.layer = self.mapDBInfo.layers.keys()[pageNum]
  1266. try:
  1267. self.OnChangeSql(None)
  1268. # update statusbar
  1269. self.log.write(_("Number of loaded records: %d") % \
  1270. self.FindWindowById(self.layerPage[self.layer]['data']).\
  1271. GetItemCount())
  1272. except:
  1273. pass
  1274. def OnLayerRightUp(self, event):
  1275. """Layer description area, context menu"""
  1276. pass
  1277. def OnChangeSql(self, event):
  1278. """Switch simple/advanced sql statement"""
  1279. if self.FindWindowById(self.layerPage[self.layer]['simple']).GetValue():
  1280. self.FindWindowById(self.layerPage[self.layer]['where']).Enable(True)
  1281. self.FindWindowById(self.layerPage[self.layer]['statement']).Enable(False)
  1282. self.FindWindowById(self.layerPage[self.layer]['builder']).Enable(False)
  1283. else:
  1284. self.FindWindowById(self.layerPage[self.layer]['where']).Enable(False)
  1285. self.FindWindowById(self.layerPage[self.layer]['statement']).Enable(True)
  1286. self.FindWindowById(self.layerPage[self.layer]['builder']).Enable(True)
  1287. def ApplyCommands(self):
  1288. """Apply changes"""
  1289. # perform GRASS commands (e.g. v.db.addcol)
  1290. if len(self.listOfCommands) > 0:
  1291. for cmd in self.listOfCommands:
  1292. Debug.msg(3, 'AttributeManager.ApplyCommands() cmd=\'%s\'' %
  1293. ' '.join(cmd))
  1294. gcmd.Command(cmd)
  1295. self.mapDBInfo = VectorDBInfo(self.vectmap)
  1296. table = self.mapDBInfo.layers[self.layer]['table']
  1297. # update table description
  1298. list = self.FindWindowById(self.layerPage[self.layer]['tableData'])
  1299. list.Update(table=self.mapDBInfo.tables[table],
  1300. columns=self.mapDBInfo.GetColumns(table))
  1301. self.OnTableReload(None)
  1302. # update data list
  1303. list = self.FindWindowById(self.layerPage[self.layer]['data'])
  1304. list.Update(self.mapDBInfo)
  1305. # reset list of commands
  1306. self.listOfCommands = []
  1307. # perform SQL non-select statements (e.g. 'delete from table where cat=1')
  1308. if len(self.listOfSQLStatements) > 0:
  1309. sqlFile = tempfile.NamedTemporaryFile(mode="w")
  1310. for sql in self.listOfSQLStatements:
  1311. sqlFile.file.write(sql + ";\n")
  1312. sqlFile.file.flush()
  1313. driver = self.mapDBInfo.layers[self.layer]["driver"]
  1314. database = self.mapDBInfo.layers[self.layer]["database"]
  1315. cmd = ['db.execute',
  1316. 'input=%s' % sqlFile.name,
  1317. 'driver=%s' % driver,
  1318. 'database=%s' % database]
  1319. Debug.msg(3, 'AttributeManger.ApplyCommands(): %s' %
  1320. ';'.join(["%s" % s for s in self.listOfSQLStatements]))
  1321. gcmd.Command(cmd)
  1322. # reset list of statements
  1323. self.listOfSQLStatements = []
  1324. def OnApplySqlStatement(self, event):
  1325. """Apply simple/advanced sql statement"""
  1326. keyColumn = -1 # index of key column
  1327. listWin = self.FindWindowById(self.layerPage[self.layer]['data'])
  1328. if self.FindWindowById(self.layerPage[self.layer]['simple']).GetValue():
  1329. # simple sql statement
  1330. where = self.FindWindowById(self.layerPage[self.layer]['where']).GetValue().strip()
  1331. try:
  1332. if len(where) > 0:
  1333. keyColumn = listWin.LoadData(self.layer, where=where)
  1334. else:
  1335. keyColumn = listWin.LoadData(self.layer)
  1336. except gcmd.CmdError, e:
  1337. wx.MessageBox(parent=self,
  1338. message=_("Loading attribute data failed.\n\n%s") % e.message,
  1339. caption=_("Error"), style=wx.OK | wx.ICON_ERROR | wx.CENTRE)
  1340. self.FindWindowById(self.layerPage[self.layer]['where']).SetValue('')
  1341. else:
  1342. # advanced sql statement
  1343. win = self.FindWindowById(self.layerPage[self.layer]['statement'])
  1344. try:
  1345. cols, where = self.ValidateSelectStatement(win.GetValue())
  1346. except TypeError:
  1347. wx.MessageBox(parent=self,
  1348. message=_("Loading attribute data failed.\n"
  1349. "Invalid SQL select statement.\n\n%s") % win.GetValue(),
  1350. caption=_("Error"), style=wx.OK | wx.ICON_ERROR | wx.CENTRE)
  1351. win.SetValue("SELECT * FROM %s" % self.mapDBInfo.layers[self.layer]['table'])
  1352. cols = None
  1353. where = None
  1354. print cols, where
  1355. if cols or where:
  1356. try:
  1357. keyColumn = listWin.LoadData(self.layer, columns=cols,
  1358. where=where)
  1359. except gcmd.CmdError, e:
  1360. wx.MessageBox(parent=self,
  1361. message=_("Loading attribute data failed.\n\n%s") % e.message,
  1362. caption=_("Error"), style=wx.OK | wx.ICON_ERROR | wx.CENTRE)
  1363. win.SetValue("SELECT * FROM %s" % self.mapDBInfo.layers[self.layer]['table'])
  1364. # sort by key column
  1365. if keyColumn > -1:
  1366. listWin.SortListItems(col=keyColumn, ascending=True)
  1367. else:
  1368. listWin.SortListItems(col=0, ascending=True)
  1369. # update statusbar
  1370. self.log.write(_("Number of loaded records: %d") % \
  1371. self.FindWindowById(self.layerPage[self.layer]['data']).GetItemCount())
  1372. def ValidateSelectStatement(self, statement):
  1373. """Validate SQL select statement
  1374. @return (columns, where)
  1375. @return None on error
  1376. """
  1377. if statement[0:7].lower() != 'select ':
  1378. return None
  1379. cols = ''
  1380. index = 7
  1381. for c in statement[index:]:
  1382. if c == ' ':
  1383. break
  1384. cols += c
  1385. index += 1
  1386. if cols == '*':
  1387. cols = None
  1388. else:
  1389. cols = cols.split(',')
  1390. tablelen = len(self.mapDBInfo.layers[self.layer]['table'])
  1391. if statement[index+1:index+6].lower() != 'from ' or \
  1392. statement[index+6:index+6+tablelen] != '%s' % \
  1393. (self.mapDBInfo.layers[self.layer]['table']):
  1394. return None
  1395. if len(statement[index+7+tablelen:]) > 0:
  1396. index = statement.lower().find('where ')
  1397. if index > -1:
  1398. where = statement[index+6:]
  1399. else:
  1400. return None
  1401. else:
  1402. where = None
  1403. return (cols, where)
  1404. def OnInfoPaneChanged(self, event):
  1405. """Collapse database connection info box"""
  1406. if self.FindWindowById(self.layerPage[self.layer]['dbinfo']).IsExpanded():
  1407. self.FindWindowById(self.layerPage[self.layer]['dbinfo']).SetLabel( \
  1408. self.infoCollapseLabelCol)
  1409. else:
  1410. self.FindWindowById(self.layerPage[self.layer]['dbinfo']).SetLabel( \
  1411. self.infoCollapseLabelExp)
  1412. # redo layout
  1413. self.Layout()
  1414. def MakeInfoPaneContent(self, layer, pane):
  1415. """Create database connection information content"""
  1416. # connection info
  1417. border = wx.BoxSizer(wx.VERTICAL)
  1418. connectionInfoBox = wx.StaticBox(parent=pane, id=wx.ID_ANY,
  1419. label=" %s " % _("Database connection"))
  1420. infoSizer = wx.StaticBoxSizer(connectionInfoBox, wx.VERTICAL)
  1421. infoFlexSizer = wx.FlexGridSizer (cols=2, hgap=1, vgap=1)
  1422. infoFlexSizer.AddGrowableCol(1)
  1423. infoFlexSizer.Add(item=wx.StaticText(parent=pane, id=wx.ID_ANY,
  1424. label="Driver:"))
  1425. infoFlexSizer.Add(item=wx.StaticText(parent=pane, id=wx.ID_ANY,
  1426. label="%s" % \
  1427. self.mapDBInfo.layers[layer]['driver']))
  1428. infoFlexSizer.Add(item=wx.StaticText(parent=pane, id=wx.ID_ANY,
  1429. label="Database:"))
  1430. infoFlexSizer.Add(item=wx.StaticText(parent=pane, id=wx.ID_ANY,
  1431. label="%s" % \
  1432. self.mapDBInfo.layers[layer]['database']))
  1433. infoFlexSizer.Add(item=wx.StaticText(parent=pane, id=wx.ID_ANY,
  1434. label="Table:"))
  1435. infoFlexSizer.Add(item=wx.StaticText(parent=pane, id=wx.ID_ANY,
  1436. label="%s" % \
  1437. self.mapDBInfo.layers[layer]['table']))
  1438. infoFlexSizer.Add(item=wx.StaticText(parent=pane, id=wx.ID_ANY,
  1439. label="Key:"))
  1440. infoFlexSizer.Add(item=wx.StaticText(parent=pane, id=wx.ID_ANY,
  1441. label="%s" % \
  1442. self.mapDBInfo.layers[layer]['key']))
  1443. infoSizer.Add(item=infoFlexSizer,
  1444. proportion=1,
  1445. flag=wx.EXPAND | wx.ALL,
  1446. border=3)
  1447. border.Add(item=infoSizer,
  1448. proportion=1,
  1449. flag=wx.EXPAND | wx.ALL,
  1450. border=3)
  1451. pane.SetSizer(border)
  1452. def OnCloseWindow(self, event):
  1453. """Cancel button pressed"""
  1454. self.Close()
  1455. event.Skip()
  1456. def OnBuilder(self,event):
  1457. """SQL Builder button pressed"""
  1458. self.builder = sqlbuilder.SQLFrame(parent=self, id=wx.ID_ANY,
  1459. title=_("SQL Builder"),
  1460. vectmap=self.vectmap)
  1461. def OnTextEnter(self, event):
  1462. pass
  1463. def OnSQLBuilder(self, event):
  1464. pass
  1465. def OnDataItemActivated(self, event):
  1466. """Item activated, highlight selected item"""
  1467. self.OnDataDrawSelected(event)
  1468. event.Skip()
  1469. def OnExtractSelected(self, event):
  1470. """
  1471. Extract vector objects selected in attribute browse window
  1472. to new vector map
  1473. """
  1474. list = self.FindWindowById(self.layerPage[self.layer]['data'])
  1475. # cats = list.selectedCats[:]
  1476. cats = list.GetSelectedItems()
  1477. if len(cats) == 0:
  1478. wx.MessageBox(parent=self,
  1479. message=_('Nothing to extract.'),
  1480. caption=_('Extract selected'), style=wx.CENTRE)
  1481. return
  1482. else:
  1483. # dialog to get file name
  1484. gdialogs.CreateNewVector(parent=self, title=_('Extract selected'),
  1485. log=self.cmdLog,
  1486. cmdDef=(["v.extract",
  1487. "input=%s" % self.vectmap,
  1488. "list=%s" % utils.ListOfCatsToRange(cats)],
  1489. "output"))
  1490. def AddQueryMapLayer(self):
  1491. """Redraw a map
  1492. Return True if map has been redrawn, False if no map is given
  1493. """
  1494. list = self.FindWindowById(self.layerPage[self.layer]['data'])
  1495. cats = list.GetSelectedItems() # FIXME: category can be hiden in list
  1496. if self.mapdisplay.Map.GetLayerIndex(self.qlayer) < 0:
  1497. self.qlayer = None
  1498. if self.qlayer:
  1499. self.qlayer.SetCmd(self.mapdisplay.AddTmpVectorMapLayer(self.vectmap, cats, addLayer=False))
  1500. else:
  1501. self.qlayer = self.mapdisplay.AddTmpVectorMapLayer(self.vectmap, cats)
  1502. def UpdateDialog(self, layer):
  1503. """Updates dialog layout for given layer"""
  1504. #
  1505. # delete page
  1506. #
  1507. if layer in self.mapDBInfo.layers.keys():
  1508. # delete page
  1509. # draging pages disallowed
  1510. # if self.browsePage.GetPageText(page).replace('Layer ', '').strip() == str(layer):
  1511. # self.browsePage.DeletePage(page)
  1512. # break
  1513. self.browsePage.DeletePage(self.mapDBInfo.layers.keys().index(layer))
  1514. self.manageTablePage.DeletePage(self.mapDBInfo.layers.keys().index(layer))
  1515. # set current page selection
  1516. self.notebook.SetSelection(2)
  1517. # fetch fresh db info
  1518. self.mapDBInfo = VectorDBInfo(self.vectmap)
  1519. #
  1520. # add new page
  1521. #
  1522. if layer in self.mapDBInfo.layers.keys():
  1523. # 'browse data' page
  1524. self.__createBrowsePage(layer)
  1525. # 'manage tables' page
  1526. self.__createManageTablePage(layer)
  1527. # set current page selection
  1528. self.notebook.SetSelection(2)
  1529. #
  1530. # 'manage layers' page
  1531. #
  1532. # update list of layers
  1533. self.layerList.Update(self.mapDBInfo.layers)
  1534. self.layerList.Populate(update=True)
  1535. # update selected widgets
  1536. listOfLayers = map(str, self.mapDBInfo.layers.keys())
  1537. ### delete layer page
  1538. self.manageLayerBook.deleteLayer.SetItems(listOfLayers)
  1539. if len(listOfLayers) > 0:
  1540. self.manageLayerBook.deleteLayer.SetStringSelection(listOfLayers[0])
  1541. tableName = self.mapDBInfo.layers[int(listOfLayers[0])]['table']
  1542. maxLayer = max(self.mapDBInfo.layers.keys())
  1543. else:
  1544. tableName = ''
  1545. maxLayer = 0
  1546. self.manageLayerBook.deleteTable.SetLabel( \
  1547. _('Drop also linked attribute table (%s)') % \
  1548. tableName)
  1549. ### add layer page
  1550. self.manageLayerBook.addLayerWidgets['layer'][1].SetValue(\
  1551. maxLayer+1)
  1552. ### modify layer
  1553. self.manageLayerBook.modifyLayerWidgets['layer'][1].SetItems(listOfLayers)
  1554. self.manageLayerBook.OnChangeLayer(event=None)
  1555. # def OnMapClick(self, event):
  1556. # """
  1557. # Gets coordinates from mouse clicking on display window
  1558. # """
  1559. # Debug.msg(3, "VirtualAttributeList.OnMapClick()")
  1560. # # map coordinates
  1561. # x, y = self.mapdisp.MapWindow.Pixel2Cell(event.GetPositionTuple()[:])
  1562. # category = ""
  1563. # for line in os.popen("v.what east_north=%f,%f map=%s" %\
  1564. # (x,y,self.vectmap)).readlines():
  1565. # if "Category:" in line:
  1566. # category = line.strip().split(" ")[1]
  1567. # #print category
  1568. # for idx in range(self.GetItemCount()):
  1569. # item = self.GetItem(idx, 0)
  1570. # if item.GetText() == category:
  1571. # #print idx
  1572. # # self.Select(idx,True)
  1573. # self.EnsureVisible( idx )
  1574. # self.SetItemState(idx, wx.LIST_STATE_SELECTED, wx.LIST_STATE_SELECTED)
  1575. # else:
  1576. # # self.SetItemState(idx, wx.LIST_STATE_DESELECTED, wx.LIST_STATE_DESELECTED)
  1577. # self.Select(idx,False)
  1578. # # try:
  1579. # # os.environ["GRASS_MESSAGE_FORMAT"] = "gui"
  1580. # # cmd = "v.what -a east_north=%d,%d distance=%d map=%@%" % (x,y,100,self.tablename, self.self.mapset)
  1581. # # self.cmd_output.write(cmd+"\n----------\n")
  1582. # # p = Popen(cmd +" --verbose", shell=True, stdin=PIPE, stdout=PIPE, stderr=PIPE, close_fds=True)
  1583. # #
  1584. # # oline = p.stderr.readline()
  1585. # # while oline:
  1586. # # oline = oline.strip()
  1587. # # oline = p.stderr.readline()
  1588. # # if oline.find("GRASS_INFO_MESSAGE")>-1:
  1589. # # self.cmd_output.write(string.split(oline,maxsplit=1)[1]+"\n")
  1590. # # elif oline.find("GRASS_INFO_WARNING")>-1:
  1591. # # self.cmd_output.write("WARNING: "+string.split(oline,maxsplit=1)[1]+"\n")
  1592. # # elif oline.find("GRASS_INFO_ERROR")>-1:
  1593. # # self.cmd_output.write("ERROR: "+string.split(oline,maxsplit=1)[1]+"\n")
  1594. # #
  1595. # # oline = p.stdout.readline()
  1596. # # while oline:
  1597. # # oline = oline.strip()
  1598. # # self.cmd_output.write(oline+"\n")
  1599. # # print oline+"\n"
  1600. # # print >> sys.stderr, oline
  1601. # # oline = p.stdout.readline()
  1602. # #
  1603. # # if p.stdout < 0:
  1604. # # print >> sys.stderr, "Child was terminated by signal", p.stdout
  1605. # # elif p.stdout > 0:
  1606. # # print >> sys.stderr, p.stdout
  1607. # # pass
  1608. # # except OSError, e:
  1609. # # print >> sys.stderr, "Execution failed:", e
  1610. # # try:
  1611. # # os.environ["GRASS_MESSAGE_FORMAT"] = "gui"
  1612. # # cmd = "v.what -a east_north=%d,%d distance=%d map=%@%" % (x,y,100,self.tablename, self.self.mapset)
  1613. # # self.cmd_output.write(cmd+"\n----------\n")
  1614. # # p = Popen(cmd +" --verbose", shell=True, stdin=PIPE, stdout=PIPE, stderr=PIPE, close_fds=True)
  1615. # #
  1616. # # oline = p.stderr.readline()
  1617. # # while oline:
  1618. # # oline = oline.strip()
  1619. # # oline = p.stderr.readline()
  1620. # # if oline.find("GRASS_INFO_MESSAGE")>-1:
  1621. # # self.cmd_output.write(string.split(oline,maxsplit=1)[1]+"\n")
  1622. # # elif oline.find("GRASS_INFO_WARNING")>-1:
  1623. # # self.cmd_output.write("WARNING: "+string.split(oline,maxsplit=1)[1]+"\n")
  1624. # # elif oline.find("GRASS_INFO_ERROR")>-1:
  1625. # # self.cmd_output.write("ERROR: "+string.split(oline,maxsplit=1)[1]+"\n")
  1626. # #
  1627. # # oline = p.stdout.readline()
  1628. # # while oline:
  1629. # # oline = oline.strip()
  1630. # # self.cmd_output.write(oline+"\n")
  1631. # # print oline+"\n"
  1632. # # print >> sys.stderr, oline
  1633. # # oline = p.stdout.readline()
  1634. # #
  1635. # # if p.stdout < 0:
  1636. # # print >> sys.stderr, "Child was terminated by signal", p.stdout
  1637. # # elif p.stdout > 0:
  1638. # # print >> sys.stderr, p.stdout
  1639. # # pass
  1640. # # except OSError, e:
  1641. # # print >> sys.stderr, "Execution failed:", e
  1642. class TableListCtrl(wx.ListCtrl,
  1643. listmix.ListCtrlAutoWidthMixin):
  1644. # listmix.TextEditMixin):
  1645. """Table description list"""
  1646. def __init__(self, parent, id, table, columns, pos=wx.DefaultPosition,
  1647. size=wx.DefaultSize):
  1648. self.parent = parent
  1649. self.table = table
  1650. self.columns = columns
  1651. wx.ListCtrl.__init__(self, parent, id, pos, size,
  1652. style=wx.LC_REPORT | wx.LC_HRULES | wx.LC_VRULES |
  1653. wx.BORDER_NONE)
  1654. listmix.ListCtrlAutoWidthMixin.__init__(self)
  1655. # listmix.TextEditMixin.__init__(self)
  1656. def Update(self, table, columns):
  1657. """Update column description"""
  1658. self.table = table
  1659. self.columns = columns
  1660. def Populate(self, update=False):
  1661. """Populate the list"""
  1662. itemData = {} # requested by sorter
  1663. if not update:
  1664. headings = [_("Column name"), _("Data type"), _("Data length")]
  1665. i = 0
  1666. for h in headings:
  1667. self.InsertColumn(col=i, heading=h)
  1668. self.SetColumnWidth(col=i, width=150)
  1669. i += 1
  1670. else:
  1671. self.DeleteAllItems()
  1672. i = 0
  1673. for column in self.columns:
  1674. index = self.InsertStringItem(sys.maxint, str(column))
  1675. self.SetStringItem(index, 0, str(column))
  1676. self.SetStringItem(index, 1, str(self.table[column]['type']))
  1677. self.SetStringItem(index, 2, str(self.table[column]['length']))
  1678. self.SetItemData(index, i)
  1679. itemData[i] = (str(column),
  1680. str(self.table[column]['type']),
  1681. int(self.table[column]['length']))
  1682. i = i + 1
  1683. self.SendSizeEvent()
  1684. return itemData
  1685. class LayerListCtrl(wx.ListCtrl,
  1686. listmix.ListCtrlAutoWidthMixin):
  1687. # listmix.ColumnSorterMixin):
  1688. # listmix.TextEditMixin):
  1689. """Layer description list"""
  1690. def __init__(self, parent, id, layers,
  1691. pos=wx.DefaultPosition,
  1692. size=wx.DefaultSize):
  1693. self.parent = parent
  1694. self.layers = layers
  1695. wx.ListCtrl.__init__(self, parent, id, pos, size,
  1696. style=wx.LC_REPORT | wx.LC_HRULES | wx.LC_VRULES |
  1697. wx.BORDER_NONE)
  1698. listmix.ListCtrlAutoWidthMixin.__init__(self)
  1699. # listmix.TextEditMixin.__init__(self)
  1700. def Update(self, layers):
  1701. """Update description"""
  1702. self.layers = layers
  1703. def Populate(self, update=False):
  1704. """Populate the list"""
  1705. itemData = {} # requested by sorter
  1706. if not update:
  1707. headings = [_("Layer"), _("Driver"), _("Database"), _("Table"), _("Key")]
  1708. i = 0
  1709. for h in headings:
  1710. self.InsertColumn(col=i, heading=h)
  1711. i += 1
  1712. else:
  1713. self.DeleteAllItems()
  1714. i = 0
  1715. for layer in self.layers.keys():
  1716. index = self.InsertStringItem(sys.maxint, str(layer))
  1717. self.SetStringItem(index, 0, str(layer))
  1718. database = str(self.layers[layer]['database'])
  1719. driver = str(self.layers[layer]['driver'])
  1720. table = str(self.layers[layer]['table'])
  1721. key = str(self.layers[layer]['key'])
  1722. self.SetStringItem(index, 1, driver)
  1723. self.SetStringItem(index, 2, database)
  1724. self.SetStringItem(index, 3, table)
  1725. self.SetStringItem(index, 4, key)
  1726. self.SetItemData(index, i)
  1727. itemData[i] = (str(layer),
  1728. driver,
  1729. database,
  1730. table,
  1731. key)
  1732. i += 1
  1733. for i in range(self.GetColumnCount()):
  1734. self.SetColumnWidth(col=i, width=wx.LIST_AUTOSIZE)
  1735. if self.GetColumnWidth(col=i) < 60:
  1736. self.SetColumnWidth(col=i, width=60)
  1737. self.SendSizeEvent()
  1738. return itemData
  1739. class LayerBook(wx.Notebook):
  1740. """Manage layers (add, delete, modify)"""
  1741. def __init__(self, parent, id,
  1742. parentDialog,
  1743. style=wx.BK_DEFAULT):
  1744. wx.Notebook.__init__(self, parent, id, style=style)
  1745. self.parent = parent
  1746. self.parentDialog = parentDialog
  1747. self.mapDBInfo = self.parentDialog.mapDBInfo
  1748. #
  1749. # drivers
  1750. #
  1751. cmdDriver = gcmd.Command(['db.drivers',
  1752. '-p',
  1753. '--q'])
  1754. self.listOfDrivers = []
  1755. for drv in cmdDriver.ReadStdOutput():
  1756. self.listOfDrivers.append(drv.strip())
  1757. #
  1758. # get default values
  1759. #
  1760. self.defaultConnect = {}
  1761. cmdConnect = gcmd.Command(['db.connect',
  1762. '-p',
  1763. '--q'])
  1764. for line in cmdConnect.ReadStdOutput():
  1765. item, value = line.split(':')
  1766. self.defaultConnect[item.strip()] = value.strip()
  1767. if len(self.defaultConnect['driver']) == 0 or \
  1768. len(self.defaultConnect['database']) == 0:
  1769. raise gcmd.DBMError(_('Unable to determine default DB connection settings. '
  1770. 'Please define DB connection using db.connect module.'))
  1771. self.defaultTables = self.__getTables(self.defaultConnect['driver'],
  1772. self.defaultConnect['database'])
  1773. try:
  1774. self.defaultColumns = self.__getColumns(self.defaultConnect['driver'],
  1775. self.defaultConnect['database'],
  1776. self.defaultTables[0])
  1777. except IndexError:
  1778. self.defaultColumns = []
  1779. self.__createAddPage()
  1780. self.__createDeletePage()
  1781. self.__createModifyPage()
  1782. def __createAddPage(self):
  1783. """Add new layer"""
  1784. self.addPanel = wx.Panel(parent=self, id=wx.ID_ANY)
  1785. self.AddPage(page=self.addPanel, text=_("Add layer"))
  1786. try:
  1787. maxLayer = max(self.mapDBInfo.layers.keys())
  1788. except ValueError:
  1789. maxLayer = 0
  1790. #
  1791. # list of layer widgets (label, value)
  1792. #
  1793. self.addLayerWidgets = {'layer':
  1794. (wx.StaticText(parent=self.addPanel, id=wx.ID_ANY,
  1795. label='%s:' % _("Layer")),
  1796. wx.SpinCtrl(parent=self.addPanel, id=wx.ID_ANY, size=(65, -1),
  1797. initial=maxLayer+1,
  1798. min=1, max=1e6)),
  1799. 'driver':
  1800. (wx.StaticText(parent=self.addPanel, id=wx.ID_ANY,
  1801. label='%s:' % _("Driver")),
  1802. wx.Choice(parent=self.addPanel, id=wx.ID_ANY, size=(200, -1),
  1803. choices=self.listOfDrivers)),
  1804. 'database':
  1805. (wx.StaticText(parent=self.addPanel, id=wx.ID_ANY,
  1806. label='%s:' % _("Database")),
  1807. wx.TextCtrl(parent=self.addPanel, id=wx.ID_ANY,
  1808. value='',
  1809. style=wx.TE_PROCESS_ENTER)),
  1810. 'table':
  1811. (wx.StaticText(parent=self.addPanel, id=wx.ID_ANY,
  1812. label='%s:' % _("Table")),
  1813. wx.Choice(parent=self.addPanel, id=wx.ID_ANY, size=(200, -1),
  1814. choices=self.defaultTables)),
  1815. 'key':
  1816. (wx.StaticText(parent=self.addPanel, id=wx.ID_ANY,
  1817. label='%s:' % _("Key column")),
  1818. wx.Choice(parent=self.addPanel, id=wx.ID_ANY, size=(200, -1),
  1819. choices=self.defaultColumns))}
  1820. # set default values for widgets
  1821. self.addLayerWidgets['driver'][1].SetStringSelection(self.defaultConnect['driver'])
  1822. self.addLayerWidgets['database'][1].SetValue(self.defaultConnect['database'])
  1823. self.addLayerWidgets['table'][1].SetSelection(0)
  1824. self.addLayerWidgets['key'][1].SetSelection(0)
  1825. # events
  1826. self.addLayerWidgets['driver'][1].Bind(wx.EVT_CHOICE, self.OnDriverChanged)
  1827. self.addLayerWidgets['database'][1].Bind(wx.EVT_TEXT_ENTER, self.OnDatabaseChanged)
  1828. self.addLayerWidgets['table'][1].Bind(wx.EVT_CHOICE, self.OnTableChanged)
  1829. #
  1830. # list of table widgets
  1831. #
  1832. self.tableWidgets = {'table': (wx.StaticText(parent=self.addPanel, id=wx.ID_ANY,
  1833. label='%s:' % _("Table name")),
  1834. wx.TextCtrl(parent=self.addPanel, id=wx.ID_ANY,
  1835. value='',
  1836. style=wx.TE_PROCESS_ENTER)),
  1837. 'key': (wx.StaticText(parent=self.addPanel, id=wx.ID_ANY,
  1838. label='%s:' % _("Key column")),
  1839. wx.TextCtrl(parent=self.addPanel, id=wx.ID_ANY,
  1840. value='',
  1841. style=wx.TE_PROCESS_ENTER))}
  1842. # events
  1843. self.tableWidgets['table'][1].Bind(wx.EVT_TEXT_ENTER, self.OnCreateTable)
  1844. self.tableWidgets['key'][1].Bind(wx.EVT_TEXT_ENTER, self.OnCreateTable)
  1845. btnTable = wx.Button(self.addPanel, wx.ID_ANY, _("&Create table"),
  1846. size=(125,-1))
  1847. btnTable.Bind(wx.EVT_BUTTON, self.OnCreateTable)
  1848. btnLayer = wx.Button(self.addPanel, wx.ID_ANY, _("&Add layer"),
  1849. size=(125,-1))
  1850. btnLayer.Bind(wx.EVT_BUTTON, self.OnAddLayer)
  1851. btnDefault = wx.Button(self.addPanel, wx.ID_ANY, _("&Set default"),
  1852. size=(125,-1))
  1853. btnDefault.Bind(wx.EVT_BUTTON, self.OnSetDefault)
  1854. #
  1855. # do layout
  1856. #
  1857. pageSizer = wx.BoxSizer(wx.HORIZONTAL)
  1858. #
  1859. # layer description
  1860. #
  1861. layerBox = wx.StaticBox (parent=self.addPanel, id=wx.ID_ANY,
  1862. label=" %s " % (_("Layer description")))
  1863. layerSizer = wx.StaticBoxSizer(layerBox, wx.VERTICAL)
  1864. # data area
  1865. dataSizer = wx.FlexGridSizer(cols=2, hgap=5, vgap=5)
  1866. dataSizer.AddGrowableCol(1)
  1867. for key in ('layer', 'driver', 'database', 'table', 'key'):
  1868. label, value = self.addLayerWidgets[key]
  1869. dataSizer.Add(item=label,
  1870. flag=wx.ALIGN_CENTER_VERTICAL)
  1871. if label.GetLabel() == "Layer:":
  1872. dataSizer.Add(item=value,
  1873. flag=wx.ALIGN_CENTER_VERTICAL | wx.ALIGN_LEFT)
  1874. else:
  1875. dataSizer.Add(item=value,
  1876. flag=wx.ALIGN_CENTER_VERTICAL | wx.EXPAND)
  1877. layerSizer.Add(item=dataSizer,
  1878. proportion=1,
  1879. flag=wx.ALL | wx.EXPAND,
  1880. border=5)
  1881. btnSizer = wx.BoxSizer(wx.HORIZONTAL)
  1882. btnSizer.Add(item=btnDefault,
  1883. proportion=0,
  1884. flag=wx.ALL | wx.ALIGN_LEFT,
  1885. border=5)
  1886. btnSizer.Add(item=(5,5),
  1887. proportion=1,
  1888. flag=wx.ALL | wx.EXPAND,
  1889. border=5)
  1890. btnSizer.Add(item=btnLayer,
  1891. proportion=0,
  1892. flag=wx.ALL | wx.ALIGN_RIGHT,
  1893. border=5)
  1894. layerSizer.Add(item=btnSizer,
  1895. proportion=0,
  1896. flag=wx.ALL | wx.EXPAND,
  1897. border=0)
  1898. #
  1899. # table description
  1900. #
  1901. tableBox = wx.StaticBox (parent=self.addPanel, id=wx.ID_ANY,
  1902. label=" %s " % (_("Table description")))
  1903. tableSizer = wx.StaticBoxSizer(tableBox, wx.VERTICAL)
  1904. # data area
  1905. dataSizer = wx.FlexGridSizer(cols=2, hgap=5, vgap=5)
  1906. dataSizer.AddGrowableCol(1)
  1907. for key in ['table', 'key']:
  1908. label, value = self.tableWidgets[key]
  1909. dataSizer.Add(item=label,
  1910. flag=wx.ALIGN_CENTER_VERTICAL)
  1911. dataSizer.Add(item=value,
  1912. flag=wx.ALIGN_CENTER_VERTICAL | wx.EXPAND)
  1913. tableSizer.Add(item=dataSizer,
  1914. proportion=1,
  1915. flag=wx.ALL | wx.EXPAND,
  1916. border=5)
  1917. tableSizer.Add(item=btnTable,
  1918. proportion=0,
  1919. flag=wx.ALL | wx.ALIGN_BOTTOM | wx.ALIGN_RIGHT,
  1920. border=5)
  1921. pageSizer.Add(item=layerSizer,
  1922. proportion=1,
  1923. flag=wx.ALL | wx.EXPAND,
  1924. border=5)
  1925. pageSizer.Add(item=tableSizer,
  1926. proportion=1,
  1927. flag=wx.ALL | wx.EXPAND,
  1928. border=5)
  1929. self.addPanel.SetSizer(pageSizer)
  1930. def __createDeletePage(self):
  1931. """Delete layer"""
  1932. self.deletePanel = wx.Panel(parent=self, id=wx.ID_ANY)
  1933. self.AddPage(page=self.deletePanel, text=_("Delete selected layer"))
  1934. label = wx.StaticText(parent=self.deletePanel, id=wx.ID_ANY,
  1935. label='%s:' % _("Layer to detele"))
  1936. self.deleteLayer = wx.ComboBox(parent=self.deletePanel, id=wx.ID_ANY, size=(100, -1),
  1937. style=wx.CB_SIMPLE | wx.CB_READONLY,
  1938. choices=map(str, self.mapDBInfo.layers.keys()))
  1939. self.deleteLayer.SetSelection(0)
  1940. self.deleteLayer.Bind(wx.EVT_COMBOBOX, self.OnChangeLayer)
  1941. try:
  1942. tableName = self.mapDBInfo.layers[int(self.deleteLayer.GetStringSelection())]['table']
  1943. except ValueError:
  1944. tableName = ''
  1945. self.deleteTable = wx.CheckBox(parent=self.deletePanel, id=wx.ID_ANY,
  1946. label=_('Drop also linked attribute table (%s)') % \
  1947. tableName)
  1948. if tableName == '':
  1949. self.deleteLayer.Enable(False)
  1950. self.deleteTable.Enable(False)
  1951. btnDelete = wx.Button(self.deletePanel, wx.ID_DELETE, _("&Delete layer"),
  1952. size=(125,-1))
  1953. btnDelete.Bind(wx.EVT_BUTTON, self.OnDeleteLayer)
  1954. #
  1955. # do layout
  1956. #
  1957. pageSizer = wx.BoxSizer(wx.VERTICAL)
  1958. dataSizer = wx.BoxSizer(wx.VERTICAL)
  1959. flexSizer = wx.FlexGridSizer(cols=2, hgap=5, vgap=5)
  1960. flexSizer.AddGrowableCol(2)
  1961. flexSizer.Add(item=label,
  1962. flag=wx.ALIGN_CENTER_VERTICAL)
  1963. flexSizer.Add(item=self.deleteLayer,
  1964. flag=wx.ALIGN_CENTER_VERTICAL)
  1965. dataSizer.Add(item=flexSizer,
  1966. proportion=0,
  1967. flag=wx.ALL | wx.EXPAND,
  1968. border=5)
  1969. dataSizer.Add(item=self.deleteTable,
  1970. proportion=0,
  1971. flag=wx.ALL | wx.EXPAND,
  1972. border=5)
  1973. pageSizer.Add(item=dataSizer,
  1974. proportion=1,
  1975. flag=wx.ALL | wx.EXPAND,
  1976. border=5)
  1977. pageSizer.Add(item=btnDelete,
  1978. proportion=0,
  1979. flag=wx.ALL | wx.ALIGN_RIGHT,
  1980. border=5)
  1981. self.deletePanel.SetSizer(pageSizer)
  1982. def __createModifyPage(self):
  1983. """Modify layer"""
  1984. self.modifyPanel = wx.Panel(parent=self, id=wx.ID_ANY)
  1985. self.AddPage(page=self.modifyPanel, text=_("Modify selected layer"))
  1986. #
  1987. # list of layer widgets (label, value)
  1988. #
  1989. self.modifyLayerWidgets = {'layer':
  1990. (wx.StaticText(parent=self.modifyPanel, id=wx.ID_ANY,
  1991. label='%s:' % _("Layer")),
  1992. wx.ComboBox(parent=self.modifyPanel, id=wx.ID_ANY,
  1993. size=(100, -1),
  1994. style=wx.CB_SIMPLE | wx.CB_READONLY,
  1995. choices=map(str,
  1996. self.mapDBInfo.layers.keys()))),
  1997. 'driver':
  1998. (wx.StaticText(parent=self.modifyPanel, id=wx.ID_ANY,
  1999. label='%s:' % _("Driver")),
  2000. wx.Choice(parent=self.modifyPanel, id=wx.ID_ANY,
  2001. size=(200, -1),
  2002. choices=self.listOfDrivers)),
  2003. 'database':
  2004. (wx.StaticText(parent=self.modifyPanel, id=wx.ID_ANY,
  2005. label='%s:' % _("Database")),
  2006. wx.TextCtrl(parent=self.modifyPanel, id=wx.ID_ANY,
  2007. value='', size=(350, -1),
  2008. style=wx.TE_PROCESS_ENTER)),
  2009. 'table':
  2010. (wx.StaticText(parent=self.modifyPanel, id=wx.ID_ANY,
  2011. label='%s:' % _("Table")),
  2012. wx.Choice(parent=self.modifyPanel, id=wx.ID_ANY,
  2013. size=(200, -1),
  2014. choices=self.defaultTables)),
  2015. 'key':
  2016. (wx.StaticText(parent=self.modifyPanel, id=wx.ID_ANY,
  2017. label='%s:' % _("Key column")),
  2018. wx.Choice(parent=self.modifyPanel, id=wx.ID_ANY,
  2019. size=(200, -1),
  2020. choices=self.defaultColumns))}
  2021. # set default values for widgets
  2022. self.modifyLayerWidgets['layer'][1].SetSelection(0)
  2023. try:
  2024. layer = int(self.modifyLayerWidgets['layer'][1].GetStringSelection())
  2025. except ValueError:
  2026. layer = None
  2027. for label in self.modifyLayerWidgets.keys():
  2028. self.modifyLayerWidgets[label][1].Enable(False)
  2029. if layer:
  2030. driver = self.mapDBInfo.layers[layer]['driver']
  2031. database = self.mapDBInfo.layers[layer]['database']
  2032. table = self.mapDBInfo.layers[layer]['table']
  2033. listOfColumns = self.__getColumns(driver, database, table)
  2034. self.modifyLayerWidgets['driver'][1].SetStringSelection(driver)
  2035. self.modifyLayerWidgets['database'][1].SetValue(database)
  2036. if table in self.modifyLayerWidgets['table'][1].GetItems():
  2037. self.modifyLayerWidgets['table'][1].SetStringSelection(table)
  2038. else:
  2039. if self.defaultConnect['schema'] != '':
  2040. table = self.defaultConnect['schema'] + table # try with default schema
  2041. else:
  2042. table = 'public.' + table # try with 'public' schema
  2043. self.modifyLayerWidgets['table'][1].SetStringSelection(table)
  2044. self.modifyLayerWidgets['key'][1].SetItems(listOfColumns)
  2045. self.modifyLayerWidgets['key'][1].SetSelection(0)
  2046. # events
  2047. self.modifyLayerWidgets['layer'][1].Bind(wx.EVT_COMBOBOX, self.OnChangeLayer)
  2048. # self.modifyLayerWidgets['driver'][1].Bind(wx.EVT_CHOICE, self.OnDriverChanged)
  2049. # self.modifyLayerWidgets['database'][1].Bind(wx.EVT_TEXT_ENTER, self.OnDatabaseChanged)
  2050. # self.modifyLayerWidgets['table'][1].Bind(wx.EVT_CHOICE, self.OnTableChanged)
  2051. btnModify = wx.Button(self.modifyPanel, wx.ID_DELETE, _("&Modify layer"),
  2052. size=(125,-1))
  2053. btnModify.Bind(wx.EVT_BUTTON, self.OnModifyLayer)
  2054. #
  2055. # do layout
  2056. #
  2057. pageSizer = wx.BoxSizer(wx.VERTICAL)
  2058. # data area
  2059. dataSizer = wx.FlexGridSizer(cols=2, hgap=5, vgap=5)
  2060. dataSizer.AddGrowableCol(1)
  2061. for key in ('layer', 'driver', 'database', 'table', 'key'):
  2062. label, value = self.modifyLayerWidgets[key]
  2063. dataSizer.Add(item=label,
  2064. flag=wx.ALIGN_CENTER_VERTICAL)
  2065. if label.GetLabel() == "Layer:":
  2066. dataSizer.Add(item=value,
  2067. flag=wx.ALIGN_CENTER_VERTICAL | wx.ALIGN_LEFT)
  2068. else:
  2069. dataSizer.Add(item=value,
  2070. flag=wx.ALIGN_CENTER_VERTICAL)
  2071. pageSizer.Add(item=dataSizer,
  2072. proportion=1,
  2073. flag=wx.ALL | wx.EXPAND,
  2074. border=5)
  2075. pageSizer.Add(item=btnModify,
  2076. proportion=0,
  2077. flag=wx.ALL | wx.ALIGN_RIGHT,
  2078. border=5)
  2079. self.modifyPanel.SetSizer(pageSizer)
  2080. def __getTables(self, driver, database):
  2081. """Get list of tables for given driver and database"""
  2082. tables = []
  2083. cmdTable = gcmd.Command(['db.tables',
  2084. '-p', '--q',
  2085. 'driver=%s' % driver,
  2086. 'database=%s' % database])
  2087. for table in cmdTable.ReadStdOutput():
  2088. tables.append(table)
  2089. return tables
  2090. def __getColumns(self, driver, database, table):
  2091. """Get list of column of given table"""
  2092. columns = []
  2093. cmdColumn = gcmd.Command(['db.columns',
  2094. '--q',
  2095. 'driver=%s' % driver,
  2096. 'database=%s' % database,
  2097. 'table=%s' % table])
  2098. for column in cmdColumn.ReadStdOutput():
  2099. columns.append(column)
  2100. return columns
  2101. def OnDriverChanged(self, event):
  2102. """Driver selection changed, update list of tables"""
  2103. driver = event.GetString()
  2104. database = self.addLayerWidgets['database'][1].GetValue()
  2105. winTable = self.addLayerWidgets['table'][1]
  2106. winKey = self.addLayerWidgets['key'][1]
  2107. tables = self.__getTables(driver, database)
  2108. winTable.SetItems(tables)
  2109. winTable.SetSelection(0)
  2110. if len(tables) == 0:
  2111. winKey.SetItems([])
  2112. event.Skip()
  2113. def OnDatabaseChanged(self, event):
  2114. """Database selection changed, update list of tables"""
  2115. event.Skip()
  2116. def OnTableChanged(self, event):
  2117. """Table name changed, update list of columns"""
  2118. driver = self.addLayerWidgets['driver'][1].GetStringSelection()
  2119. database = self.addLayerWidgets['database'][1].GetValue()
  2120. table = event.GetString()
  2121. win = self.addLayerWidgets['key'][1]
  2122. cols = self.__getColumns(driver, database, table)
  2123. win.SetItems(cols)
  2124. win.SetSelection(0)
  2125. event.Skip()
  2126. def OnSetDefault(self, event):
  2127. """Set default values"""
  2128. driver = self.addLayerWidgets['driver'][1]
  2129. database = self.addLayerWidgets['database'][1]
  2130. table = self.addLayerWidgets['table'][1]
  2131. key = self.addLayerWidgets['key'][1]
  2132. driver.SetStringSelection(self.defaultConnect['driver'])
  2133. database.SetValue(self.defaultConnect['database'])
  2134. tables = self.__getTables(self.defaultConnect['driver'],
  2135. self.defaultConnect['database'])
  2136. table.SetItems(tables)
  2137. table.SetSelection(0)
  2138. if len(tables) == 0:
  2139. key.SetItems([])
  2140. else:
  2141. cols = self.__getColumns(self.defaultConnect['driver'],
  2142. self.defaultConnect['database'],
  2143. tables[0])
  2144. key.SetItems(cols)
  2145. key.SetSelection(0)
  2146. event.Skip()
  2147. def OnCreateTable(self, event):
  2148. """Create new table (name and key column given)"""
  2149. driver = self.addLayerWidgets['driver'][1].GetStringSelection()
  2150. database = self.addLayerWidgets['database'][1].GetValue()
  2151. table = self.tableWidgets['table'][1].GetValue()
  2152. key = self.tableWidgets['key'][1].GetValue()
  2153. if not table or not key:
  2154. wx.MessageBox(parent=self,
  2155. message=_("Unable to create new table. "
  2156. "Table name or key column name is missing."),
  2157. caption=_("Error"), style=wx.OK | wx.ICON_ERROR | wx.CENTRE)
  2158. return
  2159. # create table
  2160. sql = 'CREATE TABLE %s (%s INTEGER)' % (table, key)
  2161. gcmd.Command(['db.execute',
  2162. '--q',
  2163. 'driver=%s' % driver,
  2164. 'database=%s' % database],
  2165. stdin=sql)
  2166. # update list of tables
  2167. tableList = self.addLayerWidgets['table'][1]
  2168. tableList.SetItems(self.__getTables(driver, database))
  2169. tableList.SetStringSelection(table)
  2170. # update key column selection
  2171. keyList = self.addLayerWidgets['key'][1]
  2172. keyList.SetItems(self.__getColumns(driver, database, table))
  2173. keyList.SetStringSelection(key)
  2174. event.Skip()
  2175. def OnAddLayer(self, event):
  2176. """Add new layer to vector map"""
  2177. layer = int(self.addLayerWidgets['layer'][1].GetValue())
  2178. layerWin = self.addLayerWidgets['layer'][1]
  2179. driver = self.addLayerWidgets['driver'][1].GetStringSelection()
  2180. database = self.addLayerWidgets['database'][1].GetValue()
  2181. table = self.addLayerWidgets['table'][1].GetStringSelection()
  2182. key = self.addLayerWidgets['key'][1].GetStringSelection()
  2183. if layer in self.mapDBInfo.layers.keys():
  2184. wx.MessageBox(parent=self,
  2185. message=_("Unable to add new layer to vector map <%(vector)s>. "
  2186. "Layer %(layer)d already exists.") %
  2187. {'vector' : self.mapDBInfo.map, 'layer' : layer},
  2188. caption=_("Error"), style=wx.OK | wx.ICON_ERROR | wx.CENTRE)
  2189. return
  2190. connectCmd = gcmd.Command(['v.db.connect',
  2191. '--q',
  2192. 'map=%s' % self.mapDBInfo.map,
  2193. 'driver=%s' % driver,
  2194. 'database=%s' % database,
  2195. 'table=%s' % table,
  2196. 'key=%s' % key,
  2197. 'layer=%d' % layer])
  2198. if connectCmd.returncode == 0:
  2199. # update dialog (only for new layer)
  2200. self.parentDialog.UpdateDialog(layer=layer)
  2201. # update db info
  2202. self.mapDBInfo = self.parentDialog.mapDBInfo
  2203. # increase layer number
  2204. layerWin.SetValue(layer+1)
  2205. if len(self.mapDBInfo.layers.keys()) == 1:
  2206. # first layer add --- enable previously disabled widgets
  2207. self.deleteLayer.Enable()
  2208. self.deleteTable.Enable()
  2209. for label in self.modifyLayerWidgets.keys():
  2210. self.modifyLayerWidgets[label][1].Enable()
  2211. def OnDeleteLayer(self, event):
  2212. """Delete layer"""
  2213. try:
  2214. layer = int(self.deleteLayer.GetValue())
  2215. except:
  2216. return
  2217. gcmd.Command(['v.db.connect',
  2218. '-d', '--q',
  2219. 'map=%s' % self.mapDBInfo.map,
  2220. 'layer=%d' % layer])
  2221. # drop also table linked to layer which is deleted
  2222. if self.deleteTable.IsChecked():
  2223. driver = self.addLayerWidgets['driver'][1].GetStringSelection()
  2224. database = self.addLayerWidgets['database'][1].GetValue()
  2225. table = self.mapDBInfo.layers[layer]['table']
  2226. sql = 'DROP TABLE %s' % (table)
  2227. gcmd.Command(['db.execute',
  2228. '--q',
  2229. 'driver=%s' % driver,
  2230. 'database=%s' % database],
  2231. stdin=sql)
  2232. # update dialog
  2233. self.parentDialog.UpdateDialog(layer=layer)
  2234. # update db info
  2235. self.mapDBInfo = self.parentDialog.mapDBInfo
  2236. if len(self.mapDBInfo.layers.keys()) == 0:
  2237. # disable selected widgets
  2238. self.deleteLayer.Enable(False)
  2239. self.deleteTable.Enable(False)
  2240. for label in self.modifyLayerWidgets.keys():
  2241. self.modifyLayerWidgets[label][1].Enable(False)
  2242. event.Skip()
  2243. def OnChangeLayer(self, event):
  2244. """Layer number of layer to be deleted is changed"""
  2245. try:
  2246. layer = int(event.GetString())
  2247. except:
  2248. try:
  2249. layer = self.mapDBInfo.layers.keys()[0]
  2250. except:
  2251. return
  2252. if self.GetCurrentPage() == self.modifyPanel:
  2253. driver = self.mapDBInfo.layers[layer]['driver']
  2254. database = self.mapDBInfo.layers[layer]['database']
  2255. table = self.mapDBInfo.layers[layer]['table']
  2256. listOfColumns = self.__getColumns(driver, database, table)
  2257. self.modifyLayerWidgets['driver'][1].SetStringSelection(driver)
  2258. self.modifyLayerWidgets['database'][1].SetValue(database)
  2259. self.modifyLayerWidgets['table'][1].SetStringSelection(table)
  2260. self.modifyLayerWidgets['key'][1].SetItems(listOfColumns)
  2261. self.modifyLayerWidgets['key'][1].SetSelection(0)
  2262. else:
  2263. self.deleteTable.SetLabel(_('Drop also linked attribute table (%s)') % \
  2264. self.mapDBInfo.layers[layer]['table'])
  2265. if event:
  2266. event.Skip()
  2267. def OnModifyLayer(self, event):
  2268. """Modify layer connection settings"""
  2269. layer = int(self.modifyLayerWidgets['layer'][1].GetStringSelection())
  2270. modify = False
  2271. if self.modifyLayerWidgets['driver'][1].GetStringSelection() != \
  2272. self.mapDBInfo.layers[layer]['driver'] or \
  2273. self.modifyLayerWidgets['database'][1].GetStringSelection() != \
  2274. self.mapDBInfo.layers[layer]['database'] or \
  2275. self.modifyLayerWidgets['table'][1].GetStringSelection() != \
  2276. self.mapDBInfo.layers[layer]['table'] or \
  2277. self.modifyLayerWidgets['key'][1].GetStringSelection() != \
  2278. self.mapDBInfo.layers[layer]['key']:
  2279. modify = True
  2280. if modify:
  2281. # delete layer
  2282. gcmd.Command(['v.db.connect',
  2283. '-d', '--q',
  2284. 'map=%s' % self.mapDBInfo.map,
  2285. 'layer=%d' % layer])
  2286. # add modified layer
  2287. gcmd.Command(['v.db.connect',
  2288. '--q',
  2289. 'map=%s' % self.mapDBInfo.map,
  2290. 'driver=%s' % \
  2291. self.modifyLayerWidgets['driver'][1].GetStringSelection(),
  2292. 'database=%s' % \
  2293. self.modifyLayerWidgets['database'][1].GetValue(),
  2294. 'table=%s' % \
  2295. self.modifyLayerWidgets['table'][1].GetStringSelection(),
  2296. 'key=%s' % \
  2297. self.modifyLayerWidgets['key'][1].GetStringSelection(),
  2298. 'layer=%d' % layer])
  2299. # update dialog (only for new layer)
  2300. self.parentDialog.UpdateDialog(layer=layer)
  2301. # update db info
  2302. self.mapDBInfo = self.parentDialog.mapDBInfo
  2303. event.Skip()
  2304. class DisplayAttributesDialog(wx.Dialog):
  2305. """
  2306. Standard dialog used to add/update/display attributes linked
  2307. to the vector map.
  2308. Attribute data can be selected based on layer and category number
  2309. or coordinates.
  2310. @param parent
  2311. @param map vector map
  2312. @param query query coordinates and distance (used for v.edit)
  2313. @param cats {layer: cats}
  2314. @param line feature id (requested for cats)
  2315. @param style
  2316. @param pos
  2317. @param action (add, update, display)
  2318. """
  2319. def __init__(self, parent, map,
  2320. query=None, cats=None, line=None,
  2321. style=wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER,
  2322. pos=wx.DefaultPosition,
  2323. action="add"):
  2324. self.parent = parent # mapdisplay.BufferedWindow
  2325. self.map = map
  2326. self.action = action
  2327. # id of selected line
  2328. self.line = None
  2329. # get layer/table/column information
  2330. self.mapDBInfo = VectorDBInfo(self.map)
  2331. layers = self.mapDBInfo.layers.keys() # get available layers
  2332. # check if db connection / layer exists
  2333. if len(layers) <= 0:
  2334. label = _("Database connection "
  2335. "is not defined in DB file.")
  2336. wx.MessageBox(parent=self.parent,
  2337. message=_("No attribute table linked to "
  2338. "vector map <%(vector)s> found. %(msg)s"
  2339. "\nYou can disable this message from digitization settings. Or "
  2340. "you can create and link attribute table to the vector map "
  2341. "using Attribute Table Manager.") %
  2342. {'vector' : self.map, 'msg' : label},
  2343. caption=_("Message"), style=wx.OK | wx.ICON_EXCLAMATION | wx.CENTRE)
  2344. self.mapDBInfo = None
  2345. wx.Dialog.__init__(self, parent=self.parent, id=wx.ID_ANY,
  2346. title="", style=style, pos=pos)
  2347. # dialog body
  2348. mainSizer = wx.BoxSizer(wx.VERTICAL)
  2349. # notebook
  2350. self.notebook = wx.Notebook(parent=self, id=wx.ID_ANY, style=wx.BK_DEFAULT)
  2351. self.closeDialog = wx.CheckBox(parent=self, id=wx.ID_ANY,
  2352. label=_("Close dialog on submit"))
  2353. self.closeDialog.SetValue(True)
  2354. self.UpdateDialog(query, cats, line)
  2355. # set title
  2356. if self.action == "update":
  2357. self.SetTitle(_("Update attributes"))
  2358. elif self.action == "add":
  2359. self.SetTitle(_("Add attributes"))
  2360. else:
  2361. self.SetTitle(_("Display attributes"))
  2362. # buttons
  2363. btnCancel = wx.Button(self, wx.ID_CANCEL)
  2364. btnReset = wx.Button(self, wx.ID_UNDO, _("&Reload"))
  2365. btnSubmit = wx.Button(self, wx.ID_OK, _("&Submit"))
  2366. btnSizer = wx.StdDialogButtonSizer()
  2367. btnSizer.AddButton(btnCancel)
  2368. btnSizer.AddButton(btnReset)
  2369. btnSizer.SetNegativeButton(btnReset)
  2370. btnSubmit.SetDefault()
  2371. btnSizer.AddButton(btnSubmit)
  2372. btnSizer.Realize()
  2373. mainSizer.Add(item=self.notebook, proportion=1, flag=wx.EXPAND | wx.ALL, border=5)
  2374. mainSizer.Add(item=self.closeDialog, proportion=0, flag=wx.EXPAND | wx.LEFT | wx.RIGHT,
  2375. border=5)
  2376. mainSizer.Add(item=btnSizer, proportion=0,
  2377. flag=wx.EXPAND | wx.ALL | wx.ALIGN_CENTER, border=5)
  2378. # bindigs
  2379. btnReset.Bind(wx.EVT_BUTTON, self.OnReset)
  2380. btnSubmit.Bind(wx.EVT_BUTTON, self.OnSubmit)
  2381. btnCancel.Bind(wx.EVT_BUTTON, self.OnCancel)
  2382. self.SetSizer(mainSizer)
  2383. mainSizer.Fit(self)
  2384. # set min size for dialog
  2385. self.SetMinSize(self.GetBestSize())
  2386. if self.notebook.GetPageCount() == 0:
  2387. Debug.msg(2, "DisplayAttributesDialog(): Nothing found!")
  2388. self.mapDBInfo = None
  2389. def __SelectAttributes(self, layer):
  2390. """Select attributes"""
  2391. pass
  2392. def OnSQLStatement(self, event):
  2393. """Update SQL statement"""
  2394. pass
  2395. def GetSQLString(self, updateValues=False):
  2396. """Create SQL statement string based on self.sqlStatement
  2397. If updateValues is True, update dataFrame according to values
  2398. in textfields.
  2399. """
  2400. sqlCommands = []
  2401. # find updated values for each layer/category
  2402. for layer in self.mapDBInfo.layers.keys(): # for each layer
  2403. table = self.mapDBInfo.layers[layer]["table"]
  2404. key = self.mapDBInfo.layers[layer]["key"]
  2405. columns = self.mapDBInfo.tables[table]
  2406. for idx in range(len(columns[key]['values'])): # for each category
  2407. updatedColumns = []
  2408. updatedValues = []
  2409. for name in columns.keys():
  2410. if name == key:
  2411. cat = columns[name]['values'][idx]
  2412. continue
  2413. type = columns[name]['type']
  2414. value = columns[name]['values'][idx]
  2415. id = columns[name]['ids'][idx]
  2416. try:
  2417. newvalue = self.FindWindowById(id).GetValue()
  2418. except:
  2419. newvalue = self.FindWindowById(id).GetLabel()
  2420. if newvalue == '':
  2421. newvalue = None
  2422. if newvalue != value:
  2423. updatedColumns.append(name)
  2424. if newvalue is None:
  2425. updatedValues.append('NULL')
  2426. else:
  2427. if type != 'character':
  2428. updatedValues.append(newvalue)
  2429. else:
  2430. updatedValues.append("'" + newvalue + "'")
  2431. columns[name]['values'][idx] = newvalue
  2432. if self.action != "add" and len(updatedValues) == 0:
  2433. continue
  2434. if self.action == "add":
  2435. sqlString = "INSERT INTO %s (%s," % (table, key)
  2436. else:
  2437. sqlString = "UPDATE %s SET " % table
  2438. for idx in range(len(updatedColumns)):
  2439. name = updatedColumns[idx]
  2440. if self.action == "add":
  2441. sqlString += name + ","
  2442. else:
  2443. sqlString += name + "=" + updatedValues[idx] + ","
  2444. sqlString = sqlString[:-1] # remove last comma
  2445. if self.action == "add":
  2446. sqlString += ") VALUES (%s," % cat
  2447. for value in updatedValues:
  2448. sqlString += str(value) + ","
  2449. sqlString = sqlString[:-1] # remove last comma
  2450. sqlString += ")"
  2451. else:
  2452. sqlString += " WHERE cat=%s" % cat
  2453. sqlCommands.append(sqlString)
  2454. # for each category
  2455. # for each layer END
  2456. Debug.msg(3, "DisplayAttributesDialog.GetSQLString(): %s" % sqlCommands)
  2457. return sqlCommands
  2458. def OnReset(self, event):
  2459. """Reset form"""
  2460. for layer in self.mapDBInfo.layers.keys():
  2461. table = self.mapDBInfo.layers[layer]["table"]
  2462. key = self.mapDBInfo.layers[layer]["key"]
  2463. columns = self.mapDBInfo.tables[table]
  2464. for idx in range(len(columns[key]['values'])):
  2465. for name in columns.keys():
  2466. type = columns[name]['type']
  2467. value = columns[name]['values'][idx]
  2468. id = columns[name]['ids'][idx]
  2469. if name != key:
  2470. self.FindWindowById(id).SetValue(str(value))
  2471. def OnCancel(self, event):
  2472. """Cancel button pressed"""
  2473. self.parent.parent.dialogs['attributes'] = None
  2474. if self.parent.parent.digit:
  2475. self.parent.parent.digit.driver.SetSelected([])
  2476. self.parent.UpdateMap(render=False)
  2477. else:
  2478. self.parent.parent.OnRender(None)
  2479. self.Close()
  2480. def OnSubmit(self, event):
  2481. """Submit records"""
  2482. sqlCommands = self.GetSQLString(updateValues=True)
  2483. if len(sqlCommands) > 0:
  2484. sqlfile = tempfile.NamedTemporaryFile(mode="w")
  2485. for sql in sqlCommands:
  2486. sqlfile.file.write(sql + ";\n")
  2487. sqlfile.file.flush()
  2488. gcmd.Command(cmd=["db.execute",
  2489. "--q",
  2490. "input=%s" % sqlfile.name])
  2491. if self.closeDialog.IsChecked():
  2492. self.OnCancel(event)
  2493. def GetLine(self):
  2494. """Get id of selected vector object or 'None' if nothing selected"""
  2495. return self.line
  2496. def UpdateDialog(self, query=None, cats=None, line=None):
  2497. """Update dialog
  2498. Return True if updated otherwise False
  2499. """
  2500. self.line = line
  2501. if not self.mapDBInfo:
  2502. return False
  2503. self.mapDBInfo.Reset()
  2504. layers = self.mapDBInfo.layers.keys() # get available layers
  2505. # id of selected line
  2506. if query: # select by position
  2507. self.line, nselected = self.mapDBInfo.SelectByPoint(query[0],
  2508. query[1])
  2509. # reset notebook
  2510. self.notebook.DeleteAllPages()
  2511. for layer in layers: # for each layer
  2512. if not query: # select by layer/cat
  2513. if cats.has_key(layer):
  2514. for cat in cats[layer]:
  2515. nselected = self.mapDBInfo.SelectFromTable(layer,
  2516. where="%s=%d" % \
  2517. (self.mapDBInfo.layers[layer]['key'],
  2518. cat))
  2519. else:
  2520. nselected = 0
  2521. # if nselected <= 0 and self.action != "add":
  2522. # continue # nothing selected ...
  2523. if self.action == "add":
  2524. if nselected <= 0:
  2525. if cats.has_key(layer):
  2526. table = self.mapDBInfo.layers[layer]["table"]
  2527. key = self.mapDBInfo.layers[layer]["key"]
  2528. columns = self.mapDBInfo.tables[table]
  2529. for name in columns.keys():
  2530. if name == key:
  2531. for cat in cats[layer]:
  2532. self.mapDBInfo.tables[table][name]['values'].append(cat)
  2533. else:
  2534. self.mapDBInfo.tables[table][name]['values'].append(None)
  2535. else: # change status 'add' -> 'update'
  2536. self.action = "update"
  2537. # use scrolled panel instead (and fix initial max height of the window to 480px)
  2538. ### panel = wx.Panel(parent=self.notebook, id=wx.ID_ANY)
  2539. panel = scrolled.ScrolledPanel(parent=self.notebook, id=wx.ID_ANY,
  2540. size=(-1, 150))
  2541. panel.SetupScrolling(scroll_x=False)
  2542. self.notebook.AddPage(page=panel, text=" %s %d " % (_("Layer"), layer))
  2543. # notebook body
  2544. border = wx.BoxSizer(wx.VERTICAL)
  2545. table = self.mapDBInfo.layers[layer]["table"]
  2546. key = self.mapDBInfo.layers[layer]["key"]
  2547. columns = self.mapDBInfo.tables[table]
  2548. # value
  2549. if len(columns[key]['values']) == 0: # no cats
  2550. sizer = wx.BoxSizer(wx.VERTICAL)
  2551. txt = wx.StaticText(parent=panel, id=wx.ID_ANY,
  2552. label=_("No database record available."))
  2553. sizer.Add(txt, proportion=1,
  2554. flag=wx.ALIGN_CENTER_VERTICAL | wx.ALIGN_CENTER | wx.EXPAND)
  2555. border.Add(item=sizer, proportion=1,
  2556. flag=wx.ALL | wx.EXPAND | wx.ALIGN_CENTER_VERTICAL | wx.ALIGN_CENTER,
  2557. border=10)
  2558. panel.SetSizer(border)
  2559. continue
  2560. for idx in range(len(columns[key]['values'])):
  2561. flexSizer = wx.FlexGridSizer (cols=4, hgap=3, vgap=3)
  2562. flexSizer.AddGrowableCol(3)
  2563. # columns
  2564. for name in columns.keys():
  2565. vtype = columns[name]['type']
  2566. if columns[name]['values'][idx] is not None:
  2567. value = str(columns[name]['values'][idx])
  2568. else:
  2569. value = ''
  2570. if name == key:
  2571. box = wx.StaticBox (parent=panel, id=wx.ID_ANY,
  2572. label=" %s %s " % (_("Category"), value))
  2573. boxFont = self.GetFont()
  2574. boxFont.SetWeight(wx.FONTWEIGHT_BOLD)
  2575. box.SetFont(boxFont)
  2576. sizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  2577. colValue = box
  2578. else:
  2579. colName = wx.StaticText(parent=panel, id=wx.ID_ANY,
  2580. label=name)
  2581. colType = wx.StaticText(parent=panel, id=wx.ID_ANY,
  2582. label="[" + vtype.lower() + "]")
  2583. delimiter = wx.StaticText(parent=panel, id=wx.ID_ANY, label=":")
  2584. colValue = wx.TextCtrl(parent=panel, id=wx.ID_ANY, value=value) # TODO: validator
  2585. colValue.SetName(name)
  2586. self.Bind(wx.EVT_TEXT, self.OnSQLStatement, colValue)
  2587. flexSizer.Add(colName, proportion=0,
  2588. flag=wx.FIXED_MINSIZE | wx.ALIGN_CENTER_VERTICAL)
  2589. flexSizer.Add(colType, proportion=0,
  2590. flag=wx.FIXED_MINSIZE | wx.ALIGN_CENTER_VERTICAL)
  2591. flexSizer.Add(delimiter, proportion=0,
  2592. flag=wx.FIXED_MINSIZE | wx.ALIGN_CENTER_VERTICAL)
  2593. flexSizer.Add(colValue, proportion=1,
  2594. flag=wx.EXPAND | wx.ALIGN_CENTER_VERTICAL)
  2595. # add widget reference to self.columns
  2596. columns[name]['ids'].append(colValue.GetId()) # name, type, values, id
  2597. # for each attribute (including category) END
  2598. sizer.Add(item=flexSizer, proportion=1, flag=wx.ALL | wx.EXPAND, border=1)
  2599. border.Add(item=sizer, proportion=1, flag=wx.ALL | wx.EXPAND, border=5)
  2600. # for each category END
  2601. panel.SetSizer(border)
  2602. # for each layer END
  2603. self.Layout()
  2604. return True
  2605. class VectorDBInfo(gselect.VectorDBInfo):
  2606. """Class providing information about attribute tables
  2607. linked to the vector map"""
  2608. def __init__(self, map):
  2609. gselect.VectorDBInfo.__init__(self, map)
  2610. def GetColumns(self, table):
  2611. """Return list of columns names (based on their index)"""
  2612. names = [''] * len(self.tables[table].keys())
  2613. for name, desc in self.tables[table].iteritems():
  2614. names[desc['index']] = name
  2615. return names
  2616. def SelectByPoint(self, queryCoords, qdist):
  2617. """Get attributes by coordinates (all available layers)
  2618. Return line id or None if no line is found"""
  2619. line = None
  2620. nselected = 0
  2621. cmdWhat = gcmd.Command(cmd=['v.what',
  2622. '-a', '--q',
  2623. 'map=%s' % self.map,
  2624. 'east_north=%f,%f' % \
  2625. (float(queryCoords[0]), float(queryCoords[1])),
  2626. 'distance=%f' % qdist], stderr=None)
  2627. if cmdWhat.returncode == 0:
  2628. read = False
  2629. for item in cmdWhat.ReadStdOutput():
  2630. litem = item.lower()
  2631. if read:
  2632. name, value = item.split(':')
  2633. name = name.strip()
  2634. # append value to the column
  2635. try:
  2636. # casting ...
  2637. value = self.tables[table][name]['ctype'] (value.strip())
  2638. self.tables[table][name]['values'].append(value)
  2639. except:
  2640. read = False
  2641. if "id:" in litem: # get line id
  2642. line = int(item.split(':')[1].strip())
  2643. elif "key column:" in litem: # start reading attributes
  2644. read = True
  2645. nselected = nselected + 1
  2646. elif "layer:" in litem: # get layer id
  2647. layer = int(item.split(':')[1].strip())
  2648. table = self.layers[layer]["table"] # get table desc
  2649. read = False
  2650. return (line, nselected)
  2651. def SelectFromTable(self, layer, cols='*', where=None):
  2652. """Select records from the table
  2653. Return number of selected records, -1 on error
  2654. """
  2655. if layer <= 0:
  2656. return -1
  2657. nselected = 0
  2658. table = self.layers[layer]["table"] # get table desc
  2659. # select values (only one record)
  2660. if where is None or where is '':
  2661. sql="SELECT %s FROM %s" % (cols, table)
  2662. else:
  2663. sql="SELECT %s FROM %s WHERE %s" % (cols, table, where)
  2664. selectCommand = gcmd.Command(["db.select", "-v", "--q",
  2665. "sql=%s" % sql,
  2666. "database=%s" % self.layers[layer]["database"],
  2667. "driver=%s" % self.layers[layer]["driver"]])
  2668. # self.tables[table][key][1] = str(cat)
  2669. if selectCommand.returncode == 0:
  2670. for line in selectCommand.ReadStdOutput():
  2671. name, value = line.split('|')
  2672. # casting ...
  2673. if value:
  2674. value = self.tables[table][name]['ctype'] (value)
  2675. else:
  2676. value = None
  2677. self.tables[table][name]['values'].append(value)
  2678. nselected = 1
  2679. return nselected
  2680. class ModifyTableRecord(wx.Dialog):
  2681. """Dialog for inserting/updating table record"""
  2682. def __init__(self, parent, id, title, data, keyEditable=(-1, True),
  2683. style=wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER):
  2684. """
  2685. Notes:
  2686. 'Data' is a list: [(column, value)]
  2687. 'KeyEditable' (id, editable?) indicates if textarea for key column
  2688. is editable(True) or not.
  2689. """
  2690. wx.Dialog.__init__(self, parent, id, title, style=style)
  2691. self.keyId = keyEditable[0]
  2692. self.panel = wx.Panel(parent=self, id=wx.ID_ANY)
  2693. self.dataPanel = scrolled.ScrolledPanel(parent=self.panel, id=wx.ID_ANY,
  2694. style=wx.TAB_TRAVERSAL)
  2695. self.dataPanel.SetupScrolling(scroll_x=False)
  2696. #
  2697. # buttons
  2698. #
  2699. self.btnCancel = wx.Button(self.panel, wx.ID_CANCEL)
  2700. self.btnSubmit = wx.Button(self.panel, wx.ID_OK, _("Submit"))
  2701. self.btnSubmit.SetDefault()
  2702. #
  2703. # data area
  2704. #
  2705. self.widgets = []
  2706. id = 0
  2707. self.box = False
  2708. self.cat = None
  2709. for column, value in data:
  2710. if keyEditable[0] == id:
  2711. self.cat = int(value)
  2712. if keyEditable[1] == False:
  2713. self.box = True
  2714. id += 1
  2715. continue
  2716. else:
  2717. valueWin = wx.SpinCtrl(parent=self.dataPanel, id=wx.ID_ANY,
  2718. value=value, min=-1e9, max=1e9, size=(250, -1))
  2719. else:
  2720. valueWin = wx.TextCtrl(parent=self.dataPanel, id=wx.ID_ANY,
  2721. value=value, size=(250, -1))
  2722. label = wx.StaticText(parent=self.dataPanel, id=wx.ID_ANY,
  2723. label=column + ":")
  2724. self.widgets.append((label.GetId(),
  2725. valueWin.GetId()))
  2726. id += 1
  2727. self.__Layout()
  2728. # winSize = self.GetSize()
  2729. # fix height of window frame if needed
  2730. # if winSize[1] > 480:
  2731. # winSize[1] = 480
  2732. # self.SetSize(winSize)
  2733. # self.SetMinSize(winSize)
  2734. def __Layout(self):
  2735. """Do layout"""
  2736. sizer = wx.BoxSizer(wx.VERTICAL)
  2737. if self.box:
  2738. box = wx.StaticBox(parent=self.panel, id=wx.ID_ANY,
  2739. label=" %s %d " % (_("Category"), self.cat))
  2740. boxSizer = wx.StaticBoxSizer(box, wx.VERTICAL)
  2741. # data area
  2742. dataSizer = wx.FlexGridSizer (cols=2, hgap=3, vgap=3)
  2743. dataSizer.AddGrowableCol(1)
  2744. for labelId, valueId in self.widgets:
  2745. label = self.FindWindowById(labelId)
  2746. value = self.FindWindowById(valueId)
  2747. dataSizer.Add(label, proportion=0,
  2748. flag=wx.ALIGN_CENTER_VERTICAL)
  2749. dataSizer.Add(value, proportion=0,
  2750. flag=wx.EXPAND | wx.ALIGN_CENTER_VERTICAL)
  2751. self.dataPanel.SetAutoLayout(True)
  2752. self.dataPanel.SetSizer(dataSizer)
  2753. dataSizer.Fit(self.dataPanel)
  2754. if self.box:
  2755. boxSizer.Add(item=self.dataPanel, proportion=1,
  2756. flag=wx.EXPAND | wx.ALL, border=5)
  2757. # buttons
  2758. btnSizer = wx.StdDialogButtonSizer()
  2759. btnSizer.AddButton(self.btnCancel)
  2760. btnSizer.AddButton(self.btnSubmit)
  2761. btnSizer.Realize()
  2762. if not self.box:
  2763. sizer.Add(item=self.dataPanel, proportion=1,
  2764. flag=wx.EXPAND | wx.ALL, border=5)
  2765. else:
  2766. sizer.Add(item=boxSizer, proportion=1,
  2767. flag=wx.EXPAND | wx.ALL, border=5)
  2768. sizer.Add(item=btnSizer, proportion=0,
  2769. flag=wx.EXPAND | wx.ALL, border=5)
  2770. #sizer.SetSizeHints(self.panel)
  2771. self.panel.SetAutoLayout(True)
  2772. self.panel.SetSizer(sizer)
  2773. sizer.Fit(self.panel)
  2774. self.Layout()
  2775. # set window frame size (min & max)
  2776. minFrameHeight = 150
  2777. maxFrameHeight = 2 * minFrameHeight
  2778. if self.GetSize()[1] > minFrameHeight:
  2779. self.SetMinSize((self.GetSize()[0], minFrameHeight))
  2780. else:
  2781. self.SetMinSize(self.GetSize())
  2782. if self.GetSize()[1] > maxFrameHeight:
  2783. self.SetSize((self.GetSize()[0], maxFrameHeight))
  2784. else:
  2785. self.SetSize(self.panel.GetSize())
  2786. def GetValues(self, columns=None):
  2787. """Return list of values (casted to string).
  2788. If columns is given (list), return only values of given columns.
  2789. """
  2790. valueList = []
  2791. for labelId, valueId in self.widgets:
  2792. column = self.FindWindowById(labelId).GetLabel().replace(':', '')
  2793. if columns is None or column in columns:
  2794. value = str(self.FindWindowById(valueId).GetValue()) # -> string
  2795. valueList.append(value)
  2796. # add key value
  2797. if self.box:
  2798. valueList.insert(self.keyId, str(self.cat))
  2799. return valueList
  2800. def main(argv=None):
  2801. if argv is None:
  2802. argv = sys.argv
  2803. if len(argv) != 2:
  2804. print >> sys.stderr, __doc__
  2805. sys.exit()
  2806. # Command line arguments of the script to be run are preserved by the
  2807. # hotswap.py wrapper but hotswap.py and its options are removed that
  2808. # sys.argv looks as if no wrapper was present.
  2809. #print "argv:", `argv`
  2810. #some applications might require image handlers
  2811. wx.InitAllImageHandlers()
  2812. app = wx.PySimpleApp()
  2813. f = AttributeManager(parent=None, id=wx.ID_ANY,
  2814. title="%s - <%s>" % (_("GRASS GIS Attribute Table Manager"),
  2815. argv[1]),
  2816. size=(900,600), vectmap=argv[1])
  2817. f.Show()
  2818. app.MainLoop()
  2819. if __name__ == '__main__':
  2820. main()