dbm.py 134 KB

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