dbm.py 138 KB

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