base.py 134 KB

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