base.py 133 KB

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