manager.py 126 KB

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