base.py 133 KB

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