base.py 139 KB

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