base.py 138 KB

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