123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180318131823183318431853186318731883189319031913192319331943195319631973198319932003201320232033204320532063207320832093210321132123213321432153216321732183219322032213222322332243225322632273228322932303231323232333234323532363237323832393240324132423243324432453246324732483249325032513252325332543255325632573258325932603261326232633264326532663267326832693270327132723273327432753276327732783279328032813282328332843285328632873288328932903291329232933294329532963297329832993300330133023303330433053306330733083309331033113312331333143315331633173318331933203321332233233324332533263327332833293330333133323333333433353336333733383339334033413342334333443345334633473348334933503351335233533354335533563357335833593360336133623363336433653366336733683369337033713372337333743375337633773378337933803381338233833384338533863387338833893390339133923393339433953396 |
- """
- @package dbm.py
- @brief GRASS Attribute Table Manager
- This program is based on FileHunter, published in 'The wxPython Linux
- Tutorial' on wxPython WIKI pages.
- It also uses some functions at
- http://aspn.activestate.com/ASPN/Cookbook/Python/Recipe/426407
- @code
- python dbm.py vector@mapset
- @endcode
- List of classes:
- - Log
- - VirtualAttributeList
- - AttributeManager
- - DisplayAttributesDialog
- - VectorDBInfo
- - ModifyTableRecord
- (C) 2007-2008 by the GRASS Development Team
- This program is free software under the GNU General Public
- License (>=v2). Read the file COPYING that comes with GRASS
- for details.
- @author Jachym Cepicky <jachym.cepicky gmail.com>
- @author Martin Landa <landa.martin gmail.com>
- """
- import sys
- import os
- import locale
- import tempfile
- import copy
- ### i18N
- import gettext
- gettext.install('grasswxpy', os.path.join(os.getenv("GISBASE"), 'locale'), unicode=True)
- import globalvar
- globalvar.CheckForWx()
- import wx
- import wx.lib.mixins.listctrl as listmix
- import wx.lib.flatnotebook as FN
- import wx.lib.scrolledpanel as scrolled
- import sqlbuilder
- import grassenv
- import gcmd
- import utils
- import gdialogs
- import gselect
- from debug import Debug as Debug
- from preferences import globalSettings as UserSettings
- class Log:
- """
- The log output is redirected to the status bar of the containing frame.
- """
- def __init__(self, parent):
- self.parent = parent
- def write(self, text_string):
- """Update status bar"""
- self.parent.SetStatusText(text_string.strip())
- class VirtualAttributeList(wx.ListCtrl,
- listmix.ListCtrlAutoWidthMixin,
- listmix.ColumnSorterMixin):
- """
- Support virtual list class
- """
- def __init__(self, parent, log, mapDBInfo, layer):
- wx.ListCtrl.__init__(self, parent=parent, id=wx.ID_ANY,
- style=wx.LC_REPORT | wx.LC_HRULES |
- wx.LC_VRULES | wx.LC_VIRTUAL | wx.LC_SORT_ASCENDING)
- #
- # initialize variables
- #
- self.log = log
- self.mapDBInfo = mapDBInfo
- self.layer = layer
- self.columns = {} # <- LoadData()
-
- # self.selectedCats = []
- # self.lastTurnSelectedCats = [] # just temporary, for comparation
- #
- # add some attributes (colourful background for each item rows)
- #
- self.attr1 = wx.ListItemAttr()
- # self.attr1.SetBackgroundColour("light blue")
- self.attr1.SetBackgroundColour(wx.Colour(238,238,238))
- self.attr2 = wx.ListItemAttr()
- self.attr2.SetBackgroundColour("white")
- self.il = wx.ImageList(16, 16)
- self.sm_up = self.il.Add(wx.ArtProvider_GetBitmap(wx.ART_GO_UP, wx.ART_TOOLBAR,
- (16,16)))
- self.sm_dn = self.il.Add(wx.ArtProvider_GetBitmap(wx.ART_GO_DOWN, wx.ART_TOOLBAR,
- (16,16)))
- self.SetImageList(self.il, wx.IMAGE_LIST_SMALL)
- keyColumn = self.LoadData(layer)
- # setup mixins
- listmix.ListCtrlAutoWidthMixin.__init__(self)
- listmix.ColumnSorterMixin.__init__(self, len(self.columns))
- # sort item by category (id)
- if keyColumn > -1:
- self.SortListItems(col=keyColumn, ascending=True)
- else:
- self.SortListItems(col=0, ascending=True)
- # events
- self.Bind(wx.EVT_LIST_ITEM_SELECTED, self.OnItemSelected)
- self.Bind(wx.EVT_LIST_ITEM_DESELECTED, self.OnItemDeselected)
- self.Bind(wx.EVT_LIST_COL_CLICK, self.OnColumnClick) # sorting
- # self.Bind(wx.EVT_LIST_DELETE_ITEM, self.OnItemDelete, self.list)
- # self.Bind(wx.EVT_LIST_COL_RIGHT_CLICK, self.OnColRightClick, self.list)
- # self.Bind(wx.EVT_LIST_COL_BEGIN_DRAG, self.OnColBeginDrag, self.list)
- # self.Bind(wx.EVT_LIST_COL_DRAGGING, self.OnColDragging, self.list)
- # self.Bind(wx.EVT_LIST_COL_END_DRAG, self.OnColEndDrag, self.list)
- # self.Bind(wx.EVT_LIST_BEGIN_LABEL_EDIT, self.OnBeginEdit, self.list)
- # self.list.Bind(wx.EVT_LEFT_DCLICK, self.OnDoubleClick)
- # self.list.Bind(wx.EVT_RIGHT_DOWN, self.OnRightDown)
- def Update(self, mapDBInfo):
- """Update list according new mapDBInfo description"""
- self.mapDBInfo = mapDBInfo
- self.LoadData(self.layer)
- def LoadData(self, layer, columns=None, where=None):
- """Load data into list
- @param layer layer number
- @param columns list of columns for output
- @param where where statement
-
- @return id of key column
- @return -1 if key column is not displayed
- """
- self.log.write(_("Loading data..."))
-
- tableName = self.mapDBInfo.layers[layer]['table']
- keyColumn = self.mapDBInfo.layers[layer]['key']
- self.columns = self.mapDBInfo.tables[tableName]
-
- cmd = ["v.db.select",
- "-c", "--q",
- "map=%s" % self.mapDBInfo.map,
- "layer=%d" % layer]
-
- if not columns:
- columns = self.mapDBInfo.GetColumns(tableName)
- else:
- all = self.mapDBInfo.GetColumns(tableName)
- cmd.append("columns=%s" % ','.join(columns))
- for col in columns:
- if col not in all:
- wx.MessageBox(parent=self,
- message=_("Column <%(column)s> not found in "
- "in the table <%(table)s>.") % \
- { 'column' : col, 'table' : tableName },
- caption=_("Error"), style=wx.OK | wx.ICON_ERROR | wx.CENTRE)
- return
- if where:
- cmd.append("where=%s" % where)
-
- try:
- # for maps connected via v.external
- keyId = columns.index(keyColumn)
- except:
- keyId = -1
-
- #
- # read data
- #
- # FIXME: Max. number of rows, while the GUI is still usable
- # stdout can be very large, do not use PIPE, redirect to temp file
- # TODO: more effective way should be implemented...
- outFile = tempfile.NamedTemporaryFile(mode='w+b')
-
- selectCommand = gcmd.Command(cmd,
- stdout=outFile)
- # These two should probably be passed to init more cleanly
- # setting the numbers of items = number of elements in the dictionary
- self.itemDataMap = {}
- self.itemIndexMap = []
- self.itemCatsMap = {}
-
- self.DeleteAllItems()
-
- # self.ClearAll()
- for i in range(self.GetColumnCount()):
- self.DeleteColumn(0)
-
- i = 0
- info = wx.ListItem()
- info.m_mask = wx.LIST_MASK_TEXT | wx.LIST_MASK_IMAGE | wx.LIST_MASK_FORMAT
- info.m_image = -1
- info.m_format = 0
- for column in columns:
- info.m_text = column
- self.InsertColumnInfo(i, info)
- i += 1
-
- if i >= 256:
- self.log.write(_("Can display only 256 columns."))
-
- i = 0
- outFile.seek(0)
-
- while True:
- # os.linesep doesn't work here (MSYS)
- record = outFile.readline().replace('\n', '')
- if not record:
- break
-
- self.itemDataMap[i] = []
- j = 0
-
- for value in record.split('|'):
- # casting ...
- try:
- self.itemDataMap[i].append(self.columns[columns[j]]['ctype'] (value))
- except:
- self.itemDataMap[i].append('')
- if keyId > -1 and keyId == j:
- try:
- cat = self.columns[columns[j]]['ctype'] (value)
- except ValueError, e:
- cat = -1
- wx.MessageBox(parent=self,
- message=_("Error loading attribute data. "
- "Record number: %(rec)d. Unable to cast value '%(val)s' in "
- "key column (%(key)s) to integer.\n\n"
- "Details: %(detail)s") % \
- { 'rec' : i + 1, 'val' : value,
- 'key' : keyColumn, 'detail' : e},
- caption=_("Error"),
- style=wx.OK | wx.ICON_ERROR | wx.CENTRE)
- j += 1
- # insert to table
- # index = self.InsertStringItem(index=sys.maxint, label=str(self.itemDataMap[i][0]))
- # for j in range(len(self.itemDataMap[i][1:])):
- # self.SetStringItem(index=index, col=j+1, label=str(self.itemDataMap[i][j+1]))
-
- # self.SetItemData(item=index, data=i)
-
- self.itemIndexMap.append(i)
- if keyId > -1: # load cats only when LoadData() is called first time
- self.itemCatsMap[i] = cat
- i += 1
- if i >= 100000:
- self.log.write(_("Limit 100000 records."))
- break
- self.SetItemCount(i)
-
- i = 0
- for col in columns:
- width = self.columns[col]['length'] * 6 # FIXME
- if width < 60:
- width = 60
- if width > 300:
- width = 300
- self.SetColumnWidth(col=i, width=width)
- i += 1
- self.SendSizeEvent()
- return keyId
-
- def OnItemSelected(self, event):
- """Item selected. Add item to selected cats..."""
- # cat = int(self.GetItemText(event.m_itemIndex))
- # if cat not in self.selectedCats:
- # self.selectedCats.append(cat)
- # self.selectedCats.sort()
-
- event.Skip()
- def OnItemDeselected(self, event):
- """Item deselected. Remove item from selected cats..."""
- # cat = int(self.GetItemText(event.m_itemIndex))
- # if cat in self.selectedCats:
- # self.selectedCats.remove(cat)
- # self.selectedCats.sort()
- event.Skip()
- def GetSelectedItems(self):
- """Return list of selected items (category numbers)"""
- cats = []
- item = self.GetFirstSelected()
- while item != -1:
- cats.append(self.GetItemText(item))
- item = self.GetNextSelected(item)
- return cats
- def GetColumnText(self, index, col):
- """Return column text"""
- item = self.GetItem(index, col)
- return item.GetText()
- def GetListCtrl(self):
- """Returt list"""
- return self
- def OnGetItemText(self, item, col):
- """Get item text"""
- index = self.itemIndexMap[item]
- s = str(self.itemDataMap[index][col])
- return s
- def OnGetItemAttr(self, item):
- """Get item attributes"""
- index = self.itemIndexMap[item]
- if ( index % 2) == 0:
- return self.attr2
- else:
- return self.attr1
- def OnColumnClick(self, event):
- """Column heading clicked -> sorting"""
- self._col = event.GetColumn()
- # remove duplicated arrow symbol from column header
- # FIXME: should be done automatically
- info = wx.ListItem()
- info.m_mask = wx.LIST_MASK_TEXT | wx.LIST_MASK_IMAGE
- info.m_image = -1
- for column in range(self.GetColumnCount()):
- info.m_text = self.GetColumn(column).GetText()
- self.SetColumn(column, info)
- event.Skip()
- def SortItems(self, sorter=cmp):
- """Sort items"""
- items = list(self.itemDataMap.keys())
- items.sort(self.Sorter)
- self.itemIndexMap = items
- # redraw the list
- self.Refresh()
-
- def Sorter(self, key1, key2):
- colName = self.GetColumn(self._col).GetText()
- ascending = self._colSortFlag[self._col]
- try:
- item1 = self.columns[colName]["ctype"](self.itemDataMap[key1][self._col])
- item2 = self.columns[colName]["ctype"](self.itemDataMap[key2][self._col])
- except ValueError:
- item1 = self.itemDataMap[key1][self._col]
- item2 = self.itemDataMap[key2][self._col]
- if type(item1) == type('') or type(item2) == type(''):
- cmpVal = locale.strcoll(str(item1), str(item2))
- else:
- cmpVal = cmp(item1, item2)
- # If the items are equal then pick something else to make the sort value unique
- if cmpVal == 0:
- cmpVal = apply(cmp, self.GetSecondarySortValues(self._col, key1, key2))
- if ascending:
- return cmpVal
- else:
- return -cmpVal
- def GetSortImages(self):
- """Used by the ColumnSorterMixin, see wx/lib/mixins/listctrl.py"""
- return (self.sm_dn, self.sm_up)
- class AttributeManager(wx.Frame):
- """
- GRASS Attribute manager main window
- """
- def __init__(self, parent, id, title, vectmap,
- size = wx.DefaultSize, style = wx.DEFAULT_FRAME_STYLE,
- pointdata=None, log=None):
- self.vectmap = vectmap
- self.pointdata = pointdata
- self.parent = parent # GMFrame
- self.cmdLog = log # self.parent.goutput
-
- wx.Frame.__init__(self, parent, id, title, style=style)
- # icon
- self.SetIcon(wx.Icon(os.path.join(globalvar.ETCICONDIR, 'grass_sql.ico'), wx.BITMAP_TYPE_ICO))
- self.panel = wx.Panel(parent=self, id=wx.ID_ANY)
- try:
- self.map = self.parent.curr_page.maptree.Map
- self.mapdisplay = self.parent.curr_page.maptree.mapdisplay
- except:
- self.map = self.mapdisplay = None
- if pointdata:
- self.icon = pointdata[0]
- self.pointsize = pointdata[1]
- else:
- self.icon = None
- self.pointsize = None
- # status bar log class
- self.log = Log(self) # -> statusbar
- # query map layer (if parent (GMFrame) is given)
- self.qlayer = None
- # -> layers / tables description
- self.mapDBInfo = VectorDBInfo(self.vectmap)
- if len(self.mapDBInfo.layers.keys()) == 0:
- wx.MessageBox(parent=self.parent,
- message=_("Database connection for vector map <%s> "
- "is not defined in DB file. "
- "You can define new connection in "
- "'Manage layers' tab.") % self.vectmap,
- caption=_("Attribute Table Manager"),
- style=wx.OK | wx.ICON_INFORMATION | wx.CENTRE)
- #
- # list of command/SQL statements to be performed
- #
- self.listOfCommands = []
- self.listOfSQLStatements = []
- self.CreateStatusBar(number=1)
- # set up virtual lists (each layer)
- ### {layer: list, widgets...}
- self.layerPage = {}
- # auinotebook (browse, manage, settings)
- #self.notebook = wx.aui.AuiNotebook(parent=self, id=wx.ID_ANY,
- # style=wx.aui.AUI_NB_BOTTOM)
- # really needed (ML)
- # self.notebook.SetFont(wx.Font(10, wx.FONTFAMILY_MODERN, wx.NORMAL, wx.NORMAL, 0, ''))
- self.notebook = FN.FlatNotebook(parent=self.panel, id=wx.ID_ANY,
- style=FN.FNB_BOTTOM |
- FN.FNB_NO_NAV_BUTTONS |
- FN.FNB_FANCY_TABS)
- dbmStyle = globalvar.FNPageStyle
- self.browsePage = FN.FlatNotebook(self.panel, id=wx.ID_ANY,
- style=dbmStyle)
- # self.notebook.AddPage(self.browsePage, caption=_("Browse data"))
- self.notebook.AddPage(self.browsePage, text=_("Browse data")) # FN
- self.browsePage.SetTabAreaColour(globalvar.FNPageColor)
- self.manageTablePage = FN.FlatNotebook(self.panel, id=wx.ID_ANY,
- style=dbmStyle)
- #self.notebook.AddPage(self.manageTablePage, caption=_("Manage tables"))
- self.notebook.AddPage(self.manageTablePage, text=_("Manage tables")) # FN
- self.manageTablePage.SetTabAreaColour(globalvar.FNPageColor)
- self.manageLayerPage = FN.FlatNotebook(self.panel, id=wx.ID_ANY,
- style=dbmStyle)
- #self.notebook.AddPage(self.manageLayerPage, caption=_("Manage layers"))
- self.notebook.AddPage(self.manageLayerPage, text=_("Manage layers")) # FN
- self.manageLayerPage.SetTabAreaColour(globalvar.FNPageColor)
- self.infoCollapseLabelExp = _("Click here to show database connection information")
- self.infoCollapseLabelCol = _("Click here to hide database connection information")
- self.__createBrowsePage()
- self.__createManageTablePage()
- self.__createManageLayerPage()
- self.notebook.SetSelection(0) # select browse tab
- #
- # buttons
- #
- self.btnQuit = wx.Button(parent=self.panel, id=wx.ID_EXIT)
- # events
- self.btnQuit.Bind(wx.EVT_BUTTON, self.OnCloseWindow)
- self.Bind(FN.EVT_FLATNOTEBOOK_PAGE_CHANGED, self.OnLayerPageChanged, self.browsePage)
- self.Bind(FN.EVT_FLATNOTEBOOK_PAGE_CHANGED, self.OnLayerPageChanged, self.manageTablePage)
- # do layout
- self.__layout()
- # self.SetMinSize(self.GetBestSize())
- self.SetSize((680, 550)) # FIXME hard-coded size
- self.SetMinSize(self.GetSize())
- def __createBrowsePage(self, onlyLayer=-1):
- """Create browse tab page"""
- for layer in self.mapDBInfo.layers.keys():
- if onlyLayer > 0 and layer != onlyLayer:
- continue
- panel = wx.Panel(parent=self.browsePage, id=wx.ID_ANY)
- self.layerPage[layer] = {'browsePage': panel.GetId()}
- self.browsePage.AddPage(page=panel, text=" %d / %s %s" % \
- (layer, _("Table"), self.mapDBInfo.layers[layer]['table']))
- pageSizer = wx.BoxSizer(wx.VERTICAL)
- # attribute data
- listBox = wx.StaticBox(parent=panel, id=wx.ID_ANY,
- label=" %s " % _("Attribute data - right-click to edit/manage records"))
- listSizer = wx.StaticBoxSizer(listBox, wx.VERTICAL)
-
- sqlBox = wx.StaticBox(parent=panel, id=wx.ID_ANY,
- label=" %s " % _("SQL Query"))
- sqlSizer = wx.StaticBoxSizer(sqlBox, wx.VERTICAL)
- win = VirtualAttributeList(panel, self.log,
- self.mapDBInfo, layer)
- win.Bind(wx.EVT_LIST_ITEM_ACTIVATED, self.OnDataItemActivated)
- win.Bind(wx.EVT_COMMAND_RIGHT_CLICK, self.OnDataRightUp) #wxMSW
- win.Bind(wx.EVT_RIGHT_UP, self.OnDataRightUp) #wxGTK
- if UserSettings.Get(group='atm', key='leftDbClick', subkey='selection') == 0:
- win.Bind(wx.EVT_LEFT_DCLICK, self.OnDataItemEdit)
- win.Bind(wx.EVT_COMMAND_LEFT_DCLICK, self.OnDataItemEdit)
- else:
- win.Bind(wx.EVT_LEFT_DCLICK, self.OnDataDrawSelected)
- win.Bind(wx.EVT_COMMAND_LEFT_DCLICK, self.OnDataDrawSelected)
-
-
- listSizer.Add(item=win, proportion=1,
- flag=wx.EXPAND | wx.ALL,
- border=3)
- # sql statement box
- btnApply = wx.Button(parent=panel, id=wx.ID_APPLY)
- btnApply.SetToolTipString(_("Apply SELECT statement and reload data records"))
- btnApply.Bind(wx.EVT_BUTTON, self.OnApplySqlStatement)
- btnSqlBuilder = wx.Button(parent=panel, id=wx.ID_ANY, label=_("SQL Builder"))
- btnSqlBuilder.Bind(wx.EVT_BUTTON, self.OnBuilder)
- sqlSimple = wx.RadioButton(parent=panel, id=wx.ID_ANY,
- label=_("Simple"))
- sqlSimple.SetValue(True)
- sqlAdvanced = wx.RadioButton(parent=panel, id=wx.ID_ANY,
- label=_("Advanced"))
- sqlSimple.Bind(wx.EVT_RADIOBUTTON, self.OnChangeSql)
- sqlAdvanced.Bind(wx.EVT_RADIOBUTTON, self.OnChangeSql)
- sqlWhere = wx.TextCtrl(parent=panel, id=wx.ID_ANY, value="",
- style=wx.TE_PROCESS_ENTER)
- sqlStatement = wx.TextCtrl(parent=panel, id=wx.ID_ANY,
- value="SELECT * FROM %s" % \
- self.mapDBInfo.layers[layer]['table'],
- style=wx.TE_PROCESS_ENTER)
- sqlWhere.Bind(wx.EVT_TEXT_ENTER, self.OnApplySqlStatement)
- sqlStatement.Bind(wx.EVT_TEXT_ENTER, self.OnApplySqlStatement)
- sqlLabel = wx.StaticText(parent=panel, id=wx.ID_ANY,
- label="SELECT * FROM %s WHERE " % \
- self.mapDBInfo.layers[layer]['table'])
- label_query = wx.StaticText(parent=panel, id=wx.ID_ANY,
- label="")
- sqlFlexSizer = wx.FlexGridSizer (cols=3, hgap=5, vgap=5)
- sqlFlexSizer.AddGrowableCol(1)
- sqlFlexSizer.Add(item=sqlSimple,
- flag=wx.ALIGN_CENTER_VERTICAL)
- sqlSimpleSizer = wx.BoxSizer(wx.HORIZONTAL)
- sqlSimpleSizer.Add(item=sqlLabel,
- flag=wx.ALIGN_CENTER_VERTICAL)
- sqlSimpleSizer.Add(item=sqlWhere, proportion=1,
- flag=wx.EXPAND | wx.ALIGN_CENTER_VERTICAL)
- sqlFlexSizer.Add(item=sqlSimpleSizer,
- flag=wx.ALIGN_CENTER_VERTICAL | wx.EXPAND)
- sqlFlexSizer.Add(item=btnApply,
- flag=wx.ALIGN_RIGHT)
- sqlFlexSizer.Add(item=sqlAdvanced,
- flag=wx.ALIGN_CENTER_VERTICAL)
- sqlFlexSizer.Add(item=sqlStatement,
- flag=wx.EXPAND)
- sqlFlexSizer.Add(item=btnSqlBuilder,
- flag=wx.ALIGN_RIGHT)
- sqlSizer.Add(item=sqlFlexSizer,
- flag=wx.ALL | wx.EXPAND,
- border=3)
- pageSizer.Add(item=listSizer,
- proportion=1,
- flag=wx.ALL | wx.EXPAND,
- border=5)
- pageSizer.Add(item=sqlSizer,
- proportion=0,
- flag=wx.BOTTOM | wx.LEFT | wx.RIGHT | wx.EXPAND,
- border=5)
- panel.SetSizer(pageSizer)
- self.layerPage[layer]['data'] = win.GetId()
- self.layerPage[layer]['simple'] = sqlSimple.GetId()
- self.layerPage[layer]['advanced'] = sqlAdvanced.GetId()
- self.layerPage[layer]['where'] = sqlWhere.GetId()
- self.layerPage[layer]['builder'] = btnSqlBuilder.GetId()
- self.layerPage[layer]['statement'] = sqlStatement.GetId()
- self.browsePage.SetSelection(0) # select first layer
- try:
- self.layer = self.mapDBInfo.layers.keys()[0]
- self.OnChangeSql(None)
- self.log.write(_("Number of loaded records: %d") % \
- self.FindWindowById(self.layerPage[self.layer]['data']).GetItemCount())
- except IndexError:
- self.layer = None
-
- def __createManageTablePage(self, onlyLayer=-1):
- """Create manage page (create/link and alter tables)"""
- for layer in self.mapDBInfo.layers.keys():
- if onlyLayer > 0 and layer != onlyLayer:
- continue
-
- panel = wx.Panel(parent=self.manageTablePage, id=wx.ID_ANY)
- self.layerPage[layer]['tablePage'] = panel.GetId()
- self.manageTablePage.AddPage(page=panel, text=" %d / %s %s" % (layer, _("Table"), self.mapDBInfo.layers[layer]['table']))
-
- pageSizer = wx.BoxSizer(wx.VERTICAL)
-
- #
- # dbInfo
- #
- infoCollapse = wx.CollapsiblePane(parent=panel,
- label=self.infoCollapseLabelExp,
- style=wx.CP_DEFAULT_STYLE |
- wx.CP_NO_TLW_RESIZE | wx.EXPAND)
- self.MakeInfoPaneContent(layer, infoCollapse.GetPane())
- infoCollapse.Collapse(False)
- self.Bind(wx.EVT_COLLAPSIBLEPANE_CHANGED, self.OnInfoPaneChanged, infoCollapse)
-
- #
- # table description
- #
- table = self.mapDBInfo.layers[layer]['table']
- tableBox = wx.StaticBox(parent=panel, id=wx.ID_ANY,
- label=" %s " % _("Table <%s> - right-click to delete column(s)") % table)
-
- tableSizer = wx.StaticBoxSizer(tableBox, wx.VERTICAL)
-
- list = self.__createTableDesc(panel, table)
- list.Bind(wx.EVT_COMMAND_RIGHT_CLICK, self.OnTableRightUp) #wxMSW
- list.Bind(wx.EVT_RIGHT_UP, self.OnTableRightUp) #wxGTK
- self.layerPage[layer]['tableData'] = list.GetId()
-
- #
- # add column
- #
- columnBox = wx.StaticBox(parent=panel, id=wx.ID_ANY,
- label=" %s " % _("Manage columns"))
-
- columnSizer = wx.StaticBoxSizer(columnBox, wx.VERTICAL)
-
- addSizer = wx.FlexGridSizer (cols=5, hgap=3, vgap=3)
- addSizer.AddGrowableCol(3)
-
- label = wx.StaticText(parent=panel, id=wx.ID_ANY, label=_("Column name"))
- column = wx.TextCtrl(parent=panel, id=wx.ID_ANY, value='',
- size=(150, -1), style=wx.TE_PROCESS_ENTER)
- column.Bind(wx.EVT_TEXT, self.OnTableAddColumnName)
- column.Bind(wx.EVT_TEXT_ENTER, self.OnTableItemAdd)
- self.layerPage[layer]['addColName'] = column.GetId()
- addSizer.Add(item=label,
- flag=wx.ALIGN_CENTER_VERTICAL)
- addSizer.Add(item=column,
- flag=wx.ALIGN_CENTER_VERTICAL)
- # data type
- label = wx.StaticText(parent=panel, id=wx.ID_ANY, label=_("Data type"))
- addSizer.Add(item=label,
- flag=wx.ALIGN_CENTER_VERTICAL)
-
- subSizer = wx.BoxSizer(wx.HORIZONTAL)
- type = wx.Choice (parent=panel, id=wx.ID_ANY,
- choices = ["integer",
- "double",
- "varchar",
- "date"]) # FIXME
- type.SetSelection(0)
- type.Bind(wx.EVT_CHOICE, self.OnTableChangeType)
- self.layerPage[layer]['addColType'] = type.GetId()
- subSizer.Add(item=type,
- flag=wx.ALIGN_CENTER_VERTICAL | wx.LEFT | wx.RIGHT,
- border=3)
- # length
- label = wx.StaticText(parent=panel, id=wx.ID_ANY, label=_("Data length"))
- length = wx.SpinCtrl(parent=panel, id=wx.ID_ANY, size=(65, -1),
- initial=250,
- min=1, max=1e6)
- length.Enable(False)
- self.layerPage[layer]['addColLength'] = length.GetId()
- subSizer.Add(item=label,
- flag=wx.ALIGN_CENTER_VERTICAL | wx.LEFT | wx.RIGHT,
- border=3)
- subSizer.Add(item=length,
- flag=wx.ALIGN_CENTER_VERTICAL | wx.LEFT | wx.RIGHT,
- border=3)
-
- addSizer.Add(item=subSizer,
- flag=wx.ALIGN_CENTER_VERTICAL | wx.LEFT | wx.RIGHT,
- border=3)
-
- btnAddCol = wx.Button(parent=panel, id=wx.ID_ANY, label=_("Add"))
- btnAddCol.Bind(wx.EVT_BUTTON, self.OnTableItemAdd)
- btnAddCol.Enable(False)
- self.layerPage[layer]['addColButton'] = btnAddCol.GetId()
- addSizer.Add(item=btnAddCol,
- proportion=0,
- flag=wx.EXPAND | wx.ALIGN_RIGHT | wx.FIXED_MINSIZE |
- wx.ALIGN_CENTER_VERTICAL )
-
- # rename col
- label = wx.StaticText(parent=panel, id=wx.ID_ANY, label=_("Rename column"))
- column = wx.ComboBox(parent=panel, id=wx.ID_ANY, size=(150, -1),
- style=wx.CB_SIMPLE | wx.CB_READONLY,
- choices=self.mapDBInfo.GetColumns(table))
- column.SetSelection(0)
- self.layerPage[layer]['renameCol'] = column.GetId()
- addSizer.Add(item=label,
- flag=wx.ALIGN_CENTER_VERTICAL)
- addSizer.Add(item=column,
- flag=wx.ALIGN_CENTER_VERTICAL)
- label = wx.StaticText(parent=panel, id=wx.ID_ANY, label=_("To"))
- columnTo = wx.TextCtrl(parent=panel, id=wx.ID_ANY, value='',
- size=(150, -1), style=wx.TE_PROCESS_ENTER)
- columnTo.Bind(wx.EVT_TEXT, self.OnTableRenameColumnName)
- columnTo.Bind(wx.EVT_TEXT_ENTER, self.OnTableItemChange)
- self.layerPage[layer]['renameColTo'] = columnTo.GetId()
- addSizer.Add(item=label,
- flag=wx.ALIGN_CENTER_VERTICAL | wx.ALIGN_CENTER)
- addSizer.Add(item=columnTo,
- flag=wx.ALIGN_CENTER_VERTICAL)
- btnRenameCol = wx.Button(parent=panel, id=wx.ID_ANY, label=_("&Rename"))
- btnRenameCol.Bind(wx.EVT_BUTTON, self.OnTableItemChange)
- btnRenameCol.Enable(False)
- self.layerPage[layer]['renameColButton'] = btnRenameCol.GetId()
-
- addSizer.Add(item=btnRenameCol,
- proportion=0,
- flag=wx.EXPAND | wx.ALIGN_RIGHT | wx.FIXED_MINSIZE |
- wx.ALIGN_CENTER_VERTICAL)
- columnSizer.Add(item=addSizer, proportion=1,
- flag=wx.ALL | wx.EXPAND, border=3)
-
- tableSizer.Add(item=list,
- flag=wx.ALL | wx.EXPAND,
- proportion=1,
- border=3)
-
- pageSizer.Add(item=infoCollapse,
- flag=wx.ALL | wx.EXPAND,
- proportion=0,
- border=3)
-
- pageSizer.Add(item=tableSizer,
- flag=wx.LEFT | wx.RIGHT | wx.BOTTOM | wx.EXPAND,
- proportion=1,
- border=5)
-
- pageSizer.Add(item=columnSizer,
- flag=wx.LEFT | wx.RIGHT | wx.BOTTOM | wx.EXPAND,
- proportion=0,
- border=5)
-
- panel.SetSizer(pageSizer)
-
- self.layerPage[layer]['dbinfo'] = infoCollapse.GetId()
-
- self.manageTablePage.SetSelection(0) # select first layer
- try:
- self.layer = self.mapDBInfo.layers.keys()[0]
- except IndexError:
- self.layer = None
-
- def __createTableDesc(self, parent, table):
- """Create list with table description"""
- list = TableListCtrl(parent=parent, id=wx.ID_ANY,
- table=self.mapDBInfo.tables[table],
- columns=self.mapDBInfo.GetColumns(table))
- list.Populate()
- # sorter
- # itemDataMap = list.Populate()
- # listmix.ColumnSorterMixin.__init__(self, 2)
- return list
- def __createManageLayerPage(self):
- """Create manage page"""
- splitterWin = wx.SplitterWindow(parent=self.manageLayerPage, id=wx.ID_ANY)
- self.manageLayerPage.AddPage(page=splitterWin,
- text=_("Layers of vector map")) # dummy page
-
- #
- # list of layers
- #
- panelList = wx.Panel(parent=splitterWin, id=wx.ID_ANY)
- panelListSizer = wx.BoxSizer(wx.VERTICAL)
- layerBox = wx.StaticBox(parent=panelList, id=wx.ID_ANY,
- label=" %s " % _("List of layers"))
- layerSizer = wx.StaticBoxSizer(layerBox, wx.VERTICAL)
- self.layerList = self.__createLayerDesc(panelList)
- self.layerList.Bind(wx.EVT_COMMAND_RIGHT_CLICK, self.OnLayerRightUp) #wxMSW
- self.layerList.Bind(wx.EVT_RIGHT_UP, self.OnLayerRightUp) #wxGTK
- layerSizer.Add(item=self.layerList,
- flag=wx.ALL | wx.EXPAND,
- proportion=1,
- border=3)
- panelListSizer.Add(item=layerSizer,
- flag=wx.ALL | wx.EXPAND,
- proportion=1,
- border=3)
- panelList.SetSizer(panelListSizer)
- #
- # manage part
- #
- panelManage = wx.Panel(parent=splitterWin, id=wx.ID_ANY)
- manageSizer = wx.BoxSizer(wx.VERTICAL)
- self.manageLayerBook = LayerBook(parent=panelManage, id=wx.ID_ANY,
- parentDialog=self)
- manageSizer.Add(item=self.manageLayerBook,
- proportion=1,
- flag=wx.LEFT | wx.RIGHT | wx.BOTTOM | wx.EXPAND,
- border=5)
- panelManage.SetSizer(manageSizer)
- splitterWin.SplitHorizontally(panelList, panelManage, 100)
- splitterWin.Fit()
- def __createLayerDesc(self, parent):
- """Create list of linked layers"""
- list = LayerListCtrl(parent=parent, id=wx.ID_ANY,
- layers=self.mapDBInfo.layers)
- list.Populate()
- # sorter
- # itemDataMap = list.Populate()
- # listmix.ColumnSorterMixin.__init__(self, 2)
- return list
- def __layout(self):
- """Do layout"""
- # frame body
- mainSizer = wx.BoxSizer(wx.VERTICAL)
- # buttons
- btnSizer = wx.BoxSizer(wx.HORIZONTAL)
- btnSizer.Add(item=self.btnQuit, proportion=1,
- flag=wx.ALL | wx.ALIGN_RIGHT | wx.SHAPED, border=5)
- # btnSizer.Add(item=self.btnApply,
- # flag=wx.ALL, border=5)
- mainSizer.Add(item=self.notebook, proportion=1, flag=wx.EXPAND)
- mainSizer.Add(item=btnSizer, proportion=0, flag=wx.EXPAND | wx.ALL, border=5)
- self.panel.SetAutoLayout(True)
- self.panel.SetSizer(mainSizer)
- mainSizer.Fit(self.panel)
- self.Layout()
-
- def OnDataRightUp(self, event):
- """Table description area, context menu"""
- if not hasattr(self, "popupDataID1"):
- self.popupDataID1 = wx.NewId()
- self.popupDataID2 = wx.NewId()
- self.popupDataID3 = wx.NewId()
- self.popupDataID4 = wx.NewId()
- self.popupDataID5 = wx.NewId()
- self.popupDataID6 = wx.NewId()
- self.popupDataID7 = wx.NewId()
- self.popupDataID8 = wx.NewId()
- self.popupDataID9 = wx.NewId()
- self.popupDataID10 = wx.NewId()
- self.Bind(wx.EVT_MENU, self.OnDataItemEdit, id=self.popupDataID1)
- self.Bind(wx.EVT_MENU, self.OnDataItemAdd, id=self.popupDataID2)
- self.Bind(wx.EVT_MENU, self.OnDataItemDelete, id=self.popupDataID3)
- self.Bind(wx.EVT_MENU, self.OnDataItemDeleteAll, id=self.popupDataID4)
- self.Bind(wx.EVT_MENU, self.OnDataSelectAll, id=self.popupDataID5)
- self.Bind(wx.EVT_MENU, self.OnDataSelectNone, id=self.popupDataID6)
- self.Bind(wx.EVT_MENU, self.OnDataDrawSelected, id=self.popupDataID7)
- self.Bind(wx.EVT_MENU, self.OnDataDrawSelectedZoom, id=self.popupDataID8)
- self.Bind(wx.EVT_MENU, self.OnExtractSelected, id=self.popupDataID9)
- self.Bind(wx.EVT_MENU, self.OnDataReload, id=self.popupDataID10)
- list = self.FindWindowById(self.layerPage[self.layer]['data'])
- # generate popup-menu
- menu = wx.Menu()
- menu.Append(self.popupDataID1, _("Edit selected record"))
- selected = list.GetFirstSelected()
- if selected == -1 or list.GetNextSelected(selected) != -1:
- menu.Enable(self.popupDataID1, False)
- menu.Append(self.popupDataID2, _("Insert new record"))
- menu.Append(self.popupDataID3, _("Delete selected record(s)"))
- menu.Append(self.popupDataID4, _("Delete all records"))
- menu.AppendSeparator()
- menu.Append(self.popupDataID5, _("Select all"))
- menu.Append(self.popupDataID6, _("Deselect all"))
- menu.AppendSeparator()
- menu.Append(self.popupDataID7, _("Highlight selected"))
- menu.Append(self.popupDataID8, _("Highlight selected and zoom"))
- if not self.map or len(list.GetSelectedItems()) == 0:
- menu.Enable(self.popupDataID7, False)
- menu.Enable(self.popupDataID8, False)
- menu.Append(self.popupDataID9, _("Extract selected"))
- if list.GetFirstSelected() == -1:
- menu.Enable(self.popupDataID3, False)
- menu.Enable(self.popupDataID9, False)
- menu.AppendSeparator()
- menu.Append(self.popupDataID10, _("Reload"))
- self.PopupMenu(menu)
- menu.Destroy()
- # update statusbar
- self.log.write(_("Number of loaded records: %d") % \
- list.GetItemCount())
- def OnDataItemDelete(self, event):
- """Delete selected item(s) from the list (layer/category pair)"""
- list = self.FindWindowById(self.layerPage[self.layer]['data'])
- item = list.GetFirstSelected()
- table = self.mapDBInfo.layers[self.layer]["table"]
- key = self.mapDBInfo.layers[self.layer]["key"]
- indeces = []
- # collect SQL statements
- while item != -1:
- index = list.itemIndexMap[item]
- indeces.append(index)
- cat = list.itemCatsMap[index]
-
- self.listOfSQLStatements.append('DELETE FROM %s WHERE %s=%d' % \
- (table, key, cat))
- item = list.GetNextSelected(item)
- if UserSettings.Get(group='atm', key='askOnDeleteRec', subkey='enabled'):
- deleteDialog = wx.MessageBox(parent=self,
- message=_("Selected data records (%d) will permanently deleted "
- "from table. Do you want to delete them?") % \
- (len(self.listOfSQLStatements)),
- caption=_("Delete records"),
- style=wx.YES_NO | wx.CENTRE)
- if deleteDialog != wx.YES:
- self.listOfSQLStatements = []
- return
- # restore maps
- i = 0
- indexTemp = copy.copy(list.itemIndexMap)
- list.itemIndexMap = []
- dataTemp = copy.deepcopy(list.itemDataMap)
- list.itemDataMap = {}
- catsTemp = copy.deepcopy(list.itemCatsMap)
- list.itemCatsMap = {}
- i = 0
- for index in indexTemp:
- if index in indeces:
- continue
- list.itemIndexMap.append(i)
- list.itemDataMap[i] = dataTemp[index]
- list.itemCatsMap[i] = catsTemp[index]
- i += 1
-
- list.SetItemCount(len(list.itemIndexMap))
- # deselect items
- item = list.GetFirstSelected()
- while item != -1:
- list.SetItemState(item, 0, wx.LIST_STATE_SELECTED | wx.LIST_STATE_FOCUSED)
- item = list.GetNextSelected(item)
-
- # submit SQL statements
- self.ApplyCommands()
-
- event.Skip()
- def OnDataItemDeleteAll(self, event):
- """Delete all items from the list"""
- list = self.FindWindowById(self.layerPage[self.layer]['data'])
- if UserSettings.Get(group='atm', key='askOnDeleteRec', subkey='enabled'):
- deleteDialog = wx.MessageBox(parent=self,
- message=_("All data records (%d) will permanently deleted "
- "from table. Do you want to delete them?") % \
- (len(list.itemIndexMap)),
- caption=_("Delete records"),
- style=wx.YES_NO | wx.CENTRE)
- if deleteDialog != wx.YES:
- return
- list.DeleteAllItems()
- list.itemDataMap = {}
- list.itemIndexMap = []
- list.SetItemCount(0)
- table = self.mapDBInfo.layers[self.layer]["table"]
- self.listOfSQLStatements.append('DELETE FROM %s' % table)
- self.ApplyCommands()
-
- event.Skip()
- def _drawSelected(self, zoom):
- """Highlight selected features"""
- if not self.map or not self.mapdisplay:
- return
-
- list = self.FindWindowById(self.layerPage[self.layer]['data'])
- cats = map(int, list.GetSelectedItems())
- digitToolbar = self.mapdisplay.toolbars['vdigit']
- if digitToolbar and \
- digitToolbar.GetLayer().GetName() == self.vectmap:
- self.mapdisplay.digit.driver.SetSelected(cats, cats=True)
- if zoom:
- n, s, w, e = self.mapdisplay.digit.driver.GetRegionSelected()
- self.mapdisplay.Map.GetRegion(n=n, s=s, w=w, e=e,
- update=True)
- else:
- # add map layer with higlighted vector features
- self.AddQueryMapLayer()
- if zoom:
- keyColumn = self.mapDBInfo.layers[self.layer]['key']
- where = ''
- for range in utils.ListOfCatsToRange(cats).split(','):
- if '-' in range:
- min, max = range.split('-')
- where += '%s >= %d and %s <= %d or ' % \
- (keyColumn, int(min),
- keyColumn, int(max))
- else:
- where += '%s = %d or ' % (keyColumn, int(range))
- where = where.rstrip('or ')
-
- cmd = gcmd.Command(["v.db.select",
- "-r", "--q",
- "map=%s" % self.mapDBInfo.map,
- "layer=%d" % self.layer,
- "where=%s" % where])
-
- region = {}
- for line in cmd.ReadStdOutput():
- key, value = line.split('=')
- region[key] = float(value)
-
- self.mapdisplay.Map.GetRegion(n=region['n'], s=region['s'],
- w=region['w'], e=region['e'],
- update=True)
-
- if zoom:
- self.mapdisplay.Map.AdjustRegion() # adjust resolution
- self.mapdisplay.Map.AlignExtentFromDisplay() # adjust extent
- self.mapdisplay.MapWindow.UpdateMap(render=True, renderVector=True)
- else:
- self.mapdisplay.MapWindow.UpdateMap(render=False, renderVector=True)
-
- def OnDataDrawSelected(self, event):
- """Reload table description"""
- self._drawSelected(zoom=False)
- event.Skip()
- def OnDataDrawSelectedZoom(self, event):
- self._drawSelected(zoom=True)
- event.Skip()
-
- def OnDataItemAdd(self, event):
- """Add new record to the attribute table"""
- list = self.FindWindowById(self.layerPage[self.layer]['data'])
- table = self.mapDBInfo.layers[self.layer]['table']
- keyColumn = self.mapDBInfo.layers[self.layer]['key']
- # (column name, value)
- data = []
- # collect names of all visible columns
- columnName = []
- for i in range(list.GetColumnCount()):
- columnName.append(list.GetColumn(i).GetText())
- # maximal category number
- maxCat = max(list.itemCatsMap.values())
- # key column must be always presented
- if keyColumn not in columnName:
- columnName.insert(0, keyColumn) # insert key column on first position
- data.append((keyColumn, str(maxCat + 1)))
- missingKey = True
- else:
- missingKey = False
-
- # add other visible columns
- colIdx = 0
- keyId = -1
- for col in columnName:
- if col == keyColumn: # key
- if missingKey is False:
- data.append((col, str(maxCat + 1)))
- keyId = colIdx
- else:
- data.append((col, ''))
- colIdx += 1
-
- dlg = ModifyTableRecord(parent=self, id=wx.ID_ANY,
- title=_("Insert new record"),
- data=data, keyEditable=(keyId, True))
- if dlg.ShowModal() == wx.ID_OK:
- try: # get category number
- cat = int(dlg.GetValues(columns=[keyColumn])[0])
- except:
- cat = -1
- try:
- if cat in list.itemCatsMap.values():
- raise ValueError(_("Record with category number %d "
- "already exists in the table.") % cat)
- values = dlg.GetValues() # values (need to be casted)
- columnsString = ''
- valuesString = ''
- for i in range(len(values)):
- if len(values[i]) == 0: # NULL
- if columnName[i] == keyColumn:
- raise ValueError(_("Category number (column %s)"
- " is missing.") % keyColumn)
- else:
- continue
- try:
- values[i] = list.columns[columnName[i]]['ctype'] (values[i])
- except:
- raise ValueError(_("Casting value '%(value)s' to %(type)s failed.") %
- {'value' : str(values[i]),
- 'type' : list.columns[columnName[i]]['type']})
- columnsString += '%s,' % columnName[i]
- if list.columns[columnName[i]]['ctype'] == str:
- valuesString += "'%s'," % values[i]
- else:
- valuesString += "%s," % values[i]
- except ValueError, err:
- wx.MessageBox(parent=self,
- message="%s%s%s" % (_("Unable to insert new record."),
- os.linesep, err),
- caption=_("Error"), style=wx.OK | wx.ICON_ERROR | wx.CENTRE)
- return
- # remove category if need
- if missingKey is True:
- del values[0]
-
- # add new item to the list
- index = max(list.itemIndexMap) + 1
- list.itemIndexMap.append(index)
- list.itemDataMap[index] = values
- list.itemCatsMap[index] = cat
- list.SetItemCount(list.GetItemCount() + 1)
- self.listOfSQLStatements.append('INSERT INTO %s (%s) VALUES(%s)' % \
- (table,
- columnsString.strip(','),
- valuesString.strip(',')))
- self.ApplyCommands()
-
- def OnDataItemEdit(self, event):
- """Edit selected record of the attribute table"""
- list = self.FindWindowById(self.layerPage[self.layer]['data'])
- item = list.GetFirstSelected()
- if item == -1:
- return
- table = self.mapDBInfo.layers[self.layer]['table']
- keyColumn = self.mapDBInfo.layers[self.layer]['key']
- cat = list.itemCatsMap[list.itemIndexMap[item]]
- # (column name, value)
- data = []
- # collect names of all visible columns
- columnName = []
- for i in range(list.GetColumnCount()):
- columnName.append(list.GetColumn(i).GetText())
- # key column must be always presented
- if keyColumn not in columnName:
- columnName.insert(0, keyColumn) # insert key column on first position
- data.append((keyColumn, str(cat)))
- keyId = 0
- missingKey = True
- else:
- missingKey = False
-
- # add other visible columns
- for i in range(len(columnName)):
- if columnName[i] == keyColumn: # key
- if missingKey is False:
- data.append((columnName[i], str(cat)))
- keyId = i
- else:
- if missingKey is True:
- value = list.GetItem(item, i-1).GetText()
- else:
- value = list.GetItem(item, i).GetText()
- data.append((columnName[i], value))
- dlg = ModifyTableRecord(parent=self, id=wx.ID_ANY,
- title=_("Update existing record"),
- data=data, keyEditable=(keyId, False))
- if dlg.ShowModal() == wx.ID_OK:
- values = dlg.GetValues() # string
- updateString = ''
- try:
- for i in range(len(values)):
- if i == keyId: # skip key column
- continue
- if str(list.GetItem(item, i).GetText()) != str(values[i]):
- if len(values[i]) > 0:
- try:
- if missingKey is True:
- idx = i - 1
- else:
- idx = i
- list.itemDataMap[item][idx] = \
- list.columns[columnName[i]]['ctype'] (values[i])
- except:
- raise ValueError(_("Casting value '%(value)s' to %(type)s failed.") % \
- {'value' : str(values[i]),
- 'type' : list.columns[columnName[i]]['type']})
- if list.columns[columnName[i]]['ctype'] == str:
- updateString += "%s='%s'," % (columnName[i], values[i])
- else:
- updateString += "%s=%s," % (columnName[i], values[i])
- else: # NULL
- updateString += "%s=NULL," % (columnName[i])
- except ValueError, err:
- wx.MessageBox(parent=self,
- message="%s%s%s" % (_("Unable to update existing record."),
- os.linesep, err),
- caption=_("Error"), style=wx.OK | wx.ICON_ERROR | wx.CENTRE)
- return
- if len(updateString) > 0:
- self.listOfSQLStatements.append('UPDATE %s SET %s WHERE %s=%d' % \
- (table, updateString.strip(','),
- keyColumn, cat))
- self.ApplyCommands()
-
- def OnDataReload(self, event):
- """Reload list of records"""
- self.OnApplySqlStatement(None)
- self.listOfSQLStatements = []
- def OnDataSelectAll(self, event):
- """Select all items"""
- list = self.FindWindowById(self.layerPage[self.layer]['data'])
- item = -1
- while True:
- item = list.GetNextItem(item)
- if item == -1:
- break
- list.SetItemState(item, wx.LIST_STATE_SELECTED, wx.LIST_STATE_SELECTED)
- event.Skip()
- def OnDataSelectNone(self, event):
- """Deselect items"""
- list = self.FindWindowById(self.layerPage[self.layer]['data'])
- item = -1
- while True:
- item = list.GetNextItem(item, wx.LIST_STATE_SELECTED)
- if item == -1:
- break
- list.SetItemState(item, 0, wx.LIST_STATE_SELECTED | wx.LIST_STATE_FOCUSED)
- event.Skip()
- def OnTableChangeType(self, event):
- """Data type for new column changed. Enable or disable
- data length widget"""
- win = self.FindWindowById(self.layerPage[self.layer]['addColLength'])
- if event.GetString() == "varchar":
- win.Enable(True)
- else:
- win.Enable(False)
- def OnTableRenameColumnName(self, event):
- """Editing column name to be added to the table"""
- btn = self.FindWindowById(self.layerPage[self.layer]['renameColButton'])
- col = self.FindWindowById(self.layerPage[self.layer]['renameCol'])
- colTo = self.FindWindowById(self.layerPage[self.layer]['renameColTo'])
- if len(col.GetValue()) > 0 and len(colTo.GetValue()) > 0:
- btn.Enable(True)
- else:
- btn.Enable(False)
- event.Skip()
- def OnTableAddColumnName(self, event):
- """Editing column name to be added to the table"""
- btn = self.FindWindowById(self.layerPage[self.layer]['addColButton'])
- if len(event.GetString()) > 0:
- btn.Enable(True)
- else:
- btn.Enable(False)
- event.Skip()
- def OnTableItemChange(self, event):
- """Rename column in the table"""
- list = self.FindWindowById(self.layerPage[self.layer]['tableData'])
- name = self.FindWindowById(self.layerPage[self.layer]['renameCol']).GetValue()
- nameTo = self.FindWindowById(self.layerPage[self.layer]['renameColTo']).GetValue()
- table = self.mapDBInfo.layers[self.layer]["table"]
- if not name or not nameTo:
- wx.MessageBox(self=self,
- message=_("Unable to rename column. "
- "No column name defined."),
- caption=_("Error"), style=wx.OK | wx.ICON_ERROR | wx.CENTRE)
- return
- else:
- item = list.FindItem(start=-1, str=name)
- if item > -1:
- if list.FindItem(start=-1, str=nameTo) > -1:
- wx.MessageBox(parent=self,
- message=_("Unable to rename column <%(column)s> to "
- "<%(columnTo)s>. Column already exists "
- "in the table <%(table)s>.") % \
- {'column' : name, 'columnTo' : nameTo,
- 'table' : table},
- caption=_("Error"), style=wx.OK | wx.ICON_ERROR | wx.CENTRE)
- return
- else:
- list.SetItemText(item, nameTo)
- self.listOfCommands.append(['v.db.renamecol',
- 'map=%s' % self.vectmap,
- 'layer=%d' % self.layer,
- 'column=%s,%s' % (name, nameTo)])
- else:
- wx.MessageBox(parent=self,
- message=_("Unable to rename column. "
- "Column <%(column)s> doesn't exist in the table <%(table)s>.") %
- {'column' : name, 'table' : table},
- caption=_("Error"), style=wx.OK | wx.ICON_ERROR | wx.CENTRE)
- return
-
- # apply changes
- self.ApplyCommands()
- # update widgets
- self.FindWindowById(self.layerPage[self.layer]['renameCol']).SetItems(self.mapDBInfo.GetColumns(table))
- self.FindWindowById(self.layerPage[self.layer]['renameCol']).SetSelection(0)
- self.FindWindowById(self.layerPage[self.layer]['renameColTo']).SetValue('')
- event.Skip()
- def OnTableRightUp(self, event):
- """Table description area, context menu"""
- if not hasattr(self, "popupTableID"):
- self.popupTableID1 = wx.NewId()
- self.popupTableID2 = wx.NewId()
- self.popupTableID3 = wx.NewId()
- self.Bind(wx.EVT_MENU, self.OnTableItemDelete, id=self.popupTableID1)
- self.Bind(wx.EVT_MENU, self.OnTableItemDeleteAll, id=self.popupTableID2)
- self.Bind(wx.EVT_MENU, self.OnTableReload, id=self.popupTableID3)
- # generate popup-menu
- menu = wx.Menu()
- menu.Append(self.popupTableID1, _("Drop selected column"))
- if self.FindWindowById(self.layerPage[self.layer]['tableData']).GetFirstSelected() == -1:
- menu.Enable(self.popupTableID1, False)
- menu.Append(self.popupTableID2, _("Drop all columns"))
- menu.AppendSeparator()
- menu.Append(self.popupTableID3, _("Reload"))
- self.PopupMenu(menu)
- menu.Destroy()
- def OnTableItemDelete(self, event):
- """Delete selected item(s) from the list"""
- list = self.FindWindowById(self.layerPage[self.layer]['tableData'])
- item = list.GetFirstSelected()
- while item != -1:
- self.listOfCommands.append(['v.db.dropcol',
- 'map=%s' % self.vectmap,
- 'layer=%d' % self.layer,
- 'column=%s' % list.GetItemText(item)])
- list.DeleteItem(item)
- item = list.GetFirstSelected()
- # apply changes
- self.ApplyCommands()
- # update widgets
- table = self.mapDBInfo.layers[self.layer]['table']
- self.FindWindowById(self.layerPage[self.layer]['renameCol']).SetItems(self.mapDBInfo.GetColumns(table))
- self.FindWindowById(self.layerPage[self.layer]['renameCol']).SetSelection(0)
- event.Skip()
- def OnTableItemDeleteAll(self, event):
- """Delete all items from the list"""
- table = self.mapDBInfo.layers[self.layer]['table']
- cols = self.mapDBInfo.GetColumns(table)
- for col in cols:
- self.listOfCommands = [['v.db.dropcol',
- 'map=%s' % self.vectmap,
- 'layer=%d' % self.layer,
- 'column=%s' % col]]
- self.FindWindowById(self.layerPage[self.layer]['tableData']).DeleteAllItems()
- # apply changes
- self.ApplyCommands()
- # update widgets
- table = self.mapDBInfo.layers[self.layer]['table']
- self.FindWindowById(self.layerPage[self.layer]['renameCol']).SetItems(self.mapDBInfo.GetColumns(table))
- self.FindWindowById(self.layerPage[self.layer]['renameCol']).SetSelection(0)
- event.Skip()
- def OnTableReload(self, event=None):
- """Reload table description"""
- self.FindWindowById(self.layerPage[self.layer]['tableData']).Populate(update=True)
- self.listOfCommands = []
- def OnTableItemAdd(self, event):
- """Add new column to the table"""
- table = self.mapDBInfo.layers[self.layer]['table']
- name = self.FindWindowById(self.layerPage[self.layer]['addColName']).GetValue()
- if not name:
- wx.MessageBox(parent=self,
- message=_("Unable to add column to the table. "
- "No column name defined."),
- caption=_("Error"), style=wx.OK | wx.ICON_ERROR | wx.CENTRE)
- return
- type = self.FindWindowById(self.layerPage[self.layer]['addColType']). \
- GetStringSelection()
-
- # cast type if needed
- if type == 'double':
- type = 'double precision'
- if type == 'varchar':
- length = int(self.FindWindowById(self.layerPage[self.layer]['addColLength']). \
- GetValue())
- else:
- length = '' # FIXME
- # add item to the list of table columns
- list = self.FindWindowById(self.layerPage[self.layer]['tableData'])
- # check for duplicate items
- if list.FindItem(start=-1, str=name) > -1:
- wx.MessageBox(parent=self,
- message=_("Column <%(column)s> already exists in table <%(table)s>.") % \
- {'column' : name, 'table' : self.mapDBInfo.layers[self.layer]["table"]},
- caption=_("Error"), style=wx.OK | wx.ICON_ERROR | wx.CENTRE)
- return
- index = list.InsertStringItem(sys.maxint, str(name))
- list.SetStringItem(index, 0, str(name))
- list.SetStringItem(index, 1, str(type))
- list.SetStringItem(index, 2, str(length))
-
- # add v.db.addcol command to the list
- if type == 'varchar':
- type += ' (%d)' % length
- self.listOfCommands.append(['v.db.addcol',
- 'map=%s' % self.vectmap,
- 'layer=%d' % self.layer,
- 'columns=%s %s' % (name, type)])
- # apply changes
- self.ApplyCommands()
-
- # update widgets
- self.FindWindowById(self.layerPage[self.layer]['addColName']).SetValue('')
- self.FindWindowById(self.layerPage[self.layer]['renameCol']).SetItems(self.mapDBInfo.GetColumns(table))
- self.FindWindowById(self.layerPage[self.layer]['renameCol']).SetSelection(0)
-
- event.Skip()
-
- def OnLayerPageChanged(self, event):
- """Layer tab changed"""
- pageNum = event.GetSelection()
- self.layer = self.mapDBInfo.layers.keys()[pageNum]
- try:
- self.OnChangeSql(None)
- # update statusbar
- self.log.write(_("Number of loaded records: %d") % \
- self.FindWindowById(self.layerPage[self.layer]['data']).\
- GetItemCount())
- except:
- pass
- def OnLayerRightUp(self, event):
- """Layer description area, context menu"""
- pass
- def OnChangeSql(self, event):
- """Switch simple/advanced sql statement"""
- if self.FindWindowById(self.layerPage[self.layer]['simple']).GetValue():
- self.FindWindowById(self.layerPage[self.layer]['where']).Enable(True)
- self.FindWindowById(self.layerPage[self.layer]['statement']).Enable(False)
- self.FindWindowById(self.layerPage[self.layer]['builder']).Enable(False)
- else:
- self.FindWindowById(self.layerPage[self.layer]['where']).Enable(False)
- self.FindWindowById(self.layerPage[self.layer]['statement']).Enable(True)
- self.FindWindowById(self.layerPage[self.layer]['builder']).Enable(True)
- def ApplyCommands(self):
- """Apply changes"""
- # perform GRASS commands (e.g. v.db.addcol)
- if len(self.listOfCommands) > 0:
- for cmd in self.listOfCommands:
- Debug.msg(3, 'AttributeManager.ApplyCommands() cmd=\'%s\'' %
- ' '.join(cmd))
- gcmd.Command(cmd)
- self.mapDBInfo = VectorDBInfo(self.vectmap)
- table = self.mapDBInfo.layers[self.layer]['table']
- # update table description
- list = self.FindWindowById(self.layerPage[self.layer]['tableData'])
- list.Update(table=self.mapDBInfo.tables[table],
- columns=self.mapDBInfo.GetColumns(table))
- self.OnTableReload(None)
- # update data list
- list = self.FindWindowById(self.layerPage[self.layer]['data'])
- list.Update(self.mapDBInfo)
- # reset list of commands
- self.listOfCommands = []
- # perform SQL non-select statements (e.g. 'delete from table where cat=1')
- if len(self.listOfSQLStatements) > 0:
- sqlFile = tempfile.NamedTemporaryFile(mode="w")
- for sql in self.listOfSQLStatements:
- sqlFile.file.write(sql + ";\n")
- sqlFile.file.flush()
- driver = self.mapDBInfo.layers[self.layer]["driver"]
- database = self.mapDBInfo.layers[self.layer]["database"]
- cmd = ['db.execute',
- 'input=%s' % sqlFile.name,
- 'driver=%s' % driver,
- 'database=%s' % database]
- Debug.msg(3, 'AttributeManger.ApplyCommands(): %s' %
- ';'.join(["%s" % s for s in self.listOfSQLStatements]))
- gcmd.Command(cmd)
- # reset list of statements
- self.listOfSQLStatements = []
- def OnApplySqlStatement(self, event):
- """Apply simple/advanced sql statement"""
- keyColumn = -1 # index of key column
- listWin = self.FindWindowById(self.layerPage[self.layer]['data'])
- if self.FindWindowById(self.layerPage[self.layer]['simple']).GetValue():
- # simple sql statement
- where = self.FindWindowById(self.layerPage[self.layer]['where']).GetValue().strip()
- try:
- if len(where) > 0:
- keyColumn = listWin.LoadData(self.layer, where=where)
- else:
- keyColumn = listWin.LoadData(self.layer)
- except gcmd.CmdError, e:
- wx.MessageBox(parent=self,
- message=_("Loading attribute data failed.\n\n%s") % e.message,
- caption=_("Error"), style=wx.OK | wx.ICON_ERROR | wx.CENTRE)
- self.FindWindowById(self.layerPage[self.layer]['where']).SetValue('')
- else:
- # advanced sql statement
- win = self.FindWindowById(self.layerPage[self.layer]['statement'])
- try:
- cols, where = self.ValidateSelectStatement(win.GetValue())
- except TypeError:
- wx.MessageBox(parent=self,
- message=_("Loading attribute data failed.\n"
- "Invalid SQL select statement.\n\n%s") % win.GetValue(),
- caption=_("Error"), style=wx.OK | wx.ICON_ERROR | wx.CENTRE)
- win.SetValue("SELECT * FROM %s" % self.mapDBInfo.layers[self.layer]['table'])
- cols = None
- where = None
- print cols, where
- if cols or where:
- try:
- keyColumn = listWin.LoadData(self.layer, columns=cols,
- where=where)
- except gcmd.CmdError, e:
- wx.MessageBox(parent=self,
- message=_("Loading attribute data failed.\n\n%s") % e.message,
- caption=_("Error"), style=wx.OK | wx.ICON_ERROR | wx.CENTRE)
- win.SetValue("SELECT * FROM %s" % self.mapDBInfo.layers[self.layer]['table'])
-
- # sort by key column
- if keyColumn > -1:
- listWin.SortListItems(col=keyColumn, ascending=True)
- else:
- listWin.SortListItems(col=0, ascending=True)
- # update statusbar
- self.log.write(_("Number of loaded records: %d") % \
- self.FindWindowById(self.layerPage[self.layer]['data']).GetItemCount())
- def ValidateSelectStatement(self, statement):
- """Validate SQL select statement
- @return (columns, where)
- @return None on error
- """
- if statement[0:7].lower() != 'select ':
- return None
-
- cols = ''
- index = 7
- for c in statement[index:]:
- if c == ' ':
- break
- cols += c
- index += 1
- if cols == '*':
- cols = None
- else:
- cols = cols.split(',')
-
- tablelen = len(self.mapDBInfo.layers[self.layer]['table'])
-
- if statement[index+1:index+6].lower() != 'from ' or \
- statement[index+6:index+6+tablelen] != '%s' % \
- (self.mapDBInfo.layers[self.layer]['table']):
- return None
-
- if len(statement[index+7+tablelen:]) > 0:
- index = statement.lower().find('where ')
- if index > -1:
- where = statement[index+6:]
- else:
- return None
- else:
- where = None
-
- return (cols, where)
-
- def OnInfoPaneChanged(self, event):
- """Collapse database connection info box"""
- if self.FindWindowById(self.layerPage[self.layer]['dbinfo']).IsExpanded():
- self.FindWindowById(self.layerPage[self.layer]['dbinfo']).SetLabel( \
- self.infoCollapseLabelCol)
- else:
- self.FindWindowById(self.layerPage[self.layer]['dbinfo']).SetLabel( \
- self.infoCollapseLabelExp)
- # redo layout
- self.Layout()
- def MakeInfoPaneContent(self, layer, pane):
- """Create database connection information content"""
- # connection info
- border = wx.BoxSizer(wx.VERTICAL)
- connectionInfoBox = wx.StaticBox(parent=pane, id=wx.ID_ANY,
- label=" %s " % _("Database connection"))
- infoSizer = wx.StaticBoxSizer(connectionInfoBox, wx.VERTICAL)
- infoFlexSizer = wx.FlexGridSizer (cols=2, hgap=1, vgap=1)
- infoFlexSizer.AddGrowableCol(1)
- infoFlexSizer.Add(item=wx.StaticText(parent=pane, id=wx.ID_ANY,
- label="Driver:"))
- infoFlexSizer.Add(item=wx.StaticText(parent=pane, id=wx.ID_ANY,
- label="%s" % \
- self.mapDBInfo.layers[layer]['driver']))
- infoFlexSizer.Add(item=wx.StaticText(parent=pane, id=wx.ID_ANY,
- label="Database:"))
- infoFlexSizer.Add(item=wx.StaticText(parent=pane, id=wx.ID_ANY,
- label="%s" % \
- self.mapDBInfo.layers[layer]['database']))
- infoFlexSizer.Add(item=wx.StaticText(parent=pane, id=wx.ID_ANY,
- label="Table:"))
- infoFlexSizer.Add(item=wx.StaticText(parent=pane, id=wx.ID_ANY,
- label="%s" % \
- self.mapDBInfo.layers[layer]['table']))
- infoFlexSizer.Add(item=wx.StaticText(parent=pane, id=wx.ID_ANY,
- label="Key:"))
- infoFlexSizer.Add(item=wx.StaticText(parent=pane, id=wx.ID_ANY,
- label="%s" % \
- self.mapDBInfo.layers[layer]['key']))
- infoSizer.Add(item=infoFlexSizer,
- proportion=1,
- flag=wx.EXPAND | wx.ALL,
- border=3)
- border.Add(item=infoSizer,
- proportion=1,
- flag=wx.EXPAND | wx.ALL,
- border=3)
- pane.SetSizer(border)
- def OnCloseWindow(self, event):
- """Cancel button pressed"""
- self.Close()
- event.Skip()
- def OnBuilder(self,event):
- """SQL Builder button pressed"""
- self.builder = sqlbuilder.SQLFrame(parent=self, id=wx.ID_ANY,
- title=_("SQL Builder"),
- vectmap=self.vectmap)
- def OnTextEnter(self, event):
- pass
- def OnSQLBuilder(self, event):
- pass
- def OnDataItemActivated(self, event):
- """Item activated, highlight selected item"""
- self.OnDataDrawSelected(event)
- event.Skip()
- def OnExtractSelected(self, event):
- """
- Extract vector objects selected in attribute browse window
- to new vector map
- """
-
- list = self.FindWindowById(self.layerPage[self.layer]['data'])
- # cats = list.selectedCats[:]
- cats = list.GetSelectedItems()
- if len(cats) == 0:
- wx.MessageBox(parent=self,
- message=_('Nothing to extract.'),
- caption=_('Extract selected'), style=wx.CENTRE)
- return
- else:
- # dialog to get file name
- gdialogs.CreateNewVector(parent=self, title=_('Extract selected'),
- log=self.cmdLog,
- cmdDef=(["v.extract",
- "input=%s" % self.vectmap,
- "list=%s" % utils.ListOfCatsToRange(cats)],
- "output"))
-
- def AddQueryMapLayer(self):
- """Redraw a map
- Return True if map has been redrawn, False if no map is given
- """
- list = self.FindWindowById(self.layerPage[self.layer]['data'])
- cats = list.GetSelectedItems() # FIXME: category can be hiden in list
- if self.mapdisplay.Map.GetLayerIndex(self.qlayer) < 0:
- self.qlayer = None
-
- if self.qlayer:
- self.qlayer.SetCmd(self.mapdisplay.AddTmpVectorMapLayer(self.vectmap, cats, addLayer=False))
- else:
- self.qlayer = self.mapdisplay.AddTmpVectorMapLayer(self.vectmap, cats)
-
- def UpdateDialog(self, layer):
- """Updates dialog layout for given layer"""
- #
- # delete page
- #
- if layer in self.mapDBInfo.layers.keys():
- # delete page
- # draging pages disallowed
- # if self.browsePage.GetPageText(page).replace('Layer ', '').strip() == str(layer):
- # self.browsePage.DeletePage(page)
- # break
- self.browsePage.DeletePage(self.mapDBInfo.layers.keys().index(layer))
- self.manageTablePage.DeletePage(self.mapDBInfo.layers.keys().index(layer))
- # set current page selection
- self.notebook.SetSelection(2)
-
- # fetch fresh db info
- self.mapDBInfo = VectorDBInfo(self.vectmap)
- #
- # add new page
- #
- if layer in self.mapDBInfo.layers.keys():
- # 'browse data' page
- self.__createBrowsePage(layer)
- # 'manage tables' page
- self.__createManageTablePage(layer)
- # set current page selection
- self.notebook.SetSelection(2)
-
- #
- # 'manage layers' page
- #
- # update list of layers
- self.layerList.Update(self.mapDBInfo.layers)
- self.layerList.Populate(update=True)
- # update selected widgets
- listOfLayers = map(str, self.mapDBInfo.layers.keys())
- ### delete layer page
- self.manageLayerBook.deleteLayer.SetItems(listOfLayers)
- if len(listOfLayers) > 0:
- self.manageLayerBook.deleteLayer.SetStringSelection(listOfLayers[0])
- tableName = self.mapDBInfo.layers[int(listOfLayers[0])]['table']
- maxLayer = max(self.mapDBInfo.layers.keys())
- else:
- tableName = ''
- maxLayer = 0
- self.manageLayerBook.deleteTable.SetLabel( \
- _('Drop also linked attribute table (%s)') % \
- tableName)
- ### add layer page
- self.manageLayerBook.addLayerWidgets['layer'][1].SetValue(\
- maxLayer+1)
- ### modify layer
- self.manageLayerBook.modifyLayerWidgets['layer'][1].SetItems(listOfLayers)
- self.manageLayerBook.OnChangeLayer(event=None)
-
- # def OnMapClick(self, event):
- # """
- # Gets coordinates from mouse clicking on display window
- # """
- # Debug.msg(3, "VirtualAttributeList.OnMapClick()")
- # # map coordinates
- # x, y = self.mapdisp.MapWindow.Pixel2Cell(event.GetPositionTuple()[:])
- # category = ""
- # for line in os.popen("v.what east_north=%f,%f map=%s" %\
- # (x,y,self.vectmap)).readlines():
- # if "Category:" in line:
- # category = line.strip().split(" ")[1]
- # #print category
- # for idx in range(self.GetItemCount()):
- # item = self.GetItem(idx, 0)
- # if item.GetText() == category:
- # #print idx
- # # self.Select(idx,True)
- # self.EnsureVisible( idx )
- # self.SetItemState(idx, wx.LIST_STATE_SELECTED, wx.LIST_STATE_SELECTED)
- # else:
- # # self.SetItemState(idx, wx.LIST_STATE_DESELECTED, wx.LIST_STATE_DESELECTED)
- # self.Select(idx,False)
- # # try:
- # # os.environ["GRASS_MESSAGE_FORMAT"] = "gui"
- # # cmd = "v.what -a east_north=%d,%d distance=%d map=%@%" % (x,y,100,self.tablename, self.self.mapset)
- # # self.cmd_output.write(cmd+"\n----------\n")
- # # p = Popen(cmd +" --verbose", shell=True, stdin=PIPE, stdout=PIPE, stderr=PIPE, close_fds=True)
- # #
- # # oline = p.stderr.readline()
- # # while oline:
- # # oline = oline.strip()
- # # oline = p.stderr.readline()
- # # if oline.find("GRASS_INFO_MESSAGE")>-1:
- # # self.cmd_output.write(string.split(oline,maxsplit=1)[1]+"\n")
- # # elif oline.find("GRASS_INFO_WARNING")>-1:
- # # self.cmd_output.write("WARNING: "+string.split(oline,maxsplit=1)[1]+"\n")
- # # elif oline.find("GRASS_INFO_ERROR")>-1:
- # # self.cmd_output.write("ERROR: "+string.split(oline,maxsplit=1)[1]+"\n")
- # #
- # # oline = p.stdout.readline()
- # # while oline:
- # # oline = oline.strip()
- # # self.cmd_output.write(oline+"\n")
- # # print oline+"\n"
- # # print >> sys.stderr, oline
- # # oline = p.stdout.readline()
- # #
- # # if p.stdout < 0:
- # # print >> sys.stderr, "Child was terminated by signal", p.stdout
- # # elif p.stdout > 0:
- # # print >> sys.stderr, p.stdout
- # # pass
- # # except OSError, e:
- # # print >> sys.stderr, "Execution failed:", e
- # # try:
- # # os.environ["GRASS_MESSAGE_FORMAT"] = "gui"
- # # cmd = "v.what -a east_north=%d,%d distance=%d map=%@%" % (x,y,100,self.tablename, self.self.mapset)
- # # self.cmd_output.write(cmd+"\n----------\n")
- # # p = Popen(cmd +" --verbose", shell=True, stdin=PIPE, stdout=PIPE, stderr=PIPE, close_fds=True)
- # #
- # # oline = p.stderr.readline()
- # # while oline:
- # # oline = oline.strip()
- # # oline = p.stderr.readline()
- # # if oline.find("GRASS_INFO_MESSAGE")>-1:
- # # self.cmd_output.write(string.split(oline,maxsplit=1)[1]+"\n")
- # # elif oline.find("GRASS_INFO_WARNING")>-1:
- # # self.cmd_output.write("WARNING: "+string.split(oline,maxsplit=1)[1]+"\n")
- # # elif oline.find("GRASS_INFO_ERROR")>-1:
- # # self.cmd_output.write("ERROR: "+string.split(oline,maxsplit=1)[1]+"\n")
- # #
- # # oline = p.stdout.readline()
- # # while oline:
- # # oline = oline.strip()
- # # self.cmd_output.write(oline+"\n")
- # # print oline+"\n"
- # # print >> sys.stderr, oline
- # # oline = p.stdout.readline()
- # #
- # # if p.stdout < 0:
- # # print >> sys.stderr, "Child was terminated by signal", p.stdout
- # # elif p.stdout > 0:
- # # print >> sys.stderr, p.stdout
- # # pass
- # # except OSError, e:
- # # print >> sys.stderr, "Execution failed:", e
- class TableListCtrl(wx.ListCtrl,
- listmix.ListCtrlAutoWidthMixin):
- # listmix.TextEditMixin):
- """Table description list"""
- def __init__(self, parent, id, table, columns, pos=wx.DefaultPosition,
- size=wx.DefaultSize):
- self.parent = parent
- self.table = table
- self.columns = columns
- wx.ListCtrl.__init__(self, parent, id, pos, size,
- style=wx.LC_REPORT | wx.LC_HRULES | wx.LC_VRULES |
- wx.BORDER_NONE)
- listmix.ListCtrlAutoWidthMixin.__init__(self)
- # listmix.TextEditMixin.__init__(self)
- def Update(self, table, columns):
- """Update column description"""
- self.table = table
- self.columns = columns
- def Populate(self, update=False):
- """Populate the list"""
- itemData = {} # requested by sorter
- if not update:
- headings = [_("Column name"), _("Data type"), _("Data length")]
- i = 0
- for h in headings:
- self.InsertColumn(col=i, heading=h)
- self.SetColumnWidth(col=i, width=150)
- i += 1
- else:
- self.DeleteAllItems()
- i = 0
- for column in self.columns:
- index = self.InsertStringItem(sys.maxint, str(column))
- self.SetStringItem(index, 0, str(column))
- self.SetStringItem(index, 1, str(self.table[column]['type']))
- self.SetStringItem(index, 2, str(self.table[column]['length']))
- self.SetItemData(index, i)
- itemData[i] = (str(column),
- str(self.table[column]['type']),
- int(self.table[column]['length']))
- i = i + 1
- self.SendSizeEvent()
-
- return itemData
- class LayerListCtrl(wx.ListCtrl,
- listmix.ListCtrlAutoWidthMixin):
- # listmix.ColumnSorterMixin):
- # listmix.TextEditMixin):
- """Layer description list"""
- def __init__(self, parent, id, layers,
- pos=wx.DefaultPosition,
- size=wx.DefaultSize):
- self.parent = parent
- self.layers = layers
- wx.ListCtrl.__init__(self, parent, id, pos, size,
- style=wx.LC_REPORT | wx.LC_HRULES | wx.LC_VRULES |
- wx.BORDER_NONE)
- listmix.ListCtrlAutoWidthMixin.__init__(self)
- # listmix.TextEditMixin.__init__(self)
- def Update(self, layers):
- """Update description"""
- self.layers = layers
- def Populate(self, update=False):
- """Populate the list"""
- itemData = {} # requested by sorter
- if not update:
- headings = [_("Layer"), _("Driver"), _("Database"), _("Table"), _("Key")]
- i = 0
- for h in headings:
- self.InsertColumn(col=i, heading=h)
- i += 1
- else:
- self.DeleteAllItems()
- i = 0
- for layer in self.layers.keys():
- index = self.InsertStringItem(sys.maxint, str(layer))
- self.SetStringItem(index, 0, str(layer))
- database = str(self.layers[layer]['database'])
- driver = str(self.layers[layer]['driver'])
- table = str(self.layers[layer]['table'])
- key = str(self.layers[layer]['key'])
- self.SetStringItem(index, 1, driver)
- self.SetStringItem(index, 2, database)
- self.SetStringItem(index, 3, table)
- self.SetStringItem(index, 4, key)
- self.SetItemData(index, i)
- itemData[i] = (str(layer),
- driver,
- database,
- table,
- key)
- i += 1
- for i in range(self.GetColumnCount()):
- self.SetColumnWidth(col=i, width=wx.LIST_AUTOSIZE)
- if self.GetColumnWidth(col=i) < 60:
- self.SetColumnWidth(col=i, width=60)
- self.SendSizeEvent()
-
- return itemData
- class LayerBook(wx.Notebook):
- """Manage layers (add, delete, modify)"""
- def __init__(self, parent, id,
- parentDialog,
- style=wx.BK_DEFAULT):
- wx.Notebook.__init__(self, parent, id, style=style)
- self.parent = parent
- self.parentDialog = parentDialog
- self.mapDBInfo = self.parentDialog.mapDBInfo
- #
- # drivers
- #
- cmdDriver = gcmd.Command(['db.drivers',
- '-p',
- '--q'])
- self.listOfDrivers = []
- for drv in cmdDriver.ReadStdOutput():
- self.listOfDrivers.append(drv.strip())
- #
- # get default values
- #
- self.defaultConnect = {}
- cmdConnect = gcmd.Command(['db.connect',
- '-p',
- '--q'])
- for line in cmdConnect.ReadStdOutput():
- item, value = line.split(':')
- self.defaultConnect[item.strip()] = value.strip()
- if len(self.defaultConnect['driver']) == 0 or \
- len(self.defaultConnect['database']) == 0:
- raise gcmd.DBMError(_('Unable to determine default DB connection settings. '
- 'Please define DB connection using db.connect module.'))
-
- self.defaultTables = self.__getTables(self.defaultConnect['driver'],
- self.defaultConnect['database'])
- try:
- self.defaultColumns = self.__getColumns(self.defaultConnect['driver'],
- self.defaultConnect['database'],
- self.defaultTables[0])
- except IndexError:
- self.defaultColumns = []
- self.__createAddPage()
- self.__createDeletePage()
- self.__createModifyPage()
- def __createAddPage(self):
- """Add new layer"""
- self.addPanel = wx.Panel(parent=self, id=wx.ID_ANY)
- self.AddPage(page=self.addPanel, text=_("Add layer"))
- try:
- maxLayer = max(self.mapDBInfo.layers.keys())
- except ValueError:
- maxLayer = 0
- #
- # list of layer widgets (label, value)
- #
- self.addLayerWidgets = {'layer':
- (wx.StaticText(parent=self.addPanel, id=wx.ID_ANY,
- label='%s:' % _("Layer")),
- wx.SpinCtrl(parent=self.addPanel, id=wx.ID_ANY, size=(65, -1),
- initial=maxLayer+1,
- min=1, max=1e6)),
- 'driver':
- (wx.StaticText(parent=self.addPanel, id=wx.ID_ANY,
- label='%s:' % _("Driver")),
- wx.Choice(parent=self.addPanel, id=wx.ID_ANY, size=(200, -1),
- choices=self.listOfDrivers)),
- 'database':
- (wx.StaticText(parent=self.addPanel, id=wx.ID_ANY,
- label='%s:' % _("Database")),
- wx.TextCtrl(parent=self.addPanel, id=wx.ID_ANY,
- value='',
- style=wx.TE_PROCESS_ENTER)),
- 'table':
- (wx.StaticText(parent=self.addPanel, id=wx.ID_ANY,
- label='%s:' % _("Table")),
- wx.Choice(parent=self.addPanel, id=wx.ID_ANY, size=(200, -1),
- choices=self.defaultTables)),
- 'key':
- (wx.StaticText(parent=self.addPanel, id=wx.ID_ANY,
- label='%s:' % _("Key column")),
- wx.Choice(parent=self.addPanel, id=wx.ID_ANY, size=(200, -1),
- choices=self.defaultColumns))}
-
- # set default values for widgets
- self.addLayerWidgets['driver'][1].SetStringSelection(self.defaultConnect['driver'])
- self.addLayerWidgets['database'][1].SetValue(self.defaultConnect['database'])
- self.addLayerWidgets['table'][1].SetSelection(0)
- self.addLayerWidgets['key'][1].SetSelection(0)
- # events
- self.addLayerWidgets['driver'][1].Bind(wx.EVT_CHOICE, self.OnDriverChanged)
- self.addLayerWidgets['database'][1].Bind(wx.EVT_TEXT_ENTER, self.OnDatabaseChanged)
- self.addLayerWidgets['table'][1].Bind(wx.EVT_CHOICE, self.OnTableChanged)
- #
- # list of table widgets
- #
- self.tableWidgets = {'table': (wx.StaticText(parent=self.addPanel, id=wx.ID_ANY,
- label='%s:' % _("Table name")),
- wx.TextCtrl(parent=self.addPanel, id=wx.ID_ANY,
- value='',
- style=wx.TE_PROCESS_ENTER)),
- 'key': (wx.StaticText(parent=self.addPanel, id=wx.ID_ANY,
- label='%s:' % _("Key column")),
- wx.TextCtrl(parent=self.addPanel, id=wx.ID_ANY,
- value='',
- style=wx.TE_PROCESS_ENTER))}
- # events
- self.tableWidgets['table'][1].Bind(wx.EVT_TEXT_ENTER, self.OnCreateTable)
- self.tableWidgets['key'][1].Bind(wx.EVT_TEXT_ENTER, self.OnCreateTable)
- btnTable = wx.Button(self.addPanel, wx.ID_ANY, _("&Create table"),
- size=(125,-1))
- btnTable.Bind(wx.EVT_BUTTON, self.OnCreateTable)
- btnLayer = wx.Button(self.addPanel, wx.ID_ANY, _("&Add layer"),
- size=(125,-1))
- btnLayer.Bind(wx.EVT_BUTTON, self.OnAddLayer)
- btnDefault = wx.Button(self.addPanel, wx.ID_ANY, _("&Set default"),
- size=(125,-1))
- btnDefault.Bind(wx.EVT_BUTTON, self.OnSetDefault)
- #
- # do layout
- #
- pageSizer = wx.BoxSizer(wx.HORIZONTAL)
- #
- # layer description
- #
- layerBox = wx.StaticBox (parent=self.addPanel, id=wx.ID_ANY,
- label=" %s " % (_("Layer description")))
- layerSizer = wx.StaticBoxSizer(layerBox, wx.VERTICAL)
- # data area
- dataSizer = wx.FlexGridSizer(cols=2, hgap=5, vgap=5)
- dataSizer.AddGrowableCol(1)
- for key in ('layer', 'driver', 'database', 'table', 'key'):
- label, value = self.addLayerWidgets[key]
- dataSizer.Add(item=label,
- flag=wx.ALIGN_CENTER_VERTICAL)
- if label.GetLabel() == "Layer:":
- dataSizer.Add(item=value,
- flag=wx.ALIGN_CENTER_VERTICAL | wx.ALIGN_LEFT)
- else:
- dataSizer.Add(item=value,
- flag=wx.ALIGN_CENTER_VERTICAL | wx.EXPAND)
- layerSizer.Add(item=dataSizer,
- proportion=1,
- flag=wx.ALL | wx.EXPAND,
- border=5)
- btnSizer = wx.BoxSizer(wx.HORIZONTAL)
- btnSizer.Add(item=btnDefault,
- proportion=0,
- flag=wx.ALL | wx.ALIGN_LEFT,
- border=5)
- btnSizer.Add(item=(5,5),
- proportion=1,
- flag=wx.ALL | wx.EXPAND,
- border=5)
- btnSizer.Add(item=btnLayer,
- proportion=0,
- flag=wx.ALL | wx.ALIGN_RIGHT,
- border=5)
- layerSizer.Add(item=btnSizer,
- proportion=0,
- flag=wx.ALL | wx.EXPAND,
- border=0)
- #
- # table description
- #
- tableBox = wx.StaticBox (parent=self.addPanel, id=wx.ID_ANY,
- label=" %s " % (_("Table description")))
- tableSizer = wx.StaticBoxSizer(tableBox, wx.VERTICAL)
- # data area
- dataSizer = wx.FlexGridSizer(cols=2, hgap=5, vgap=5)
- dataSizer.AddGrowableCol(1)
- for key in ['table', 'key']:
- label, value = self.tableWidgets[key]
- dataSizer.Add(item=label,
- flag=wx.ALIGN_CENTER_VERTICAL)
- dataSizer.Add(item=value,
- flag=wx.ALIGN_CENTER_VERTICAL | wx.EXPAND)
- tableSizer.Add(item=dataSizer,
- proportion=1,
- flag=wx.ALL | wx.EXPAND,
- border=5)
- tableSizer.Add(item=btnTable,
- proportion=0,
- flag=wx.ALL | wx.ALIGN_BOTTOM | wx.ALIGN_RIGHT,
- border=5)
- pageSizer.Add(item=layerSizer,
- proportion=1,
- flag=wx.ALL | wx.EXPAND,
- border=5)
- pageSizer.Add(item=tableSizer,
- proportion=1,
- flag=wx.ALL | wx.EXPAND,
- border=5)
- self.addPanel.SetSizer(pageSizer)
- def __createDeletePage(self):
- """Delete layer"""
- self.deletePanel = wx.Panel(parent=self, id=wx.ID_ANY)
- self.AddPage(page=self.deletePanel, text=_("Delete selected layer"))
- label = wx.StaticText(parent=self.deletePanel, id=wx.ID_ANY,
- label='%s:' % _("Layer to detele"))
- self.deleteLayer = wx.ComboBox(parent=self.deletePanel, id=wx.ID_ANY, size=(100, -1),
- style=wx.CB_SIMPLE | wx.CB_READONLY,
- choices=map(str, self.mapDBInfo.layers.keys()))
- self.deleteLayer.SetSelection(0)
- self.deleteLayer.Bind(wx.EVT_COMBOBOX, self.OnChangeLayer)
- try:
- tableName = self.mapDBInfo.layers[int(self.deleteLayer.GetStringSelection())]['table']
- except ValueError:
- tableName = ''
-
- self.deleteTable = wx.CheckBox(parent=self.deletePanel, id=wx.ID_ANY,
- label=_('Drop also linked attribute table (%s)') % \
- tableName)
- if tableName == '':
- self.deleteLayer.Enable(False)
- self.deleteTable.Enable(False)
-
- btnDelete = wx.Button(self.deletePanel, wx.ID_DELETE, _("&Delete layer"),
- size=(125,-1))
- btnDelete.Bind(wx.EVT_BUTTON, self.OnDeleteLayer)
- #
- # do layout
- #
- pageSizer = wx.BoxSizer(wx.VERTICAL)
- dataSizer = wx.BoxSizer(wx.VERTICAL)
- flexSizer = wx.FlexGridSizer(cols=2, hgap=5, vgap=5)
- flexSizer.AddGrowableCol(2)
- flexSizer.Add(item=label,
- flag=wx.ALIGN_CENTER_VERTICAL)
- flexSizer.Add(item=self.deleteLayer,
- flag=wx.ALIGN_CENTER_VERTICAL)
- dataSizer.Add(item=flexSizer,
- proportion=0,
- flag=wx.ALL | wx.EXPAND,
- border=5)
- dataSizer.Add(item=self.deleteTable,
- proportion=0,
- flag=wx.ALL | wx.EXPAND,
- border=5)
- pageSizer.Add(item=dataSizer,
- proportion=1,
- flag=wx.ALL | wx.EXPAND,
- border=5)
- pageSizer.Add(item=btnDelete,
- proportion=0,
- flag=wx.ALL | wx.ALIGN_RIGHT,
- border=5)
- self.deletePanel.SetSizer(pageSizer)
- def __createModifyPage(self):
- """Modify layer"""
- self.modifyPanel = wx.Panel(parent=self, id=wx.ID_ANY)
- self.AddPage(page=self.modifyPanel, text=_("Modify selected layer"))
- #
- # list of layer widgets (label, value)
- #
- self.modifyLayerWidgets = {'layer':
- (wx.StaticText(parent=self.modifyPanel, id=wx.ID_ANY,
- label='%s:' % _("Layer")),
- wx.ComboBox(parent=self.modifyPanel, id=wx.ID_ANY,
- size=(100, -1),
- style=wx.CB_SIMPLE | wx.CB_READONLY,
- choices=map(str,
- self.mapDBInfo.layers.keys()))),
- 'driver':
- (wx.StaticText(parent=self.modifyPanel, id=wx.ID_ANY,
- label='%s:' % _("Driver")),
- wx.Choice(parent=self.modifyPanel, id=wx.ID_ANY,
- size=(200, -1),
- choices=self.listOfDrivers)),
- 'database':
- (wx.StaticText(parent=self.modifyPanel, id=wx.ID_ANY,
- label='%s:' % _("Database")),
- wx.TextCtrl(parent=self.modifyPanel, id=wx.ID_ANY,
- value='', size=(350, -1),
- style=wx.TE_PROCESS_ENTER)),
- 'table':
- (wx.StaticText(parent=self.modifyPanel, id=wx.ID_ANY,
- label='%s:' % _("Table")),
- wx.Choice(parent=self.modifyPanel, id=wx.ID_ANY,
- size=(200, -1),
- choices=self.defaultTables)),
- 'key':
- (wx.StaticText(parent=self.modifyPanel, id=wx.ID_ANY,
- label='%s:' % _("Key column")),
- wx.Choice(parent=self.modifyPanel, id=wx.ID_ANY,
- size=(200, -1),
- choices=self.defaultColumns))}
-
- # set default values for widgets
- self.modifyLayerWidgets['layer'][1].SetSelection(0)
- try:
- layer = int(self.modifyLayerWidgets['layer'][1].GetStringSelection())
- except ValueError:
- layer = None
- for label in self.modifyLayerWidgets.keys():
- self.modifyLayerWidgets[label][1].Enable(False)
- if layer:
- driver = self.mapDBInfo.layers[layer]['driver']
- database = self.mapDBInfo.layers[layer]['database']
- table = self.mapDBInfo.layers[layer]['table']
- listOfColumns = self.__getColumns(driver, database, table)
- self.modifyLayerWidgets['driver'][1].SetStringSelection(driver)
- self.modifyLayerWidgets['database'][1].SetValue(database)
- if table in self.modifyLayerWidgets['table'][1].GetItems():
- self.modifyLayerWidgets['table'][1].SetStringSelection(table)
- else:
- if self.defaultConnect['schema'] != '':
- table = self.defaultConnect['schema'] + table # try with default schema
- else:
- table = 'public.' + table # try with 'public' schema
- self.modifyLayerWidgets['table'][1].SetStringSelection(table)
- self.modifyLayerWidgets['key'][1].SetItems(listOfColumns)
- self.modifyLayerWidgets['key'][1].SetSelection(0)
- # events
- self.modifyLayerWidgets['layer'][1].Bind(wx.EVT_COMBOBOX, self.OnChangeLayer)
- # self.modifyLayerWidgets['driver'][1].Bind(wx.EVT_CHOICE, self.OnDriverChanged)
- # self.modifyLayerWidgets['database'][1].Bind(wx.EVT_TEXT_ENTER, self.OnDatabaseChanged)
- # self.modifyLayerWidgets['table'][1].Bind(wx.EVT_CHOICE, self.OnTableChanged)
- btnModify = wx.Button(self.modifyPanel, wx.ID_DELETE, _("&Modify layer"),
- size=(125,-1))
- btnModify.Bind(wx.EVT_BUTTON, self.OnModifyLayer)
- #
- # do layout
- #
- pageSizer = wx.BoxSizer(wx.VERTICAL)
- # data area
- dataSizer = wx.FlexGridSizer(cols=2, hgap=5, vgap=5)
- dataSizer.AddGrowableCol(1)
- for key in ('layer', 'driver', 'database', 'table', 'key'):
- label, value = self.modifyLayerWidgets[key]
- dataSizer.Add(item=label,
- flag=wx.ALIGN_CENTER_VERTICAL)
- if label.GetLabel() == "Layer:":
- dataSizer.Add(item=value,
- flag=wx.ALIGN_CENTER_VERTICAL | wx.ALIGN_LEFT)
- else:
- dataSizer.Add(item=value,
- flag=wx.ALIGN_CENTER_VERTICAL)
- pageSizer.Add(item=dataSizer,
- proportion=1,
- flag=wx.ALL | wx.EXPAND,
- border=5)
- pageSizer.Add(item=btnModify,
- proportion=0,
- flag=wx.ALL | wx.ALIGN_RIGHT,
- border=5)
- self.modifyPanel.SetSizer(pageSizer)
- def __getTables(self, driver, database):
- """Get list of tables for given driver and database"""
- tables = []
- cmdTable = gcmd.Command(['db.tables',
- '-p', '--q',
- 'driver=%s' % driver,
- 'database=%s' % database])
- for table in cmdTable.ReadStdOutput():
- tables.append(table)
- return tables
- def __getColumns(self, driver, database, table):
- """Get list of column of given table"""
- columns = []
- cmdColumn = gcmd.Command(['db.columns',
- '--q',
- 'driver=%s' % driver,
- 'database=%s' % database,
- 'table=%s' % table])
- for column in cmdColumn.ReadStdOutput():
- columns.append(column)
- return columns
- def OnDriverChanged(self, event):
- """Driver selection changed, update list of tables"""
- driver = event.GetString()
- database = self.addLayerWidgets['database'][1].GetValue()
- winTable = self.addLayerWidgets['table'][1]
- winKey = self.addLayerWidgets['key'][1]
- tables = self.__getTables(driver, database)
- winTable.SetItems(tables)
- winTable.SetSelection(0)
- if len(tables) == 0:
- winKey.SetItems([])
- event.Skip()
- def OnDatabaseChanged(self, event):
- """Database selection changed, update list of tables"""
- event.Skip()
- def OnTableChanged(self, event):
- """Table name changed, update list of columns"""
- driver = self.addLayerWidgets['driver'][1].GetStringSelection()
- database = self.addLayerWidgets['database'][1].GetValue()
- table = event.GetString()
- win = self.addLayerWidgets['key'][1]
- cols = self.__getColumns(driver, database, table)
- win.SetItems(cols)
- win.SetSelection(0)
- event.Skip()
- def OnSetDefault(self, event):
- """Set default values"""
- driver = self.addLayerWidgets['driver'][1]
- database = self.addLayerWidgets['database'][1]
- table = self.addLayerWidgets['table'][1]
- key = self.addLayerWidgets['key'][1]
- driver.SetStringSelection(self.defaultConnect['driver'])
- database.SetValue(self.defaultConnect['database'])
- tables = self.__getTables(self.defaultConnect['driver'],
- self.defaultConnect['database'])
- table.SetItems(tables)
- table.SetSelection(0)
- if len(tables) == 0:
- key.SetItems([])
- else:
- cols = self.__getColumns(self.defaultConnect['driver'],
- self.defaultConnect['database'],
- tables[0])
- key.SetItems(cols)
- key.SetSelection(0)
- event.Skip()
- def OnCreateTable(self, event):
- """Create new table (name and key column given)"""
- driver = self.addLayerWidgets['driver'][1].GetStringSelection()
- database = self.addLayerWidgets['database'][1].GetValue()
- table = self.tableWidgets['table'][1].GetValue()
- key = self.tableWidgets['key'][1].GetValue()
-
- if not table or not key:
- wx.MessageBox(parent=self,
- message=_("Unable to create new table. "
- "Table name or key column name is missing."),
- caption=_("Error"), style=wx.OK | wx.ICON_ERROR | wx.CENTRE)
- return
-
- # create table
- sql = 'CREATE TABLE %s (%s INTEGER)' % (table, key)
- gcmd.Command(['db.execute',
- '--q',
- 'driver=%s' % driver,
- 'database=%s' % database],
- stdin=sql)
- # update list of tables
- tableList = self.addLayerWidgets['table'][1]
- tableList.SetItems(self.__getTables(driver, database))
- tableList.SetStringSelection(table)
- # update key column selection
- keyList = self.addLayerWidgets['key'][1]
- keyList.SetItems(self.__getColumns(driver, database, table))
- keyList.SetStringSelection(key)
-
- event.Skip()
- def OnAddLayer(self, event):
- """Add new layer to vector map"""
- layer = int(self.addLayerWidgets['layer'][1].GetValue())
- layerWin = self.addLayerWidgets['layer'][1]
- driver = self.addLayerWidgets['driver'][1].GetStringSelection()
- database = self.addLayerWidgets['database'][1].GetValue()
- table = self.addLayerWidgets['table'][1].GetStringSelection()
- key = self.addLayerWidgets['key'][1].GetStringSelection()
-
- if layer in self.mapDBInfo.layers.keys():
- wx.MessageBox(parent=self,
- message=_("Unable to add new layer to vector map <%(vector)s>. "
- "Layer %(layer)d already exists.") %
- {'vector' : self.mapDBInfo.map, 'layer' : layer},
- caption=_("Error"), style=wx.OK | wx.ICON_ERROR | wx.CENTRE)
- return
- connectCmd = gcmd.Command(['v.db.connect',
- '--q',
- 'map=%s' % self.mapDBInfo.map,
- 'driver=%s' % driver,
- 'database=%s' % database,
- 'table=%s' % table,
- 'key=%s' % key,
- 'layer=%d' % layer])
-
- if connectCmd.returncode == 0:
- # update dialog (only for new layer)
- self.parentDialog.UpdateDialog(layer=layer)
- # update db info
- self.mapDBInfo = self.parentDialog.mapDBInfo
- # increase layer number
- layerWin.SetValue(layer+1)
- if len(self.mapDBInfo.layers.keys()) == 1:
- # first layer add --- enable previously disabled widgets
- self.deleteLayer.Enable()
- self.deleteTable.Enable()
- for label in self.modifyLayerWidgets.keys():
- self.modifyLayerWidgets[label][1].Enable()
-
- def OnDeleteLayer(self, event):
- """Delete layer"""
- try:
- layer = int(self.deleteLayer.GetValue())
- except:
- return
- gcmd.Command(['v.db.connect',
- '-d', '--q',
- 'map=%s' % self.mapDBInfo.map,
- 'layer=%d' % layer])
- # drop also table linked to layer which is deleted
- if self.deleteTable.IsChecked():
- driver = self.addLayerWidgets['driver'][1].GetStringSelection()
- database = self.addLayerWidgets['database'][1].GetValue()
- table = self.mapDBInfo.layers[layer]['table']
- sql = 'DROP TABLE %s' % (table)
- gcmd.Command(['db.execute',
- '--q',
- 'driver=%s' % driver,
- 'database=%s' % database],
- stdin=sql)
- # update dialog
- self.parentDialog.UpdateDialog(layer=layer)
- # update db info
- self.mapDBInfo = self.parentDialog.mapDBInfo
- if len(self.mapDBInfo.layers.keys()) == 0:
- # disable selected widgets
- self.deleteLayer.Enable(False)
- self.deleteTable.Enable(False)
- for label in self.modifyLayerWidgets.keys():
- self.modifyLayerWidgets[label][1].Enable(False)
-
- event.Skip()
- def OnChangeLayer(self, event):
- """Layer number of layer to be deleted is changed"""
- try:
- layer = int(event.GetString())
- except:
- try:
- layer = self.mapDBInfo.layers.keys()[0]
- except:
- return
- if self.GetCurrentPage() == self.modifyPanel:
- driver = self.mapDBInfo.layers[layer]['driver']
- database = self.mapDBInfo.layers[layer]['database']
- table = self.mapDBInfo.layers[layer]['table']
- listOfColumns = self.__getColumns(driver, database, table)
- self.modifyLayerWidgets['driver'][1].SetStringSelection(driver)
- self.modifyLayerWidgets['database'][1].SetValue(database)
- self.modifyLayerWidgets['table'][1].SetStringSelection(table)
- self.modifyLayerWidgets['key'][1].SetItems(listOfColumns)
- self.modifyLayerWidgets['key'][1].SetSelection(0)
- else:
- self.deleteTable.SetLabel(_('Drop also linked attribute table (%s)') % \
- self.mapDBInfo.layers[layer]['table'])
- if event:
- event.Skip()
- def OnModifyLayer(self, event):
- """Modify layer connection settings"""
- layer = int(self.modifyLayerWidgets['layer'][1].GetStringSelection())
- modify = False
- if self.modifyLayerWidgets['driver'][1].GetStringSelection() != \
- self.mapDBInfo.layers[layer]['driver'] or \
- self.modifyLayerWidgets['database'][1].GetStringSelection() != \
- self.mapDBInfo.layers[layer]['database'] or \
- self.modifyLayerWidgets['table'][1].GetStringSelection() != \
- self.mapDBInfo.layers[layer]['table'] or \
- self.modifyLayerWidgets['key'][1].GetStringSelection() != \
- self.mapDBInfo.layers[layer]['key']:
- modify = True
- if modify:
- # delete layer
- gcmd.Command(['v.db.connect',
- '-d', '--q',
- 'map=%s' % self.mapDBInfo.map,
- 'layer=%d' % layer])
- # add modified layer
- gcmd.Command(['v.db.connect',
- '--q',
- 'map=%s' % self.mapDBInfo.map,
- 'driver=%s' % \
- self.modifyLayerWidgets['driver'][1].GetStringSelection(),
- 'database=%s' % \
- self.modifyLayerWidgets['database'][1].GetValue(),
- 'table=%s' % \
- self.modifyLayerWidgets['table'][1].GetStringSelection(),
- 'key=%s' % \
- self.modifyLayerWidgets['key'][1].GetStringSelection(),
- 'layer=%d' % layer])
- # update dialog (only for new layer)
- self.parentDialog.UpdateDialog(layer=layer)
- # update db info
- self.mapDBInfo = self.parentDialog.mapDBInfo
- event.Skip()
- class DisplayAttributesDialog(wx.Dialog):
- """
- Standard dialog used to add/update/display attributes linked
- to the vector map.
- Attribute data can be selected based on layer and category number
- or coordinates.
- @param parent
- @param map vector map
- @param query query coordinates and distance (used for v.edit)
- @param cats {layer: cats}
- @param line feature id (requested for cats)
- @param style
- @param pos
- @param action (add, update, display)
- """
- def __init__(self, parent, map,
- query=None, cats=None, line=None,
- style=wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER,
- pos=wx.DefaultPosition,
- action="add"):
- self.parent = parent # mapdisplay.BufferedWindow
- self.map = map
- self.action = action
- # id of selected line
- self.line = None
- # get layer/table/column information
- self.mapDBInfo = VectorDBInfo(self.map)
- layers = self.mapDBInfo.layers.keys() # get available layers
- # check if db connection / layer exists
- if len(layers) <= 0:
- label = _("Database connection "
- "is not defined in DB file.")
- wx.MessageBox(parent=self.parent,
- message=_("No attribute table linked to "
- "vector map <%(vector)s> found. %(msg)s"
- "\nYou can disable this message from digitization settings. Or "
- "you can create and link attribute table to the vector map "
- "using Attribute Table Manager.") %
- {'vector' : self.map, 'msg' : label},
- caption=_("Message"), style=wx.OK | wx.ICON_EXCLAMATION | wx.CENTRE)
- self.mapDBInfo = None
- wx.Dialog.__init__(self, parent=self.parent, id=wx.ID_ANY,
- title="", style=style, pos=pos)
- # dialog body
- mainSizer = wx.BoxSizer(wx.VERTICAL)
- # notebook
- self.notebook = wx.Notebook(parent=self, id=wx.ID_ANY, style=wx.BK_DEFAULT)
- self.closeDialog = wx.CheckBox(parent=self, id=wx.ID_ANY,
- label=_("Close dialog on submit"))
- self.closeDialog.SetValue(True)
- self.UpdateDialog(query, cats, line)
- # set title
- if self.action == "update":
- self.SetTitle(_("Update attributes"))
- elif self.action == "add":
- self.SetTitle(_("Add attributes"))
- else:
- self.SetTitle(_("Display attributes"))
- # buttons
- btnCancel = wx.Button(self, wx.ID_CANCEL)
- btnReset = wx.Button(self, wx.ID_UNDO, _("&Reload"))
- btnSubmit = wx.Button(self, wx.ID_OK, _("&Submit"))
- btnSizer = wx.StdDialogButtonSizer()
- btnSizer.AddButton(btnCancel)
- btnSizer.AddButton(btnReset)
- btnSizer.SetNegativeButton(btnReset)
- btnSubmit.SetDefault()
- btnSizer.AddButton(btnSubmit)
- btnSizer.Realize()
- mainSizer.Add(item=self.notebook, proportion=1, flag=wx.EXPAND | wx.ALL, border=5)
- mainSizer.Add(item=self.closeDialog, proportion=0, flag=wx.EXPAND | wx.LEFT | wx.RIGHT,
- border=5)
- mainSizer.Add(item=btnSizer, proportion=0,
- flag=wx.EXPAND | wx.ALL | wx.ALIGN_CENTER, border=5)
- # bindigs
- btnReset.Bind(wx.EVT_BUTTON, self.OnReset)
- btnSubmit.Bind(wx.EVT_BUTTON, self.OnSubmit)
- btnCancel.Bind(wx.EVT_BUTTON, self.OnCancel)
- self.SetSizer(mainSizer)
- mainSizer.Fit(self)
- # set min size for dialog
- self.SetMinSize(self.GetBestSize())
- if self.notebook.GetPageCount() == 0:
- Debug.msg(2, "DisplayAttributesDialog(): Nothing found!")
- self.mapDBInfo = None
- def __SelectAttributes(self, layer):
- """Select attributes"""
- pass
- def OnSQLStatement(self, event):
- """Update SQL statement"""
- pass
- def GetSQLString(self, updateValues=False):
- """Create SQL statement string based on self.sqlStatement
- If updateValues is True, update dataFrame according to values
- in textfields.
- """
- sqlCommands = []
- # find updated values for each layer/category
- for layer in self.mapDBInfo.layers.keys(): # for each layer
- table = self.mapDBInfo.layers[layer]["table"]
- key = self.mapDBInfo.layers[layer]["key"]
- columns = self.mapDBInfo.tables[table]
- for idx in range(len(columns[key]['values'])): # for each category
- updatedColumns = []
- updatedValues = []
- for name in columns.keys():
- if name == key:
- cat = columns[name]['values'][idx]
- continue
- type = columns[name]['type']
- value = columns[name]['values'][idx]
- id = columns[name]['ids'][idx]
- try:
- newvalue = self.FindWindowById(id).GetValue()
- except:
- newvalue = self.FindWindowById(id).GetLabel()
- if newvalue == '':
- newvalue = None
-
- if newvalue != value:
- updatedColumns.append(name)
- if newvalue is None:
- updatedValues.append('NULL')
- else:
- if type != 'character':
- updatedValues.append(newvalue)
- else:
- updatedValues.append("'" + newvalue + "'")
- columns[name]['values'][idx] = newvalue
- if self.action != "add" and len(updatedValues) == 0:
- continue
- if self.action == "add":
- sqlString = "INSERT INTO %s (%s," % (table, key)
- else:
- sqlString = "UPDATE %s SET " % table
- for idx in range(len(updatedColumns)):
- name = updatedColumns[idx]
- if self.action == "add":
- sqlString += name + ","
- else:
- sqlString += name + "=" + updatedValues[idx] + ","
- sqlString = sqlString[:-1] # remove last comma
- if self.action == "add":
- sqlString += ") VALUES (%s," % cat
- for value in updatedValues:
- sqlString += str(value) + ","
- sqlString = sqlString[:-1] # remove last comma
- sqlString += ")"
- else:
- sqlString += " WHERE cat=%s" % cat
- sqlCommands.append(sqlString)
- # for each category
- # for each layer END
- Debug.msg(3, "DisplayAttributesDialog.GetSQLString(): %s" % sqlCommands)
- return sqlCommands
- def OnReset(self, event):
- """Reset form"""
- for layer in self.mapDBInfo.layers.keys():
- table = self.mapDBInfo.layers[layer]["table"]
- key = self.mapDBInfo.layers[layer]["key"]
- columns = self.mapDBInfo.tables[table]
- for idx in range(len(columns[key]['values'])):
- for name in columns.keys():
- type = columns[name]['type']
- value = columns[name]['values'][idx]
- id = columns[name]['ids'][idx]
- if name != key:
- self.FindWindowById(id).SetValue(str(value))
- def OnCancel(self, event):
- """Cancel button pressed"""
- self.parent.parent.dialogs['attributes'] = None
- if self.parent.parent.digit:
- self.parent.parent.digit.driver.SetSelected([])
- self.parent.UpdateMap(render=False)
- else:
- self.parent.parent.OnRender(None)
- self.Close()
- def OnSubmit(self, event):
- """Submit records"""
- sqlCommands = self.GetSQLString(updateValues=True)
- if len(sqlCommands) > 0:
- sqlfile = tempfile.NamedTemporaryFile(mode="w")
- for sql in sqlCommands:
- sqlfile.file.write(sql + ";\n")
- sqlfile.file.flush()
- gcmd.Command(cmd=["db.execute",
- "--q",
- "input=%s" % sqlfile.name])
- if self.closeDialog.IsChecked():
- self.OnCancel(event)
- def GetLine(self):
- """Get id of selected vector object or 'None' if nothing selected"""
- return self.line
- def UpdateDialog(self, query=None, cats=None, line=None):
- """Update dialog
- Return True if updated otherwise False
- """
- self.line = line
- if not self.mapDBInfo:
- return False
- self.mapDBInfo.Reset()
- layers = self.mapDBInfo.layers.keys() # get available layers
- # id of selected line
- if query: # select by position
- self.line, nselected = self.mapDBInfo.SelectByPoint(query[0],
- query[1])
- # reset notebook
- self.notebook.DeleteAllPages()
- for layer in layers: # for each layer
- if not query: # select by layer/cat
- if cats.has_key(layer):
- for cat in cats[layer]:
- nselected = self.mapDBInfo.SelectFromTable(layer,
- where="%s=%d" % \
- (self.mapDBInfo.layers[layer]['key'],
- cat))
- else:
- nselected = 0
- # if nselected <= 0 and self.action != "add":
- # continue # nothing selected ...
- if self.action == "add":
- if nselected <= 0:
- if cats.has_key(layer):
- table = self.mapDBInfo.layers[layer]["table"]
- key = self.mapDBInfo.layers[layer]["key"]
- columns = self.mapDBInfo.tables[table]
- for name in columns.keys():
- if name == key:
- for cat in cats[layer]:
- self.mapDBInfo.tables[table][name]['values'].append(cat)
- else:
- self.mapDBInfo.tables[table][name]['values'].append(None)
- else: # change status 'add' -> 'update'
- self.action = "update"
- # use scrolled panel instead (and fix initial max height of the window to 480px)
- ### panel = wx.Panel(parent=self.notebook, id=wx.ID_ANY)
- panel = scrolled.ScrolledPanel(parent=self.notebook, id=wx.ID_ANY,
- size=(-1, 150))
- panel.SetupScrolling(scroll_x=False)
-
- self.notebook.AddPage(page=panel, text=" %s %d " % (_("Layer"), layer))
-
- # notebook body
- border = wx.BoxSizer(wx.VERTICAL)
- table = self.mapDBInfo.layers[layer]["table"]
- key = self.mapDBInfo.layers[layer]["key"]
- columns = self.mapDBInfo.tables[table]
- # value
- if len(columns[key]['values']) == 0: # no cats
- sizer = wx.BoxSizer(wx.VERTICAL)
- txt = wx.StaticText(parent=panel, id=wx.ID_ANY,
- label=_("No database record available."))
- sizer.Add(txt, proportion=1,
- flag=wx.ALIGN_CENTER_VERTICAL | wx.ALIGN_CENTER | wx.EXPAND)
- border.Add(item=sizer, proportion=1,
- flag=wx.ALL | wx.EXPAND | wx.ALIGN_CENTER_VERTICAL | wx.ALIGN_CENTER,
- border=10)
- panel.SetSizer(border)
- continue
- for idx in range(len(columns[key]['values'])):
- flexSizer = wx.FlexGridSizer (cols=4, hgap=3, vgap=3)
- flexSizer.AddGrowableCol(3)
- # columns
- for name in columns.keys():
- vtype = columns[name]['type']
- if columns[name]['values'][idx] is not None:
- value = str(columns[name]['values'][idx])
- else:
- value = ''
- if name == key:
- box = wx.StaticBox (parent=panel, id=wx.ID_ANY,
- label=" %s %s " % (_("Category"), value))
- boxFont = self.GetFont()
- boxFont.SetWeight(wx.FONTWEIGHT_BOLD)
- box.SetFont(boxFont)
- sizer = wx.StaticBoxSizer(box, wx.VERTICAL)
- colValue = box
- else:
- colName = wx.StaticText(parent=panel, id=wx.ID_ANY,
- label=name)
- colType = wx.StaticText(parent=panel, id=wx.ID_ANY,
- label="[" + vtype.lower() + "]")
- delimiter = wx.StaticText(parent=panel, id=wx.ID_ANY, label=":")
- colValue = wx.TextCtrl(parent=panel, id=wx.ID_ANY, value=value) # TODO: validator
- colValue.SetName(name)
- self.Bind(wx.EVT_TEXT, self.OnSQLStatement, colValue)
- flexSizer.Add(colName, proportion=0,
- flag=wx.FIXED_MINSIZE | wx.ALIGN_CENTER_VERTICAL)
- flexSizer.Add(colType, proportion=0,
- flag=wx.FIXED_MINSIZE | wx.ALIGN_CENTER_VERTICAL)
- flexSizer.Add(delimiter, proportion=0,
- flag=wx.FIXED_MINSIZE | wx.ALIGN_CENTER_VERTICAL)
- flexSizer.Add(colValue, proportion=1,
- flag=wx.EXPAND | wx.ALIGN_CENTER_VERTICAL)
- # add widget reference to self.columns
- columns[name]['ids'].append(colValue.GetId()) # name, type, values, id
- # for each attribute (including category) END
- sizer.Add(item=flexSizer, proportion=1, flag=wx.ALL | wx.EXPAND, border=1)
- border.Add(item=sizer, proportion=1, flag=wx.ALL | wx.EXPAND, border=5)
- # for each category END
- panel.SetSizer(border)
- # for each layer END
- self.Layout()
- return True
- class VectorDBInfo(gselect.VectorDBInfo):
- """Class providing information about attribute tables
- linked to the vector map"""
- def __init__(self, map):
- gselect.VectorDBInfo.__init__(self, map)
-
- def GetColumns(self, table):
- """Return list of columns names (based on their index)"""
- names = [''] * len(self.tables[table].keys())
- for name, desc in self.tables[table].iteritems():
- names[desc['index']] = name
- return names
- def SelectByPoint(self, queryCoords, qdist):
- """Get attributes by coordinates (all available layers)
- Return line id or None if no line is found"""
- line = None
- nselected = 0
- cmdWhat = gcmd.Command(cmd=['v.what',
- '-a', '--q',
- 'map=%s' % self.map,
- 'east_north=%f,%f' % \
- (float(queryCoords[0]), float(queryCoords[1])),
- 'distance=%f' % qdist], stderr=None)
- if cmdWhat.returncode == 0:
- read = False
- for item in cmdWhat.ReadStdOutput():
- litem = item.lower()
- if read:
- name, value = item.split(':')
- name = name.strip()
- # append value to the column
- try:
- # casting ...
- value = self.tables[table][name]['ctype'] (value.strip())
- self.tables[table][name]['values'].append(value)
- except:
- read = False
- if "id:" in litem: # get line id
- line = int(item.split(':')[1].strip())
- elif "key column:" in litem: # start reading attributes
- read = True
- nselected = nselected + 1
- elif "layer:" in litem: # get layer id
- layer = int(item.split(':')[1].strip())
- table = self.layers[layer]["table"] # get table desc
- read = False
- return (line, nselected)
- def SelectFromTable(self, layer, cols='*', where=None):
- """Select records from the table
- Return number of selected records, -1 on error
- """
- if layer <= 0:
- return -1
- nselected = 0
- table = self.layers[layer]["table"] # get table desc
- # select values (only one record)
- if where is None or where is '':
- sql="SELECT %s FROM %s" % (cols, table)
- else:
- sql="SELECT %s FROM %s WHERE %s" % (cols, table, where)
- selectCommand = gcmd.Command(["db.select", "-v", "--q",
- "sql=%s" % sql,
- "database=%s" % self.layers[layer]["database"],
- "driver=%s" % self.layers[layer]["driver"]])
- # self.tables[table][key][1] = str(cat)
- if selectCommand.returncode == 0:
- for line in selectCommand.ReadStdOutput():
- name, value = line.split('|')
- # casting ...
- if value:
- value = self.tables[table][name]['ctype'] (value)
- else:
- value = None
- self.tables[table][name]['values'].append(value)
- nselected = 1
- return nselected
- class ModifyTableRecord(wx.Dialog):
- """Dialog for inserting/updating table record"""
- def __init__(self, parent, id, title, data, keyEditable=(-1, True),
- style=wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER):
- """
- Notes:
- 'Data' is a list: [(column, value)]
- 'KeyEditable' (id, editable?) indicates if textarea for key column
- is editable(True) or not.
- """
- wx.Dialog.__init__(self, parent, id, title, style=style)
- self.keyId = keyEditable[0]
-
- self.panel = wx.Panel(parent=self, id=wx.ID_ANY)
- self.dataPanel = scrolled.ScrolledPanel(parent=self.panel, id=wx.ID_ANY,
- style=wx.TAB_TRAVERSAL)
- self.dataPanel.SetupScrolling(scroll_x=False)
-
- #
- # buttons
- #
- self.btnCancel = wx.Button(self.panel, wx.ID_CANCEL)
- self.btnSubmit = wx.Button(self.panel, wx.ID_OK, _("Submit"))
- self.btnSubmit.SetDefault()
- #
- # data area
- #
- self.widgets = []
- id = 0
- self.box = False
- self.cat = None
- for column, value in data:
- if keyEditable[0] == id:
- self.cat = int(value)
- if keyEditable[1] == False:
- self.box = True
- id += 1
- continue
- else:
- valueWin = wx.SpinCtrl(parent=self.dataPanel, id=wx.ID_ANY,
- value=value, min=-1e9, max=1e9, size=(250, -1))
- else:
- valueWin = wx.TextCtrl(parent=self.dataPanel, id=wx.ID_ANY,
- value=value, size=(250, -1))
-
- label = wx.StaticText(parent=self.dataPanel, id=wx.ID_ANY,
- label=column + ":")
- self.widgets.append((label.GetId(),
- valueWin.GetId()))
- id += 1
-
- self.__Layout()
- # winSize = self.GetSize()
- # fix height of window frame if needed
- # if winSize[1] > 480:
- # winSize[1] = 480
- # self.SetSize(winSize)
- # self.SetMinSize(winSize)
- def __Layout(self):
- """Do layout"""
- sizer = wx.BoxSizer(wx.VERTICAL)
- if self.box:
- box = wx.StaticBox(parent=self.panel, id=wx.ID_ANY,
- label=" %s %d " % (_("Category"), self.cat))
- boxSizer = wx.StaticBoxSizer(box, wx.VERTICAL)
- # data area
- dataSizer = wx.FlexGridSizer (cols=2, hgap=3, vgap=3)
- dataSizer.AddGrowableCol(1)
- for labelId, valueId in self.widgets:
- label = self.FindWindowById(labelId)
- value = self.FindWindowById(valueId)
- dataSizer.Add(label, proportion=0,
- flag=wx.ALIGN_CENTER_VERTICAL)
- dataSizer.Add(value, proportion=0,
- flag=wx.EXPAND | wx.ALIGN_CENTER_VERTICAL)
- self.dataPanel.SetAutoLayout(True)
- self.dataPanel.SetSizer(dataSizer)
- dataSizer.Fit(self.dataPanel)
- if self.box:
- boxSizer.Add(item=self.dataPanel, proportion=1,
- flag=wx.EXPAND | wx.ALL, border=5)
-
- # buttons
- btnSizer = wx.StdDialogButtonSizer()
- btnSizer.AddButton(self.btnCancel)
- btnSizer.AddButton(self.btnSubmit)
- btnSizer.Realize()
- if not self.box:
- sizer.Add(item=self.dataPanel, proportion=1,
- flag=wx.EXPAND | wx.ALL, border=5)
- else:
- sizer.Add(item=boxSizer, proportion=1,
- flag=wx.EXPAND | wx.ALL, border=5)
-
- sizer.Add(item=btnSizer, proportion=0,
- flag=wx.EXPAND | wx.ALL, border=5)
- #sizer.SetSizeHints(self.panel)
- self.panel.SetAutoLayout(True)
- self.panel.SetSizer(sizer)
- sizer.Fit(self.panel)
- self.Layout()
- # set window frame size (min & max)
- minFrameHeight = 150
- maxFrameHeight = 2 * minFrameHeight
- if self.GetSize()[1] > minFrameHeight:
- self.SetMinSize((self.GetSize()[0], minFrameHeight))
- else:
- self.SetMinSize(self.GetSize())
- if self.GetSize()[1] > maxFrameHeight:
- self.SetSize((self.GetSize()[0], maxFrameHeight))
- else:
- self.SetSize(self.panel.GetSize())
-
- def GetValues(self, columns=None):
- """Return list of values (casted to string).
- If columns is given (list), return only values of given columns.
- """
- valueList = []
- for labelId, valueId in self.widgets:
- column = self.FindWindowById(labelId).GetLabel().replace(':', '')
- if columns is None or column in columns:
- value = str(self.FindWindowById(valueId).GetValue()) # -> string
- valueList.append(value)
- # add key value
- if self.box:
- valueList.insert(self.keyId, str(self.cat))
-
- return valueList
- def main(argv=None):
- if argv is None:
- argv = sys.argv
- if len(argv) != 2:
- print >> sys.stderr, __doc__
- sys.exit()
- # Command line arguments of the script to be run are preserved by the
- # hotswap.py wrapper but hotswap.py and its options are removed that
- # sys.argv looks as if no wrapper was present.
- #print "argv:", `argv`
- #some applications might require image handlers
- wx.InitAllImageHandlers()
- app = wx.PySimpleApp()
- f = AttributeManager(parent=None, id=wx.ID_ANY,
- title="%s - <%s>" % (_("GRASS GIS Attribute Table Manager"),
- argv[1]),
- size=(900,600), vectmap=argv[1])
- f.Show()
- app.MainLoop()
- if __name__ == '__main__':
- main()
|