12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648264926502651265226532654265526562657265826592660266126622663266426652666266726682669267026712672267326742675267626772678267926802681268226832684268526862687268826892690269126922693269426952696269726982699270027012702270327042705270627072708270927102711271227132714271527162717271827192720272127222723272427252726272727282729273027312732273327342735273627372738273927402741274227432744274527462747274827492750275127522753275427552756275727582759276027612762276327642765276627672768276927702771277227732774277527762777277827792780278127822783278427852786278727882789279027912792279327942795279627972798279928002801280228032804280528062807280828092810281128122813281428152816281728182819282028212822282328242825282628272828282928302831283228332834283528362837283828392840284128422843284428452846284728482849285028512852285328542855285628572858285928602861286228632864286528662867286828692870287128722873287428752876287728782879288028812882288328842885288628872888288928902891289228932894289528962897289828992900290129022903290429052906290729082909291029112912291329142915291629172918291929202921292229232924292529262927292829292930293129322933293429352936293729382939294029412942294329442945294629472948294929502951295229532954295529562957295829592960296129622963296429652966296729682969297029712972297329742975297629772978297929802981298229832984298529862987298829892990299129922993299429952996299729982999300030013002300330043005300630073008300930103011301230133014301530163017301830193020302130223023302430253026302730283029303030313032303330343035303630373038303930403041304230433044304530463047304830493050305130523053305430553056305730583059306030613062306330643065306630673068306930703071307230733074307530763077307830793080308130823083308430853086308730883089309030913092309330943095309630973098309931003101310231033104310531063107310831093110311131123113311431153116311731183119312031213122312331243125312631273128312931303131313231333134313531363137313831393140314131423143314431453146314731483149315031513152315331543155315631573158315931603161316231633164316531663167316831693170317131723173317431753176317731783179318031813182318331843185318631873188318931903191319231933194319531963197319831993200320132023203320432053206320732083209321032113212321332143215321632173218321932203221322232233224322532263227322832293230323132323233323432353236323732383239324032413242324332443245324632473248324932503251325232533254325532563257325832593260326132623263326432653266326732683269327032713272327332743275327632773278327932803281328232833284328532863287328832893290329132923293329432953296329732983299330033013302330333043305330633073308330933103311331233133314331533163317 |
- """!
- @package dbmgr.base
- @brief GRASS Attribute Table Manager base classes
- List of classes:
- - base::Log
- - base::VirtualAttributeList
- - base::DbMgrBase
- - base::DbMgrNotebookBase
- - base::DbMgrBrowsePage
- - base::DbMgrTablesPage
- - base::DbMgrLayersPage
- - base::TableListCtrl
- - base::LayerListCtrl
- - base::LayerBook
- (C) 2007-2009, 2011-2012 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>
- @author Refactoring by Stepan Turek <stepan.turek seznam.cz> (GSoC 2012, mentor: Martin Landa)
- """
- import sys
- import os
- import locale
- import tempfile
- import copy
- import types
- if __name__ == "__main__":
- gui_wx_path = os.path.join(os.getenv('GISBASE'), 'etc', 'gui', 'wxpython')
- if gui_wx_path not in sys.path:
- sys.path.append(gui_wx_path)
- from core import globalvar
- import wx
- import wx.lib.mixins.listctrl as listmix
- import wx.lib.flatnotebook as FN
- import grass.script as grass
- from dbmgr.sqlbuilder import SQLBuilderSelect, SQLBuilderUpdate
- from core.gcmd import RunCommand, GException, GError, GMessage, GWarning
- from core.utils import ListOfCatsToRange, _
- from gui_core.dialogs import CreateNewVector
- from dbmgr.vinfo import VectorDBInfo, GetUnicodeValue, CreateDbInfoDesc
- from core.debug import Debug
- from dbmgr.dialogs import ModifyTableRecord, AddColumnDialog
- from core.settings import UserSettings
- class Log:
- """!The log output SQL 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"""
- if self.parent:
- self.parent.SetStatusText(text_string.strip())
- class VirtualAttributeList(wx.ListCtrl,
- listmix.ListCtrlAutoWidthMixin,
- listmix.ColumnSorterMixin):
- """!Support virtual list class for Attribute Table Manager (browse page)
- """
- def __init__(self, parent, log, dbMgrData, layer, pages):
- # initialize variables
- self.parent = parent
- self.log = log
- self.dbMgrData = dbMgrData
- self.mapDBInfo = self.dbMgrData['mapDBInfo']
- self.layer = layer
- self.pages = pages
- self.fieldCalc = None
- self.columns = {} # <- LoadData()
-
- self.sqlFilter = {}
- 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)
-
- try:
- keyColumn = self.LoadData(layer)
- except GException, e:
- GError(parent = self,
- message = e.value)
- return
-
- # add some attributes (colourful background for each item rows)
- self.attr1 = wx.ListItemAttr()
- 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)
-
- # 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)
- elif keyColumn:
- 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.OnColumnSort)
- self.Bind(wx.EVT_LIST_COL_RIGHT_CLICK, self.OnColumnMenu)
-
- def Update(self, mapDBInfo = None):
- """!Update list according new mapDBInfo description"""
- if mapDBInfo:
- self.mapDBInfo = mapDBInfo
- self.LoadData(self.layer)
- else:
- self.LoadData(self.layer, **self.sqlFilter)
- def LoadData(self, layer, columns = None, where = None, sql = None):
- """!Load data into list
- @param layer layer number
- @param columns list of columns for output (-> v.db.select)
- @param where where statement (-> v.db.select)
- @param sql full sql statement (-> db.select)
-
- @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']
- try:
- self.columns = self.mapDBInfo.tables[tableName]
- except KeyError:
- raise GException(_("Attribute table <%s> not found. "
- "For creating the table switch to "
- "'Manage layers' tab.") % tableName)
-
- if not columns:
- columns = self.mapDBInfo.GetColumns(tableName)
- else:
- all = self.mapDBInfo.GetColumns(tableName)
- for col in columns:
- if col not in all:
- GError(parent = self,
- message = _("Column <%(column)s> not found in "
- "in the table <%(table)s>.") % \
- { 'column' : col, 'table' : tableName })
- return
-
- 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...
- # split on field sep breaks if varchar() column contains the
- # values, so while sticking with ASCII we make it something
- # highly unlikely to exist naturally.
- fs = '{_sep_}'
- outFile = tempfile.NamedTemporaryFile(mode = 'w+b')
- cmdParams = dict(quiet = True,
- parent = self,
- flags = 'c',
- separator = fs)
- if sql:
- cmdParams.update(dict(sql = sql,
- output = outFile.name,
- overwrite = True))
- ret = RunCommand('db.select',
- **cmdParams)
- self.sqlFilter = {"sql" : sql}
- else:
- cmdParams.update(dict(map = self.mapDBInfo.map,
- layer = layer,
- where = where,
- stdout = outFile))
-
- self.sqlFilter = {"where" : where}
- if columns:
- cmdParams.update(dict(columns = ','.join(columns)))
- ret = RunCommand('v.db.select',
- **cmdParams)
-
- # 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
- record = record.split(fs)
- if len(columns) != len(record):
- GError(parent = self,
- message = _("Inconsistent number of columns "
- "in the table <%(table)s>.") % \
- {'table' : tableName })
- self.columns = {} # because of IsEmpty method
- return
- self.AddDataRow(i, record, columns, keyId)
- i += 1
- if i >= 100000:
- self.log.write(_("Viewing 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()
-
- self.log.write(_("Number of loaded records: %d") % \
- self.GetItemCount())
-
- return keyId
-
- def AddDataRow(self, i, record, columns, keyId):
- """!Add row to the data list"""
- self.itemDataMap[i] = []
- keyColumn = self.mapDBInfo.layers[self.layer]['key']
- j = 0
- cat = None
-
- if keyColumn == 'OGC_FID':
- self.itemDataMap[i].append(i+1)
- j += 1
- cat = i + 1
-
- for value in record:
- if self.columns[columns[j]]['ctype'] != types.StringType:
- try:
- ### casting disabled (2009/03)
- ### self.itemDataMap[i].append(self.columns[columns[j]]['ctype'](value))
- self.itemDataMap[i].append(value)
- except ValueError:
- self.itemDataMap[i].append(_('Unknown value'))
- else:
- # encode string values
- try:
- self.itemDataMap[i].append(GetUnicodeValue(value))
- except UnicodeDecodeError:
- self.itemDataMap[i].append(_("Unable to decode value. "
- "Set encoding in GUI preferences ('Attributes')."))
-
- if not cat and keyId > -1 and keyId == j:
- try:
- cat = self.columns[columns[j]]['ctype'] (value)
- except ValueError, e:
- cat = -1
- GError(parent = self,
- message = _("Error loading attribute data. "
- "Record number: %(rec)d. Unable to convert value '%(val)s' in "
- "key column (%(key)s) to integer.\n\n"
- "Details: %(detail)s") % \
- { 'rec' : i + 1, 'val' : value,
- 'key' : keyColumn, 'detail' : e})
- j += 1
-
- self.itemIndexMap.append(i)
- if keyId > -1: # load cats only when LoadData() is called first time
- self.itemCatsMap[i] = cat
-
- 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 = self.itemDataMap[index][col]
- return s
- def OnGetItemAttr(self, item):
- """!Get item attributes"""
- if ( item % 2) == 0:
- return self.attr2
- else:
- return self.attr1
- def OnColumnMenu(self, event):
- """!Column heading right mouse button -> pop-up menu"""
- self._col = event.GetColumn()
-
- popupMenu = wx.Menu()
- if not hasattr (self, "popupID1"):
- #TODO put to dict
- self.popupID1 = wx.NewId()
- self.popupID2 = wx.NewId()
- self.popupID3 = wx.NewId()
- self.popupID4 = wx.NewId()
- self.popupID5 = wx.NewId()
- self.popupID6 = wx.NewId()
- self.popupID7 = wx.NewId()
- self.popupID8 = wx.NewId()
- self.popupID9 = wx.NewId()
- self.popupID10 = wx.NewId()
- self.popupID11 = wx.NewId()
- self.popupID12 = wx.NewId()
- self.popupID13 = wx.NewId()
- popupMenu.Append(self.popupID1, text = _("Sort ascending"))
- popupMenu.Append(self.popupID2, text = _("Sort descending"))
- popupMenu.AppendSeparator()
- subMenu = wx.Menu()
- popupMenu.AppendMenu(self.popupID3, _("Calculate (only numeric columns)"),
- subMenu)
- popupMenu.Append(self.popupID13, text = _("Field calculator"))
- if not self.pages['manageTable']:
- popupMenu.AppendSeparator()
- self.popupID14 = wx.NewId()
- popupMenu.Append(self.popupID14, text = _("Add column"))
- if not self.dbMgrData['editable']:
- popupMenu.Enable(self.popupID14, False)
- if not self.dbMgrData['editable']:
- popupMenu.Enable(self.popupID13, False)
-
- if not self.dbMgrData['editable'] or \
- self.columns[self.GetColumn(self._col).GetText()]['ctype'] not in (types.IntType, types.FloatType):
- popupMenu.Enable(self.popupID3, False)
-
- subMenu.Append(self.popupID4, text = _("Area size"))
- subMenu.Append(self.popupID5, text = _("Line length"))
- subMenu.Append(self.popupID6, text = _("Compactness of an area"))
- subMenu.Append(self.popupID7, text = _("Fractal dimension of boundary defining a polygon"))
- subMenu.Append(self.popupID8, text = _("Perimeter length of an area"))
- subMenu.Append(self.popupID9, text = _("Number of features for each category"))
- subMenu.Append(self.popupID10, text = _("Slope steepness of 3D line"))
- subMenu.Append(self.popupID11, text = _("Line sinuousity"))
- subMenu.Append(self.popupID12, text = _("Line azimuth"))
-
- self.Bind (wx.EVT_MENU, self.OnColumnSortAsc, id = self.popupID10)
- self.Bind (wx.EVT_MENU, self.OnColumnSortDesc, id = self.popupID2)
- self.Bind(wx.EVT_MENU, self.OnFiledCalculator, id = self.popupID13)
- if not self.pages['manageTable']:
- self.Bind(wx.EVT_MENU, self.OnAddColumn, id = self.popupID14)
- for id in (self.popupID4, self.popupID5, self.popupID6,
- self.popupID7, self.popupID8, self.popupID9,
- self.popupID10, self.popupID11, self.popupID12):
- self.Bind(wx.EVT_MENU, self.OnColumnCompute, id = id)
- self.PopupMenu(popupMenu)
- popupMenu.Destroy()
- def OnColumnSort(self, event):
- """!Column heading left mouse button -> sorting"""
- self._col = event.GetColumn()
-
- self.ColumnSort()
-
- event.Skip()
- def OnColumnSortAsc(self, event):
- """!Sort values of selected column (ascending)"""
- self.SortListItems(col = self._col, ascending = True)
- event.Skip()
- def OnColumnSortDesc(self, event):
- """!Sort values of selected column (descending)"""
- self.SortListItems(col = self._col, ascending = False)
- event.Skip()
-
- def OnColumnCompute(self, event):
- """!Compute values of selected column"""
- id = event.GetId()
-
- option = None
- if id == self.popupID4:
- option = 'area'
- elif id == self.popupID5:
- option = 'length'
- elif id == self.popupID6:
- option = 'compact'
- elif id == self.popupID7:
- option = 'fd'
- elif id == self.popupID8:
- option = 'perimeter'
- elif id == self.popupID9:
- option = 'count'
- elif id == self.popupID10:
- option = 'slope'
- elif id == self.popupID11:
- option = 'sinuous'
- elif id == self.popupID12:
- option = 'azimuth'
-
- if not option:
- return
-
- RunCommand('v.to.db',
- parent = self.parent,
- map = self.mapDBInfo.map,
- layer = self.layer,
- option = option,
- columns = self.GetColumn(self._col).GetText())
-
- self.LoadData(self.layer)
-
- def ColumnSort(self):
- """!Sort values of selected column (self._col)"""
- # 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)
-
- def OnFiledCalculator(self, event):
- """!Calls SQLBuilderUpdate instance"""
- if not self.fieldCalc:
- self.fieldCalc = SQLBuilderUpdate(parent = self, id = wx.ID_ANY,
- vectmap = self.dbMgrData['vectName'],
- layer = self.layer,
- column = self.GetColumn(self._col).GetText())
- self.fieldCalc.Show()
- else:
- self.fieldCalc.Raise()
- def OnAddColumn(self, event):
- """!Add column into table"""
- table = self.dbMgrData['mapDBInfo'].layers[self.layer]['table']
- dlg = AddColumnDialog(parent = self, title = _('Add column to table <%s>') % table)
- if not dlg:
- return
- if dlg.ShowModal() == wx.ID_OK:
- data = dlg.GetData()
- self.pages['browse'].AddColumn(name = data['name'],
- ctype = data['ctype'],
- length = data['length'])
- dlg.Destroy()
- 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) == types.StringType or type(item2) == types.StringTypes:
- 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)
- def IsEmpty(self):
- """!Check if list if empty"""
- if self.columns:
- return False
-
- return True
- class DbMgrBase:
- def __init__(self, id = wx.ID_ANY, mapdisplay = None,
- vectorName = None, item = None, giface = None,
- statusbar = None,
- **kwargs):
- """!Base class, which enables usage of separate pages of Attribute Table Manager
- @param id window id
- @param mapdisplay MapFrame instance
- @param vetorName name of vector map
- @param item item from Layer Tree
- @param log log window
- @param statusbar widget with statusbar
- @param kwagrs other wx.Frame's arguments
- """
- # stores all data, which are shared by pages
- self.dbMgrData = {}
- self.dbMgrData['vectName'] = vectorName
- self.dbMgrData['treeItem'] = item # item in layer tree
- self.mapdisplay = mapdisplay
- if self.mapdisplay:
- self.map = mapdisplay.Map
- else:
- self.map = None
- if not self.mapdisplay:
- pass
- elif self.mapdisplay.tree and \
- self.dbMgrData['treeItem'] and not self.dbMgrData['vectName']:
- maptree = self.mapdisplay.tree
- name = maptree.GetLayerInfo(self.dbMgrData['treeItem'], key = 'maplayer').GetName()
- self.dbMgrData['vectName'] = name
-
- # vector attributes can be changed only if vector map is in
- # the current mapset
- mapInfo = None
- if self.dbMgrData['vectName']:
- mapInfo = grass.find_file(name = self.dbMgrData['vectName'], element = 'vector')
- if not mapInfo or mapInfo['mapset'] != grass.gisenv()['MAPSET']:
- self.dbMgrData['editable'] = False
- else:
- self.dbMgrData['editable'] = True
-
- self.giface = giface
- # status bar log class
- self.log = Log(statusbar) # -> statusbar
- # -> layers / tables description
- self.dbMgrData['mapDBInfo'] = VectorDBInfo(self.dbMgrData['vectName'])
- # store information, which pages were initialized
- self.pages = {
- 'browse' : None,
- 'manageTable' : None,
- 'manageLayer' : None
- }
- def ChangeVectorMap(self, vectorName):
- """!Change of vector map
- Does not import layers of new vector map into pages.
- For the import use methods addLayer in DbMgrBrowsePage and DbMgrTablesPage
- """
- if self.pages['browse']:
- self.pages['browse'].DeleteAllPages()
- if self.pages['manageTable']:
- self.pages['manageTable'].DeleteAllPages()
- self.dbMgrData['vectName'] = vectorName
- # fetch fresh db info
- self.dbMgrData['mapDBInfo'] = VectorDBInfo(self.dbMgrData['vectName'])
- # vector attributes can be changed only if vector map is in
- # the current mapset
- mapInfo = grass.find_file(name = self.dbMgrData['vectName'], element = 'vector')
- if not mapInfo or mapInfo['mapset'] != grass.gisenv()['MAPSET']:
- self.dbMgrData['editable'] = False
- else:
- self.dbMgrData['editable'] = True
- # 'manage layers page
- if self.pages['manageLayer']:
- self.pages['manageLayer'].UpdatePage()
- def CreateDbMgrPage(self, parent, pageName, onlyLayer = -1):
- """!Creates chosen page
- @param pageName can be 'browse' or 'manageTable' or 'manageLayer' which corresponds with pages in
- Attribute Table Manager
- @return created instance of page - if the page has been already created returns the previously created instance
- @return None if wrong identifier was passed
- """
- if pageName == 'browse':
- if not self.pages['browse']:
- self.pages[pageName] = DbMgrBrowsePage(parent = parent, parentDbMgrBase = self,
- onlyLayer = onlyLayer)
- return self.pages[pageName]
- if pageName == 'manageTable':
- if not self.pages['manageTable']:
- self.pages[pageName] = DbMgrTablesPage(parent = parent, parentDbMgrBase = self,
- onlyLayer = onlyLayer)
- return self.pages[pageName]
- if pageName == 'manageLayer':
- if not self.pages['manageLayer']:
- self.pages[pageName] = DbMgrLayersPage(parent = parent, parentDbMgrBase = self)
- return self.pages[pageName]
- return None
- def UpdateDialog(self, layer):
- """!Updates dialog layout for given layer"""
- # delete page
- if layer in self.dbMgrData['mapDBInfo'].layers.keys():
- # delete page
- # draging pages disallowed
- # if self.browsePage.GetPageText(page).replace('Layer ', '').strip() == str(layer):
- # self.browsePage.DeletePage(page)
- # break
- if self.pages['browse']:
- self.pages['browse'].DeletePage(layer)
- if self.pages['manageTable']:
- self.pages['manageTable'].DeletePage(layer)
- # fetch fresh db info
- self.dbMgrData['mapDBInfo'] = VectorDBInfo(self.dbMgrData['vectName'])
- #
- # add new page
- #
- if layer in self.dbMgrData['mapDBInfo'].layers.keys():
- # 'browse data' page
- if self.pages['browse']:
- self.pages['browse'].AddLayer(layer)
- # 'manage tables' page
- if self.pages['manageTable']:
- self.pages['manageTable'].AddLayer(layer)
- # manage layers page
- if self.pages['manageLayer']:
- self.pages['manageLayer'].UpdatePage()
-
- def GetVectorName(self):
- """!Get vector name"""
- return self.dbMgrData['vectName']
- def GetVectorLayers(self):
- """!Get layers of vector map which have table"""
- return self.dbMgrData['mapDBInfo'].layers.keys()
- class DbMgrNotebookBase(FN.FlatNotebook):
- def __init__(self, parent, parentDbMgrBase):
- """!Base class for notebook with attribute tables in tabs
- @param parent GUI parent
- @param parentDbMgrBase instance of DbMgrBase class
- """
- self.parent = parent
- self.parentDbMgrBase = parentDbMgrBase
- self.log = self.parentDbMgrBase.log
- self.giface = self.parentDbMgrBase.giface
- self.map = self.parentDbMgrBase.map
- self.mapdisplay = self.parentDbMgrBase.mapdisplay
- #TODO no need to have it in class scope make it local?
- self.listOfCommands = []
- self.listOfSQLStatements = []
- #initializet pages
- self.pages = self.parentDbMgrBase.pages
- # shared data among pages
- self.dbMgrData = self.parentDbMgrBase.dbMgrData
- # set up virtual lists (each layer)
- ### {layer: list, widgets...}
- self.layerPage = {}
- # currently selected layer
- self.selLayer = None
- # list which represents layers numbers in order of tabs
- self.layers = []
- if globalvar.hasAgw:
- dbmStyle = { 'agwStyle' : globalvar.FNPageStyle }
- else:
- dbmStyle = { 'style' : globalvar.FNPageStyle }
- FN.FlatNotebook.__init__(self,parent = self.parent, id = wx.ID_ANY,
- **dbmStyle)
- self.Bind(FN.EVT_FLATNOTEBOOK_PAGE_CHANGED, self.OnLayerPageChanged)
- def OnLayerPageChanged(self, event):
- """!Layer tab changed"""
- # because of SQL Query notebook
- if event.GetEventObject() != self:
- return
- pageNum = self.GetSelection()
- self.selLayer = self.layers[pageNum]
- try:
- idCol = self.layerPage[self.selLayer]['whereColumn']
- except KeyError:
- idCol = None
-
- try:
- # update statusbar
- self.log.write(_("Number of loaded records: %d") % \
- self.FindWindowById(self.layerPage[self.selLayer]['data']).\
- GetItemCount())
- except:
- pass
-
- if idCol:
- winCol = self.FindWindowById(idCol)
- table = self.dbMgrData['mapDBInfo'].layers[self.selLayer]["table"]
- self.dbMgrData['mapDBInfo'].GetColumns(table)
- def ApplyCommands(self, listOfCommands, listOfSQLStatements):
- """!Apply changes"""
- # perform GRASS commands (e.g. v.db.addcolumn)
- wx.BeginBusyCursor()
-
- if len(listOfCommands) > 0:
- for cmd in listOfCommands:
- RunCommand(prog = cmd[0],
- quiet = True,
- parent = self,
- **cmd[1])
-
- self.dbMgrData['mapDBInfo'] = VectorDBInfo(self.dbMgrData['vectName'])
- if self.pages['manageTable']:
- self.pages['manageTable'].UpdatePage(self.selLayer)
- if self.pages['browse']:
- self.pages['browse'].UpdatePage(self.selLayer)
- # reset list of commands
- listOfCommands = []
-
- # perform SQL non-select statements (e.g. 'delete from table where cat=1')
- if len(listOfSQLStatements) > 0:
- fd, sqlFilePath = tempfile.mkstemp(text=True)
- sqlFile = open(sqlFilePath, 'w')
- for sql in listOfSQLStatements:
- enc = UserSettings.Get(group = 'atm', key = 'encoding', subkey = 'value')
- if not enc and 'GRASS_DB_ENCODING' in os.environ:
- enc = os.environ['GRASS_DB_ENCODING']
- if enc:
- sqlFile.write(sql.encode(enc) + ';')
- else:
- sqlFile.write(sql + ';')
- sqlFile.write(os.linesep)
- sqlFile.close()
- driver = self.dbMgrData['mapDBInfo'].layers[self.selLayer]["driver"]
- database = self.dbMgrData['mapDBInfo'].layers[self.selLayer]["database"]
-
- Debug.msg(3, 'AttributeManger.ApplyCommands(): %s' %
- ';'.join(["%s" % s for s in listOfSQLStatements]))
-
- RunCommand('db.execute',
- parent = self,
- input = sqlFilePath,
- driver = driver,
- database = database)
-
- os.close(fd)
- os.remove(sqlFilePath)
- # reset list of statements
- listOfSQLStatements = []
-
- wx.EndBusyCursor()
- def DeletePage(self, layer):
- """!Removes layer page"""
- if layer not in self.layers:
- return False
- FN.FlatNotebook.DeletePage(self, self.layers.index(layer))
- self.layers.remove(layer)
- del self.layerPage[layer]
- if self.GetSelection() >= 0:
- self.selLayer = self.layers[self.GetSelection()]
- else:
- self.selLayer = None
- return True
- def DeleteAllPages(self):
- """!Removes all layer pages"""
- FN.FlatNotebook.DeleteAllPages(self)
- self.layerPage = {}
- self.layers = []
- self.selLayer = None
- def AddColumn(self, name, ctype, length):
- """!Add new column to the table"""
- table = self.dbMgrData['mapDBInfo'].layers[self.selLayer]['table']
-
- if not name:
- GError(parent = self,
- message = _("Unable to add column to the table. "
- "No column name defined."))
- return False
-
- # cast type if needed
- if ctype == 'double':
- ctype = 'double precision'
- if ctype != 'varchar':
- length = '' # FIXME
-
- # check for duplicate items
- if name in self.dbMgrData['mapDBInfo'].GetColumns(table):
- GError(parent = self,
- message = _("Column <%(column)s> already exists in table <%(table)s>.") % \
- {'column' : name, 'table' : self.dbMgrData['mapDBInfo'].layers[self.selLayer]["table"]}
- )
- return False
- # add v.db.addcolumn command to the list
- if ctype == 'varchar':
- ctype += ' (%d)' % length
- self.listOfCommands.append(('v.db.addcolumn',
- { 'map' : self.dbMgrData['vectName'],
- 'layer' : self.selLayer,
- 'columns' : '%s %s' % (name, ctype) }
- ))
- # apply changes
- self.ApplyCommands(self.listOfCommands, self.listOfSQLStatements)
-
- return True
- def GetAddedLayers(self):
- """!Get list of added layers"""
- return self.layers[:]
- class DbMgrBrowsePage(DbMgrNotebookBase):
- def __init__(self, parent, parentDbMgrBase, onlyLayer = -1):
- """!Browse page class
- @param parent GUI parent
- @param parentDbMgrBase instance of DbMgrBase class
- @param onlyLayer create only tab of given layer, if -1 creates tabs of all layers
- """
- DbMgrNotebookBase.__init__(self, parent = parent,
- parentDbMgrBase = parentDbMgrBase)
- # for Sql Query notebook adaptation on current width
- self.sqlBestSize = None
- for layer in self.dbMgrData['mapDBInfo'].layers.keys():
- if onlyLayer > 0 and layer != onlyLayer:
- continue
- self.AddLayer(layer)
- if self.layers:
- self.SetSelection(0)
- self.selLayer = self.layers[0]
- self.log.write(_("Number of loaded records: %d") % \
- self.FindWindowById(self.layerPage[self.selLayer]['data']).GetItemCount())
- # query map layer (if parent (GMFrame) is given)
- self.qlayer = None
- # sqlbuilder
- self.builder = None
- def AddLayer(self, layer, pos = -1):
- """!Adds tab which represents table and enables browse it
- @param layer vector map layer conntected to table
- @param pos position of tab, if -1 it is added to end
- @return True if layer was added
- @return False if layer was not added - layer has been already added or has empty table or does not exist
- """
- if layer in self.layers or \
- layer not in self.parentDbMgrBase.GetVectorLayers():
- return False
- panel = wx.Panel(parent = self, id = wx.ID_ANY)
-
- #IMPORTANT NOTE: wx.StaticBox MUST be defined BEFORE any of the
- # controls that are placed IN the wx.StaticBox, or it will freeze
- # on the Mac
-
- 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)
-
- win = VirtualAttributeList(panel, self.log,
- self.dbMgrData, layer, self.pages)
- if win.IsEmpty():
- panel.Destroy()
- return False
- self.layers.append(layer)
-
- win.Bind(wx.EVT_LIST_ITEM_ACTIVATED, self.OnDataItemActivated)
- self.layerPage[layer] = {'browsePage': panel.GetId()}
-
- label = _("Table")
- if not self.dbMgrData['editable']:
- label += _(" (readonly)")
- if pos == -1:
- pos = self.GetPageCount()
- self.InsertPage(pos, page = panel,
- text = " %d / %s %s" % \
- (layer, label, self.dbMgrData['mapDBInfo'].layers[layer]['table']))
-
- pageSizer = wx.BoxSizer(wx.VERTICAL)
- sqlQueryPanel = wx.Panel(parent = panel, id = wx.ID_ANY)
- # attribute data
- sqlBox = wx.StaticBox(parent = sqlQueryPanel, id = wx.ID_ANY,
- label = " %s " % _("SQL Query"))
- sqlSizer = wx.StaticBoxSizer(sqlBox, wx.VERTICAL)
- 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
- FNPageStyle = FN.FNB_NO_NAV_BUTTONS | \
- FN.FNB_NO_X_BUTTON
- if globalvar.hasAgw:
- dbmStyle = { 'agwStyle' : FNPageStyle }
- else:
- dbmStyle = { 'style' : FNPageStyle }
- sqlNtb = FN.FlatNotebook(parent = sqlQueryPanel, id = wx.ID_ANY,
- **dbmStyle)
- # Simple tab
- simpleSqlPanel = wx.Panel(parent = sqlNtb, id = wx.ID_ANY)
- sqlNtb.AddPage(page = simpleSqlPanel,
- text = _('Simple'))
- btnApply = wx.Button(parent = simpleSqlPanel, id = wx.ID_APPLY, name = 'btnApply')
- btnApply.SetToolTipString(_("Apply SELECT statement and reload data records"))
- btnApply.Bind(wx.EVT_BUTTON, self.OnApplySqlStatement)
- whereSimpleSqlPanel = wx.Panel(parent = simpleSqlPanel, id = wx.ID_ANY, name = 'wherePanel')
- sqlWhereColumn = wx.ComboBox(parent = whereSimpleSqlPanel, id = wx.ID_ANY,
- size = (150,-1),
- style = wx.CB_SIMPLE | wx.CB_READONLY,
- choices = self.dbMgrData['mapDBInfo'].GetColumns(self.dbMgrData['mapDBInfo'].layers[layer]['table']))
- sqlWhereColumn.SetSelection(0)
- sqlWhereCond = wx.Choice(parent = whereSimpleSqlPanel, id = wx.ID_ANY,
- size = (55,-1),
- choices = ['=', '!=', '<', '<=', '>', '>='])
- sqlWhereValue = wx.TextCtrl(parent = whereSimpleSqlPanel, id = wx.ID_ANY, value = "",
- style = wx.TE_PROCESS_ENTER)
- sqlWhereValue.SetToolTipString(_("Example: %s") % "MULTILANE = 'no' AND OBJECTID < 10")
- sqlLabel = wx.StaticText(parent = simpleSqlPanel, id = wx.ID_ANY,
- label = "SELECT * FROM %s WHERE " % \
- self.dbMgrData['mapDBInfo'].layers[layer]['table'])
- # Advanced tab
- advancedSqlPanel = wx.Panel(parent = sqlNtb, id = wx.ID_ANY)
- sqlNtb.AddPage(page = advancedSqlPanel,
- text = _('Advanced'))
- btnSqlBuilder = wx.Button(parent = advancedSqlPanel, id = wx.ID_ANY, label = _("SQL Builder"))
- btnSqlBuilder.Bind(wx.EVT_BUTTON, self.OnBuilder)
- sqlStatement = wx.TextCtrl(parent = advancedSqlPanel, id = wx.ID_ANY,
- value = "SELECT * FROM %s" % \
- self.dbMgrData['mapDBInfo'].layers[layer]['table'],
- style = wx.TE_PROCESS_ENTER)
- sqlStatement.SetToolTipString(_("Example: %s") % "SELECT * FROM roadsmajor WHERE MULTILANE = 'no' AND OBJECTID < 10")
- sqlWhereValue.Bind(wx.EVT_TEXT_ENTER, self.OnApplySqlStatement)
- sqlStatement.Bind(wx.EVT_TEXT_ENTER, self.OnApplySqlStatement)
- # Simple tab layout
- simpleSqlSizer = wx.GridBagSizer (hgap = 5, vgap = 5)
- sqlSimpleWhereSizer= wx.BoxSizer(wx.HORIZONTAL)
- sqlSimpleWhereSizer.Add(item = sqlWhereColumn,
- flag = wx.EXPAND | wx.ALIGN_CENTER_VERTICAL| wx.LEFT,
- border = 3)
- sqlSimpleWhereSizer.Add(item = sqlWhereCond,
- flag = wx.EXPAND | wx.ALIGN_CENTER_VERTICAL | wx.LEFT,
- border = 3)
- sqlSimpleWhereSizer.Add(item = sqlWhereValue, proportion = 1,
- flag = wx.EXPAND | wx.ALIGN_CENTER_VERTICAL | wx.LEFT,
- border = 3)
- whereSimpleSqlPanel.SetSizer(sqlSimpleWhereSizer)
- simpleSqlSizer.Add(item = sqlLabel, border = 5, pos = (0, 0),
- flag = wx.ALIGN_CENTER_VERTICAL | wx.TOP | wx.LEFT)
- simpleSqlSizer.Add(item = whereSimpleSqlPanel, border = 5, pos = (0, 1),
- flag = wx.ALIGN_CENTER_VERTICAL | wx.TOP | wx.EXPAND)
- simpleSqlSizer.Add(item = btnApply, border = 5, pos = (0, 2),
- flag = wx.ALIGN_CENTER_VERTICAL | wx.TOP)
- simpleSqlSizer.AddGrowableCol(1)
- simpleSqlPanel.SetSizer(simpleSqlSizer)
- # Advanced tab layout
- advancedSqlSizer = wx.FlexGridSizer (cols = 2, hgap = 5, vgap = 5)
- advancedSqlSizer.AddGrowableCol(0)
- advancedSqlSizer.Add(item = sqlStatement,
- flag = wx.EXPAND | wx.ALL, border = 5)
- advancedSqlSizer.Add(item = btnSqlBuilder,
- flag = wx.ALIGN_RIGHT | wx.TOP | wx.RIGHT | wx.BOTTOM, border = 5)
- sqlSizer.Add(item = sqlNtb,
- flag = wx.ALL | wx.EXPAND,
- border = 3)
- advancedSqlPanel.SetSizer(advancedSqlSizer)
- pageSizer.Add(item = listSizer,
- proportion = 1,
- flag = wx.ALL | wx.EXPAND,
- border = 5)
- sqlQueryPanel.SetSizer(sqlSizer)
-
- pageSizer.Add(item = sqlQueryPanel,
- proportion = 0,
- flag = wx.BOTTOM | wx.LEFT | wx.RIGHT | wx.EXPAND,
- border = 5)
- panel.SetSizer(pageSizer)
-
- sqlNtb.Bind(wx.EVT_SIZE, self.OnSqlQuerySizeWrap(layer))
- self.layerPage[layer]['data'] = win.GetId()
- self.layerPage[layer]['sqlNtb'] = sqlNtb.GetId()
- self.layerPage[layer]['whereColumn'] = sqlWhereColumn.GetId()
- self.layerPage[layer]['whereOperator'] = sqlWhereCond.GetId()
- self.layerPage[layer]['where'] = sqlWhereValue.GetId()
- self.layerPage[layer]['builder'] = btnSqlBuilder.GetId()
- self.layerPage[layer]['statement'] = sqlStatement.GetId()
- self.layerPage[layer]['sqlIsReduced'] = False # for SQL Query adaptation on width
- return True
- def OnSqlQuerySizeWrap(self, layer):
- """!Helper function"""
- return lambda event : self.OnSqlQuerySize(event, layer)
- def OnSqlQuerySize(self, event, layer):
- """!Adapts SQL Query Simple tab on current width"""
- sqlNtb = event.GetEventObject()
- if not self.sqlBestSize:
- self.sqlBestSize = sqlNtb.GetBestSize()
- size = sqlNtb.GetSize()
- sqlReduce = self.sqlBestSize[0] > size[0]
- if (sqlReduce and self.layerPage[layer]['sqlIsReduced']) or \
- (not sqlReduce and not self.layerPage[layer]['sqlIsReduced']):
- event.Skip()
- return
- wherePanel = sqlNtb.FindWindowByName('wherePanel')
- btnApply = sqlNtb.FindWindowByName('btnApply')
- sqlSimpleSizer = btnApply.GetContainingSizer()
- if sqlReduce:
- self.layerPage[layer]['sqlIsReduced'] = True
- sqlSimpleSizer.AddGrowableCol(0)
- sqlSimpleSizer.RemoveGrowableCol(1)
- sqlSimpleSizer.SetItemPosition(wherePanel, (1, 0))
- sqlSimpleSizer.SetItemPosition(btnApply, (1, 1))
- else:
- self.layerPage[layer]['sqlIsReduced'] = False
- sqlSimpleSizer.AddGrowableCol(1)
- sqlSimpleSizer.RemoveGrowableCol(0)
- sqlSimpleSizer.SetItemPosition(wherePanel, (0, 1))
- sqlSimpleSizer.SetItemPosition(btnApply, (0, 2))
- event.Skip()
- def OnDataItemActivated(self, event):
- """!Item activated, highlight selected item"""
- self.OnDataDrawSelected(event)
- event.Skip()
- 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.popupDataID11 = 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.OnDeleteSelected, id = self.popupDataID11)
- self.Bind(wx.EVT_MENU, self.OnDataReload, id = self.popupDataID10)
- tlist = self.FindWindowById(self.layerPage[self.selLayer]['data'])
- # generate popup-menu
- menu = wx.Menu()
- menu.Append(self.popupDataID1, _("Edit selected record"))
- selected = tlist.GetFirstSelected()
- if not self.dbMgrData['editable'] or selected == -1 or tlist.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"))
- if not self.dbMgrData['editable']:
- menu.Enable(self.popupDataID2, False)
- menu.Enable(self.popupDataID3, False)
- menu.Enable(self.popupDataID4, False)
- menu.AppendSeparator()
- menu.Append(self.popupDataID5, _("Select all"))
- menu.Append(self.popupDataID6, _("Deselect all"))
- menu.AppendSeparator()
- menu.Append(self.popupDataID7, _("Highlight selected features"))
- menu.Append(self.popupDataID8, _("Highlight selected features and zoom"))
- if not self.map or len(tlist.GetSelectedItems()) == 0:
- menu.Enable(self.popupDataID7, False)
- menu.Enable(self.popupDataID8, False)
- menu.Append(self.popupDataID9, _("Extract selected features"))
- menu.Append(self.popupDataID11, _("Delete selected features"))
- if not self.dbMgrData['editable']:
- menu.Enable(self.popupDataID11, False)
- if tlist.GetFirstSelected() == -1:
- menu.Enable(self.popupDataID3, False)
- menu.Enable(self.popupDataID9, False)
- menu.Enable(self.popupDataID11, False)
- menu.AppendSeparator()
- menu.Append(self.popupDataID10, _("Reload"))
- self.PopupMenu(menu)
- menu.Destroy()
- # update statusbar
- self.log.write(_("Number of loaded records: %d") % \
- tlist.GetItemCount())
- def OnDataItemEdit(self, event):
- """!Edit selected record of the attribute table"""
- tlist = self.FindWindowById(self.layerPage[self.selLayer]['data'])
- item = tlist.GetFirstSelected()
- if item == -1:
- return
- table = self.dbMgrData['mapDBInfo'].layers[self.selLayer]['table']
- keyColumn = self.dbMgrData['mapDBInfo'].layers[self.selLayer]['key']
- cat = tlist.itemCatsMap[tlist.itemIndexMap[item]]
- # (column name, value)
- data = []
- # collect names of all visible columns
- columnName = []
- for i in range(tlist.GetColumnCount()):
- columnName.append(tlist.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)):
- ctype = self.dbMgrData['mapDBInfo'].tables[table][columnName[i]]['ctype']
- ctypeStr = self.dbMgrData['mapDBInfo'].tables[table][columnName[i]]['type']
- if columnName[i] == keyColumn: # key
- if missingKey is False:
- data.append((columnName[i], ctype, ctypeStr, str(cat)))
- keyId = i
- else:
- if missingKey is True:
- value = tlist.GetItem(item, i-1).GetText()
- else:
- value = tlist.GetItem(item, i).GetText()
- data.append((columnName[i], ctype, ctypeStr, value))
- dlg = ModifyTableRecord(parent = self,
- title = _("Update existing record"),
- data = data, keyEditable = (keyId, False))
- if dlg.ShowModal() == wx.ID_OK:
- values = dlg.GetValues() # string
- updateList = list()
- try:
- for i in range(len(values)):
- if i == keyId: # skip key column
- continue
- if tlist.GetItem(item, i).GetText() == values[i]:
- continue # no change
-
- column = tlist.columns[columnName[i]]
- if len(values[i]) > 0:
- try:
- if missingKey is True:
- idx = i - 1
- else:
- idx = i
-
- if column['ctype'] != types.StringType:
- tlist.itemDataMap[item][idx] = column['ctype'] (values[i])
- else: # -> string
- tlist.itemDataMap[item][idx] = values[i]
- except ValueError:
- raise ValueError(_("Value '%(value)s' needs to be entered as %(type)s.") % \
- {'value' : str(values[i]),
- 'type' : column['type']})
-
- if column['ctype'] == types.StringType:
- if "'" in values[i]: # replace "'" -> "''"
- values[i] = values[i].replace("'", "''")
- updateList.append("%s='%s'" % (columnName[i], values[i]))
- else:
- updateList.append("%s=%s" % (columnName[i], values[i]))
- else: # -> NULL
- updateList.append("%s=NULL" % (columnName[i]))
- except ValueError, err:
- GError(parent = self,
- message = _("Unable to update existing record.\n%s") % err,
- showTraceback = False)
- self.OnDataItemEdit(event)
- return
-
- if updateList:
- self.listOfSQLStatements.append('UPDATE %s SET %s WHERE %s=%d' % \
- (table, ','.join(updateList),
- keyColumn, cat))
- self.ApplyCommands(self.listOfCommands, self.listOfSQLStatements)
-
- tlist.Update()
-
- def OnDataItemAdd(self, event):
- """!Add new record to the attribute table"""
- tlist = self.FindWindowById(self.layerPage[self.selLayer]['data'])
- table = self.dbMgrData['mapDBInfo'].layers[self.selLayer]['table']
- keyColumn = self.dbMgrData['mapDBInfo'].layers[self.selLayer]['key']
-
- # (column name, value)
- data = []
- # collect names of all visible columns
- columnName = []
- for i in range(tlist.GetColumnCount()):
- columnName.append(tlist.GetColumn(i).GetText())
- # maximal category number
- if len(tlist.itemCatsMap.values()) > 0:
- maxCat = max(tlist.itemCatsMap.values())
- else:
- maxCat = 0 # starting category '1'
-
- # 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:
- ctype = self.dbMgrData['mapDBInfo'].tables[table][col]['ctype']
- ctypeStr = self.dbMgrData['mapDBInfo'].tables[table][col]['type']
- if col == keyColumn: # key
- if missingKey is False:
- data.append((col, ctype, ctypeStr, str(maxCat + 1)))
- keyId = colIdx
- else:
- data.append((col, ctype, ctypeStr, ''))
-
- colIdx += 1
-
- dlg = ModifyTableRecord(parent = self,
- 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 tlist.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:
- if tlist.columns[columnName[i]]['ctype'] == int:
- # values[i] is stored as text.
- value = float(values[i])
- else:
- value = values[i]
- values[i] = tlist.columns[columnName[i]]['ctype'] (value)
- except:
- raise ValueError(_("Value '%(value)s' needs to be entered as %(type)s.") %
- {'value' : str(values[i]),
- 'type' : tlist.columns[columnName[i]]['type']})
- columnsString += '%s,' % columnName[i]
- if tlist.columns[columnName[i]]['ctype'] == str:
- valuesString += "'%s'," % values[i]
- else:
- valuesString += "%s," % values[i]
- except ValueError, err:
- GError(parent = self,
- message = _("Unable to insert new record.\n%s") % err,
- showTraceback = False)
- self.OnDataItemAdd(event)
- return
-
- # remove category if need
- if missingKey is True:
- del values[0]
-
- # add new item to the tlist
- if len(tlist.itemIndexMap) > 0:
- index = max(tlist.itemIndexMap) + 1
- else:
- index = 0
-
- tlist.itemIndexMap.append(index)
- tlist.itemDataMap[index] = values
- tlist.itemCatsMap[index] = cat
- tlist.SetItemCount(tlist.GetItemCount() + 1)
- self.listOfSQLStatements.append('INSERT INTO %s (%s) VALUES(%s)' % \
- (table,
- columnsString.strip(','),
- valuesString.strip(',')))
- self.ApplyCommands(self.listOfCommands, self.listOfSQLStatements)
-
- def OnDataItemDelete(self, event):
- """!Delete selected item(s) from the tlist (layer/category pair)"""
- dlist = self.FindWindowById(self.layerPage[self.selLayer]['data'])
- item = dlist.GetFirstSelected()
-
- table = self.dbMgrData['mapDBInfo'].layers[self.selLayer]["table"]
- key = self.dbMgrData['mapDBInfo'].layers[self.selLayer]["key"]
-
- indeces = []
- # collect SQL statements
- while item != -1:
- index = dlist.itemIndexMap[item]
- indeces.append(index)
-
- cat = dlist.itemCatsMap[index]
-
- self.listOfSQLStatements.append('DELETE FROM %s WHERE %s=%d' % \
- (table, key, cat))
-
- item = dlist.GetNextSelected(item)
-
- if UserSettings.Get(group = 'atm', key = 'askOnDeleteRec', subkey = 'enabled'):
- deleteDialog = wx.MessageBox(parent = self,
- message = _("Selected data records (%d) will be 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 False
-
- # restore maps
- i = 0
- indexTemp = copy.copy(dlist.itemIndexMap)
- dlist.itemIndexMap = []
- dataTemp = copy.deepcopy(dlist.itemDataMap)
- dlist.itemDataMap = {}
- catsTemp = copy.deepcopy(dlist.itemCatsMap)
- dlist.itemCatsMap = {}
-
- i = 0
- for index in indexTemp:
- if index in indeces:
- continue
- dlist.itemIndexMap.append(i)
- dlist.itemDataMap[i] = dataTemp[index]
- dlist.itemCatsMap[i] = catsTemp[index]
-
- i += 1
-
- dlist.SetItemCount(len(dlist.itemIndexMap))
-
- # deselect items
- item = dlist.GetFirstSelected()
- while item != -1:
- dlist.SetItemState(item, 0, wx.LIST_STATE_SELECTED | wx.LIST_STATE_FOCUSED)
- item = dlist.GetNextSelected(item)
-
- # submit SQL statements
- self.ApplyCommands(self.listOfCommands, self.listOfSQLStatements)
-
- return True
- def OnDataItemDeleteAll(self, event):
- """!Delete all items from the list"""
- dlist = self.FindWindowById(self.layerPage[self.selLayer]['data'])
- if UserSettings.Get(group = 'atm', key = 'askOnDeleteRec', subkey = 'enabled'):
- deleteDialog = wx.MessageBox(parent = self,
- message = _("All data records (%d) will be permanently deleted "
- "from table. Do you want to delete them?") % \
- (len(dlist.itemIndexMap)),
- caption = _("Delete records"),
- style = wx.YES_NO | wx.CENTRE)
- if deleteDialog != wx.YES:
- return
- dlist.DeleteAllItems()
- dlist.itemDataMap = {}
- dlist.itemIndexMap = []
- dlist.SetItemCount(0)
- table = self.dbMgrData['mapDBInfo'].layers[self.selLayer]["table"]
- self.listOfSQLStatements.append('DELETE FROM %s' % table)
- self.ApplyCommands(self.listOfCommands, self.listOfSQLStatements)
-
- event.Skip()
- def _drawSelected(self, zoom):
- """!Highlight selected features"""
- if not self.map or not self.mapdisplay:
- return
-
- tlist = self.FindWindowById(self.layerPage[self.selLayer]['data'])
- cats = map(int, tlist.GetSelectedItems())
- digitToolbar = None
- if 'vdigit' in self.mapdisplay.toolbars:
- digitToolbar = self.mapdisplay.toolbars['vdigit']
- if digitToolbar and digitToolbar.GetLayer() and \
- digitToolbar.GetLayer().GetName() == self.dbMgrData['vectName']:
- display = self.mapdisplay.GetMapWindow().GetDisplay()
- display.SetSelected(cats, layer = self.selLayer)
- if zoom:
- n, s, w, e = display.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() # -> self.qlayer
- # set opacity based on queried layer
- if self.parent and self.mapdisplay.tree and \
- self.dbMgrData['treeItem']:
- maptree = self.mapdisplay.tree
- opacity = maptree.GetLayerInfo(self.dbMgrData['treeItem'], key = 'maplayer').GetOpacity()
- self.qlayer.SetOpacity(opacity)
- if zoom:
- keyColumn = self.dbMgrData['mapDBInfo'].layers[self.selLayer]['key']
- where = ''
- for range in 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 ')
-
- select = RunCommand('v.db.select',
- parent = self,
- read = True,
- quiet = True,
- flags = 'r',
- map = self.dbMgrData['mapDBInfo'].map,
- layer = int(self.selLayer),
- where = where)
-
- region = {}
- for line in select.splitlines():
- key, value = line.split('=')
- region[key.strip()] = float(value.strip())
- nsdist = ewdist = 0
- renderer = self.mapdisplay.GetMap()
- nsdist = 10 * ((renderer.GetCurrentRegion()['n'] - renderer.GetCurrentRegion()['s']) /
- renderer.height)
- ewdist = 10 * ((renderer.GetCurrentRegion()['e'] - renderer.GetCurrentRegion()['w']) /
- renderer.width)
- north = region['n'] + nsdist
- south = region['s'] - nsdist
- west = region['w'] - ewdist
- east = region['e'] + ewdist
- renderer.GetRegion(n = north, s = south, w = west, e = east, update = True)
- self.mapdisplay.GetMapWindow().ZoomHistory(n = north, s = south, w = west, e = east)
-
- 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 AddQueryMapLayer(self):
- """!Redraw a map
- Return True if map has been redrawn, False if no map is given
- """
- tlist = self.FindWindowById(self.layerPage[self.selLayer]['data'])
- cats = {
- self.selLayer : tlist.GetSelectedItems()
- }
-
- if self.mapdisplay.Map.GetLayerIndex(self.qlayer) < 0:
- self.qlayer = None
-
- if self.qlayer:
- self.qlayer.SetCmd(self.mapdisplay.AddTmpVectorMapLayer(self.dbMgrData['vectName'], cats, addLayer = False))
- else:
- self.qlayer = self.mapdisplay.AddTmpVectorMapLayer(self.dbMgrData['vectName'], cats)
- return self.qlayer
- def OnDataReload(self, event):
- """!Reload tlist of records"""
- self.OnApplySqlStatement(None)
- self.listOfSQLStatements = []
- def OnDataSelectAll(self, event):
- """!Select all items"""
- tlist = self.FindWindowById(self.layerPage[self.selLayer]['data'])
- item = -1
- while True:
- item = tlist.GetNextItem(item)
- if item == -1:
- break
- tlist.SetItemState(item, wx.LIST_STATE_SELECTED, wx.LIST_STATE_SELECTED)
- event.Skip()
- def OnDataSelectNone(self, event):
- """!Deselect items"""
- tlist = self.FindWindowById(self.layerPage[self.selLayer]['data'])
- item = -1
- while True:
- item = tlist.GetNextItem(item, wx.LIST_STATE_SELECTED)
- if item == -1:
- break
- tlist.SetItemState(item, 0, wx.LIST_STATE_SELECTED | wx.LIST_STATE_FOCUSED)
- event.Skip()
- def OnDataDrawSelected(self, event):
- """!Reload table description"""
- self._drawSelected(zoom = False)
- event.Skip()
- def OnDataDrawSelectedZoom(self, event):
- self._drawSelected(zoom = True)
- event.Skip()
-
- def OnExtractSelected(self, event):
- """!Extract vector objects selected in attribute browse window
- to new vector map
- """
- tlist = self.FindWindowById(self.layerPage[self.selLayer]['data'])
- # cats = tlist.selectedCats[:]
- cats = tlist.GetSelectedItems()
- if len(cats) == 0:
- GMessage(parent = self,
- message = _('Nothing to extract.'))
- return
- else:
- # dialog to get file name
- dlg = CreateNewVector(parent=self, title=_('Extract selected features'),
- giface=self.giface,
- cmd=(('v.extract',
- {'input': self.dbMgrData['vectName'],
- 'cats': ListOfCatsToRange(cats)},
- 'output')),
- disableTable=True)
- if not dlg:
- return
-
- name = dlg.GetName(full = True)
- if not self.mapdisplay and self.mapdisplay.tree:
- pass
- elif name and dlg.IsChecked('add'):
- # add layer to map layer tree
- self.mapdisplay.tree.AddLayer(ltype = 'vector',
- lname = name,
- lcmd = ['d.vect', 'map=%s' % name])
- dlg.Destroy()
-
- def OnDeleteSelected(self, event):
- """!Delete vector objects selected in attribute browse window
- (attribures and geometry)
- """
- tlist = self.FindWindowById(self.layerPage[self.selLayer]['data'])
- cats = tlist.GetSelectedItems()
- if len(cats) == 0:
- GMessage(parent = self,
- message = _('Nothing to delete.'))
-
- return
- display = None
- if not self.mapdisplay:
- pass
- elif 'vdigit' in self.mapdisplay.toolbars:
- digitToolbar = self.mapdisplay.toolbars['vdigit']
- if digitToolbar and digitToolbar.GetLayer() and \
- digitToolbar.GetLayer().GetName() == self.dbMgrData['vectName']:
- display = self.mapdisplay.GetMapWindow().GetDisplay()
- display.SetSelected(map(int, cats), layer = self.selLayer)
- self.mapdisplay.MapWindow.UpdateMap(render = True, renderVector = True)
-
- if self.OnDataItemDelete(None) and self.mapdisplay:
- if display:
- self.mapdisplay.GetMapWindow().digit.DeleteSelectedLines()
- else:
- RunCommand('v.edit',
- parent = self,
- quiet = True,
- map = self.dbMgrData['vectName'],
- tool = 'delete',
- cats = ListOfCatsToRange(cats))
-
- self.mapdisplay.MapWindow.UpdateMap(render = True, renderVector = True)
-
- def OnApplySqlStatement(self, event):
- """!Apply simple/advanced sql statement"""
- keyColumn = -1 # index of key column
- listWin = self.FindWindowById(self.layerPage[self.selLayer]['data'])
- sql = None
- win = self.FindWindowById(self.layerPage[self.selLayer]['sqlNtb'])
- if not win:
- return
-
- wx.BeginBusyCursor()
- if win.GetSelection() == 0:
- # simple sql statement
- whereCol = self.FindWindowById(self.layerPage[self.selLayer]['whereColumn']).GetStringSelection()
- whereOpe = self.FindWindowById(self.layerPage[self.selLayer]['whereOperator']).GetStringSelection()
- whereVal = self.FindWindowById(self.layerPage[self.selLayer]['where']).GetValue().strip()
- try:
- if len(whereVal) > 0:
- keyColumn = listWin.LoadData(self.selLayer, where = whereCol + whereOpe + whereVal)
- else:
- keyColumn = listWin.LoadData(self.selLayer)
- except GException, e:
- GError(parent = self,
- message = _("Loading attribute data failed.\n\n%s") % e.value)
- self.FindWindowById(self.layerPage[self.selLayer]['where']).SetValue('')
- else:
- # advanced sql statement
- win = self.FindWindowById(self.layerPage[self.selLayer]['statement'])
- try:
- cols, where = self.ValidateSelectStatement(win.GetValue())
- if cols is None and where is None:
- sql = win.GetValue()
- except TypeError:
- GError(parent = self,
- message = _("Loading attribute data failed.\n"
- "Invalid SQL select statement.\n\n%s") % win.GetValue())
- win.SetValue("SELECT * FROM %s" % self.dbMgrData['mapDBInfo'].layers[self.selLayer]['table'])
- cols = None
- where = None
-
- if cols or where or sql:
- try:
- keyColumn = listWin.LoadData(self.selLayer, columns = cols,
- where = where, sql = sql)
- except GException, e:
- GError(parent = self,
- message = _("Loading attribute data failed.\n\n%s") % e.value)
- win.SetValue("SELECT * FROM %s" % self.dbMgrData['mapDBInfo'].layers[self.selLayer]['table'])
-
- # sort by key column
- if sql and 'order by' in sql.lower():
- pass # don't order by key column
- else:
- if keyColumn > -1:
- listWin.SortListItems(col = keyColumn, ascending = True)
- else:
- listWin.SortListItems(col = 0, ascending = True)
-
- wx.EndBusyCursor()
-
- # update statusbar
- self.log.write(_("Number of loaded records: %d") % \
- self.FindWindowById(self.layerPage[self.selLayer]['data']).GetItemCount())
- def OnBuilder(self,event):
- """!SQL Builder button pressed -> show the SQLBuilder dialog"""
- if not self.builder:
- self.builder = SQLBuilderSelect(parent = self, id = wx.ID_ANY,
- vectmap = self.dbMgrData['vectName'],
- layer = self.selLayer,
- evtHandler = self.OnBuilderEvt)
- self.builder.Show()
- else:
- self.builder.Raise()
-
- def OnBuilderEvt(self, event):
- if event == 'apply':
- sqlstr = self.builder.GetSQLStatement()
- self.FindWindowById(self.layerPage[self.selLayer]['statement']).SetValue(sqlstr)
- # apply query
- #self.listOfSQLStatements.append(sqlstr) #TODO probably it was bug
- self.OnApplySqlStatement(None)
- # close builder on apply
- if self.builder.CloseOnApply():
- self.builder = None
- elif event == 'close':
- self.builder = None
- 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.dbMgrData['mapDBInfo'].layers[self.selLayer]['table'])
-
- if statement[index+1:index+6].lower() != 'from ' or \
- statement[index+6:index+6+tablelen] != '%s' % \
- (self.dbMgrData['mapDBInfo'].layers[self.selLayer]['table']):
- return None
-
- if len(statement[index+7+tablelen:]) > 0:
- index = statement.lower().find('where ')
- if index > -1:
- where = statement[index+6:]
- else:
- where = None
- else:
- where = None
-
- return (cols, where)
- def LoadData(self, layer, columns = None, where = None, sql = None):
- """!Load data into list
- @param layer layer number
- @param columns list of columns for output
- @param where where statement
- @param sql full sql statement
- @return id of key column
- @return -1 if key column is not displayed
- """
- listWin = self.FindWindowById(self.layerPage[layer]['data'])
- return listWin.LoadData(layer, columns, where, sql)
- def UpdatePage(self, layer):
- # update data tlist
- if layer in self.layerPage.keys():
- tlist = self.FindWindowById(self.layerPage[layer]['data'])
- tlist.Update(self.dbMgrData['mapDBInfo'])
- class DbMgrTablesPage(DbMgrNotebookBase):
- def __init__(self, parent, parentDbMgrBase, onlyLayer = -1):
- """!Page for managing tables
- @param parent GUI parent
- @param parentDbMgrBase instance of DbMgrBase class
- @param onlyLayer create only tab of given layer, if -1 creates tabs of all layers
- """
- DbMgrNotebookBase.__init__(self, parent = parent,
- parentDbMgrBase = parentDbMgrBase)
- for layer in self.dbMgrData['mapDBInfo'].layers.keys():
- if onlyLayer > 0 and layer != onlyLayer:
- continue
- self.AddLayer(layer)
- if self.layers:
- self.SetSelection(0) # select first layer
- self.selLayer = self.layers[0]
- def AddLayer(self, layer, pos = -1):
- """!Adds tab which represents table
- @param layer vector map layer connected to table
- @param pos position of tab, if -1 it is added to end
- @return True if layer was added
- @return False if layer was not added - layer has been already added or does not exist
- """
- if layer in self.layers or \
- layer not in self.parentDbMgrBase.GetVectorLayers():
- return False
- self.layers.append(layer)
- self.layerPage[layer] = {}
- panel = wx.Panel(parent = self, id = wx.ID_ANY)
- self.layerPage[layer]['tablePage'] = panel.GetId()
- label = _("Table")
- if not self.dbMgrData['editable']:
- label += _(" (readonly)")
- if pos == -1:
- pos = self.GetPageCount()
- self.InsertPage(pos, page = panel,
- text = " %d / %s %s" % (layer, label,
- self.dbMgrData['mapDBInfo'].layers[layer]['table']))
-
- pageSizer = wx.BoxSizer(wx.VERTICAL)
-
- #
- # dbInfo
- #
- dbBox = wx.StaticBox(parent = panel, id = wx.ID_ANY,
- label = " %s " % _("Database connection"))
- dbSizer = wx.StaticBoxSizer(dbBox, wx.VERTICAL)
- dbSizer.Add(item = CreateDbInfoDesc(panel, self.dbMgrData['mapDBInfo'], layer),
- proportion = 1,
- flag = wx.EXPAND | wx.ALL,
- border = 3)
-
- #
- # table description
- #
- table = self.dbMgrData['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)
-
- tlist = self._createTableDesc(panel, table)
- tlist.Bind(wx.EVT_COMMAND_RIGHT_CLICK, self.OnTableRightUp) #wxMSW
- tlist.Bind(wx.EVT_RIGHT_UP, self.OnTableRightUp) #wxGTK
- self.layerPage[layer]['tableData'] = tlist.GetId()
-
- # manage columns (add)
- addBox = wx.StaticBox(parent = panel, id = wx.ID_ANY,
- label = " %s " % _("Add column"))
- addSizer = wx.StaticBoxSizer(addBox, wx.HORIZONTAL)
-
- 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 = wx.StaticText(parent = panel, id = wx.ID_ANY, label = _("Column")),
- flag = wx.ALIGN_CENTER_VERTICAL | wx.LEFT | wx.RIGHT,
- border = 5)
- addSizer.Add(item = column, proportion = 1,
- flag = wx.ALIGN_CENTER_VERTICAL | wx.LEFT | wx.RIGHT,
- border = 5)
-
- ctype = wx.Choice (parent = panel, id = wx.ID_ANY,
- choices = ["integer",
- "double",
- "varchar",
- "date"]) # FIXME
- ctype.SetSelection(0)
- ctype.Bind(wx.EVT_CHOICE, self.OnTableChangeType)
- self.layerPage[layer]['addColType'] = ctype.GetId()
- addSizer.Add(item = wx.StaticText(parent = panel, id = wx.ID_ANY, label = _("Type")),
- flag = wx.ALIGN_CENTER_VERTICAL | wx.LEFT | wx.RIGHT,
- border = 5)
- addSizer.Add(item = ctype,
- flag = wx.ALIGN_CENTER_VERTICAL | wx.LEFT | wx.RIGHT,
- border = 5)
-
- 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()
- addSizer.Add(item = wx.StaticText(parent = panel, id = wx.ID_ANY, label = _("Length")),
- flag = wx.ALIGN_CENTER_VERTICAL | wx.LEFT | wx.RIGHT,
- border = 5)
- addSizer.Add(item = length,
- flag = wx.ALIGN_CENTER_VERTICAL | wx.LEFT | wx.RIGHT,
- border = 5)
-
- 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, flag = wx.ALL | wx.ALIGN_RIGHT | wx.EXPAND,
- border = 3)
-
- # manage columns (rename)
- renameBox = wx.StaticBox(parent = panel, id = wx.ID_ANY,
- label = " %s " % _("Rename column"))
- renameSizer = wx.StaticBoxSizer(renameBox, wx.HORIZONTAL)
-
- columnFrom = wx.ComboBox(parent = panel, id = wx.ID_ANY, size = (150, -1),
- style = wx.CB_SIMPLE | wx.CB_READONLY,
- choices = self.dbMgrData['mapDBInfo'].GetColumns(table))
- columnFrom.SetSelection(0)
- self.layerPage[layer]['renameCol'] = columnFrom.GetId()
- renameSizer.Add(item = wx.StaticText(parent = panel, id = wx.ID_ANY, label = _("Column")),
- flag = wx.ALIGN_CENTER_VERTICAL | wx.LEFT | wx.RIGHT,
- border = 5)
- renameSizer.Add(item = columnFrom, proportion = 1,
- flag = wx.ALIGN_CENTER_VERTICAL | wx.LEFT | wx.RIGHT,
- border = 5)
-
- 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()
- renameSizer.Add(item = wx.StaticText(parent = panel, id = wx.ID_ANY, label = _("To")),
- flag = wx.ALIGN_CENTER_VERTICAL | wx.LEFT | wx.RIGHT,
- border = 5)
- renameSizer.Add(item = columnTo, proportion = 1,
- flag = wx.ALIGN_CENTER_VERTICAL | wx.LEFT | wx.RIGHT,
- border = 5)
-
- 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()
- renameSizer.Add(item = btnRenameCol, flag = wx.ALL | wx.ALIGN_RIGHT | wx.EXPAND,
- border = 3)
-
- tableSizer.Add(item = tlist,
- flag = wx.ALL | wx.EXPAND,
- proportion = 1,
- border = 3)
-
- pageSizer.Add(item=dbSizer,
- 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 = 3)
-
- pageSizer.Add(item = addSizer,
- flag = wx.LEFT | wx.RIGHT | wx.BOTTOM | wx.EXPAND,
- proportion = 0,
- border = 3)
- pageSizer.Add(item = renameSizer,
- flag = wx.LEFT | wx.RIGHT | wx.BOTTOM | wx.EXPAND,
- proportion = 0,
- border = 3)
-
- panel.SetSizer(pageSizer)
-
- if not self.dbMgrData['editable']:
- for widget in [columnTo, columnFrom, length, ctype,
- column, btnAddCol, btnRenameCol]:
- widget.Enable(False)
- return True
- def _createTableDesc(self, parent, table):
- """!Create list with table description"""
- tlist = TableListCtrl(parent = parent, id = wx.ID_ANY,
- table = self.dbMgrData['mapDBInfo'].tables[table],
- columns = self.dbMgrData['mapDBInfo'].GetColumns(table))
- tlist.Populate()
- # sorter
- # itemDataMap = list.Populate()
- # listmix.ColumnSorterMixin.__init__(self, 2)
- return tlist
- def OnTableChangeType(self, event):
- """!Data type for new column changed. Enable or disable
- data length widget"""
- win = self.FindWindowById(self.layerPage[self.selLayer]['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.selLayer]['renameColButton'])
- col = self.FindWindowById(self.layerPage[self.selLayer]['renameCol'])
- colTo = self.FindWindowById(self.layerPage[self.selLayer]['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.selLayer]['addColButton'])
- if len(event.GetString()) > 0:
- btn.Enable(True)
- else:
- btn.Enable(False)
- event.Skip()
- def OnTableItemChange(self, event):
- """!Rename column in the table"""
- tlist = self.FindWindowById(self.layerPage[self.selLayer]['tableData'])
- name = self.FindWindowById(self.layerPage[self.selLayer]['renameCol']).GetValue()
- nameTo = self.FindWindowById(self.layerPage[self.selLayer]['renameColTo']).GetValue()
- table = self.dbMgrData['mapDBInfo'].layers[self.selLayer]["table"]
- if not name or not nameTo:
- GError(parent = self,
- message = _("Unable to rename column. "
- "No column name defined."))
- return
- else:
- item = tlist.FindItem(start = -1, str = name)
- if item > -1:
- if tlist.FindItem(start = -1, str = nameTo) > -1:
- GError(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})
- return
- else:
- tlist.SetItemText(item, nameTo)
- self.listOfCommands.append(('v.db.renamecolumn',
- { 'map' : self.dbMgrData['vectName'],
- 'layer' : self.selLayer,
- 'column' : '%s,%s' % (name, nameTo) }
- ))
- else:
- GError(parent = self,
- message = _("Unable to rename column. "
- "Column <%(column)s> doesn't exist in the table <%(table)s>.") %
- {'column' : name, 'table' : table})
- return
-
- # apply changes
- self.ApplyCommands(self.listOfCommands, self.listOfSQLStatements)
- # update widgets
- self.FindWindowById(self.layerPage[self.selLayer]['renameCol']).SetItems(self.dbMgrData['mapDBInfo'].GetColumns(table))
- self.FindWindowById(self.layerPage[self.selLayer]['renameCol']).SetSelection(0)
- self.FindWindowById(self.layerPage[self.selLayer]['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.selLayer]['tableData']).GetFirstSelected() == -1:
- menu.Enable(self.popupTableID1, False)
- menu.Append(self.popupTableID2, _("Drop all columns"))
- menu.AppendSeparator()
- menu.Append(self.popupTableID3, _("Reload"))
-
- if not self.dbMgrData['editable']:
- menu.Enable(self.popupTableID1, False)
- menu.Enable(self.popupTableID2, False)
- self.PopupMenu(menu)
- menu.Destroy()
- def OnTableItemDelete(self, event):
- """!Delete selected item(s) from the list"""
- tlist = self.FindWindowById(self.layerPage[self.selLayer]['tableData'])
-
- item = tlist.GetFirstSelected()
- countSelected = tlist.GetSelectedItemCount()
- if UserSettings.Get(group = 'atm', key = 'askOnDeleteRec', subkey = 'enabled'):
- # if the user select more columns to delete, all the columns name
- # will appear the the warning dialog
- if tlist.GetSelectedItemCount() > 1:
- deleteColumns = "columns '%s'" % tlist.GetItemText(item)
- while item != -1:
- item = tlist.GetNextSelected(item)
- if item != -1:
- deleteColumns += ", '%s'" % tlist.GetItemText(item)
- else:
- deleteColumns = "column '%s'" % tlist.GetItemText(item)
- deleteDialog = wx.MessageBox(parent = self,
- message = _("Selected column %s will PERMANENTLY removed "
- "from table. Do you want to drop the column?") % \
- (deleteColumns),
- caption = _("Drop column(s)"),
- style = wx.YES_NO | wx.CENTRE)
- if deleteDialog != wx.YES:
- return False
- item = tlist.GetFirstSelected()
- while item != -1:
- self.listOfCommands.append(('v.db.dropcolumn',
- { 'map' : self.dbMgrData['vectName'],
- 'layer' : self.selLayer,
- 'column' : tlist.GetItemText(item) }
- ))
- tlist.DeleteItem(item)
- item = tlist.GetFirstSelected()
-
- # apply changes
- self.ApplyCommands(self.listOfCommands, self.listOfSQLStatements)
-
- # update widgets
- table = self.dbMgrData['mapDBInfo'].layers[self.selLayer]['table']
- self.FindWindowById(self.layerPage[self.selLayer]['renameCol']).SetItems(self.dbMgrData['mapDBInfo'].GetColumns(table))
- self.FindWindowById(self.layerPage[self.selLayer]['renameCol']).SetSelection(0)
-
- event.Skip()
- def OnTableItemDeleteAll(self, event):
- """!Delete all items from the list"""
- table = self.dbMgrData['mapDBInfo'].layers[self.selLayer]['table']
- cols = self.dbMgrData['mapDBInfo'].GetColumns(table)
- keyColumn = self.dbMgrData['mapDBInfo'].layers[self.selLayer]['key']
- if keyColumn in cols:
- cols.remove(keyColumn)
-
- if UserSettings.Get(group = 'atm', key = 'askOnDeleteRec', subkey = 'enabled'):
- deleteDialog = wx.MessageBox(parent = self,
- message = _("Selected columns\n%s\nwill PERMANENTLY removed "
- "from table. Do you want to drop the columns?") % \
- ('\n'.join(cols)),
- caption = _("Drop column(s)"),
- style = wx.YES_NO | wx.CENTRE)
- if deleteDialog != wx.YES:
- return False
-
- for col in cols:
- self.listOfCommands.append(('v.db.dropcolumn',
- { 'map' : self.dbMgrData['vectName'],
- 'layer' : self.selLayer,
- 'column' : col }
- ))
- self.FindWindowById(self.layerPage[self.selLayer]['tableData']).DeleteAllItems()
- # apply changes
- self.ApplyCommands(self.listOfCommands, self.listOfSQLStatements)
- # update widgets
- table = self.dbMgrData['mapDBInfo'].layers[self.selLayer]['table']
- self.FindWindowById(self.layerPage[self.selLayer]['renameCol']).SetItems(self.dbMgrData['mapDBInfo'].GetColumns(table))
- self.FindWindowById(self.layerPage[self.selLayer]['renameCol']).SetSelection(0)
- event.Skip()
- def OnTableReload(self, event = None):
- """!Reload table description"""
- self.FindWindowById(self.layerPage[self.selLayer]['tableData']).Populate(update = True)
- self.listOfCommands = []
- def OnTableItemAdd(self, event):
- """!Add new column to the table"""
- name = self.FindWindowById(self.layerPage[self.selLayer]['addColName']).GetValue()
-
- ctype = self.FindWindowById(self.layerPage[self.selLayer]['addColType']). \
- GetStringSelection()
-
- length = int(self.FindWindowById(self.layerPage[self.selLayer]['addColLength']). \
- GetValue())
-
- # add item to the list of table columns
- tlist = self.FindWindowById(self.layerPage[self.selLayer]['tableData'])
- index = tlist.InsertStringItem(sys.maxint, str(name))
- tlist.SetStringItem(index, 0, str(name))
- tlist.SetStringItem(index, 1, str(ctype))
- tlist.SetStringItem(index, 2, str(length))
-
- self.AddColumn(name, ctype, length)
- # update widgets
- table = self.dbMgrData['mapDBInfo'].layers[self.selLayer]['table']
- self.FindWindowById(self.layerPage[self.selLayer]['addColName']).SetValue('')
- self.FindWindowById(self.layerPage[self.selLayer]['renameCol']).SetItems(self.dbMgrData['mapDBInfo'].GetColumns(table))
- self.FindWindowById(self.layerPage[self.selLayer]['renameCol']).SetSelection(0)
-
- event.Skip()
- def UpdatePage(self, layer):
- if layer in self.layerPage.keys():
- table = self.dbMgrData['mapDBInfo'].layers[layer]['table']
- # update table description
- tlist = self.FindWindowById(self.layerPage[layer]['tableData'])
- tlist.Update(table = self.dbMgrData['mapDBInfo'].tables[table],
- columns = self.dbMgrData['mapDBInfo'].GetColumns(table))
- self.OnTableReload(None)
- class DbMgrLayersPage(wx.Panel):
- def __init__(self, parent, parentDbMgrBase):
- """!Create layer manage page"""
- self.parentDbMgrBase = parentDbMgrBase
- self.dbMgrData = self.parentDbMgrBase.dbMgrData
- wx.Panel.__init__(self, parent = parent)
- splitterWin = wx.SplitterWindow(parent = self, id = wx.ID_ANY)
- splitterWin.SetMinimumPaneSize(100)
-
- #
- # 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)
- if not self.dbMgrData['editable']:
- self.manageLayerBook.Enable(False)
- manageSizer.Add(item = self.manageLayerBook,
- proportion = 1,
- flag = wx.LEFT | wx.RIGHT | wx.BOTTOM | wx.EXPAND,
- border = 5)
- panelSizer = wx.BoxSizer(wx.VERTICAL)
- panelSizer.Add(item = splitterWin,
- proportion = 1,
- flag = wx.EXPAND)
- panelManage.SetSizer(manageSizer)
- splitterWin.SplitHorizontally(panelList, panelManage, 100)
- splitterWin.Fit()
- self.SetSizer(panelSizer)
- def _createLayerDesc(self, parent):
- """!Create list of linked layers"""
- tlist = LayerListCtrl(parent = parent, id = wx.ID_ANY,
- layers = self.dbMgrData['mapDBInfo'].layers)
-
- tlist.Populate()
- # sorter
- # itemDataMap = list.Populate()
- # listmix.ColumnSorterMixin.__init__(self, 2)
- return tlist
-
- def UpdatePage(self):
- #
- # 'manage layers' page
- #
- # update list of layers
- #self.dbMgrData['mapDBInfo'] = VectorDBInfo(self.dbMgrData['vectName'])
- self.layerList.Update(self.dbMgrData['mapDBInfo'].layers)
- self.layerList.Populate(update = True)
- # update selected widgets
- listOfLayers = map(str, self.dbMgrData['mapDBInfo'].layers.keys())
- ### delete layer page
- self.manageLayerBook.deleteLayer.SetItems(listOfLayers)
- if len(listOfLayers) > 0:
- self.manageLayerBook.deleteLayer.SetStringSelection(listOfLayers[0])
- tableName = self.dbMgrData['mapDBInfo'].layers[int(listOfLayers[0])]['table']
- maxLayer = max(self.dbMgrData['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 OnLayerRightUp(self, event):
- """!Layer description area, context menu"""
- pass
- 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)
- i += 1
- self.SetColumnWidth(col = 0, width = 350)
- self.SetColumnWidth(col = 1, width = 175)
- 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.dbMgrData['mapDBInfo']
- #
- # drivers
- #
- drivers = RunCommand('db.drivers',
- quiet = True,
- read = True,
- flags = 'p')
-
- self.listOfDrivers = []
- for drv in drivers.splitlines():
- self.listOfDrivers.append(drv.strip())
-
- #
- # get default values
- #
- self.defaultConnect = {}
- connect = RunCommand('db.connect',
- flags = 'p',
- read = True,
- quiet = True)
-
- for line in connect.splitlines():
- item, value = line.split(':', 1)
- self.defaultConnect[item.strip()] = value.strip()
-
- if len(self.defaultConnect['driver']) == 0 or \
- len(self.defaultConnect['database']) == 0:
- GWarning(parent = self.parent,
- message = _("Unknown default DB connection. "
- "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
- # layer description
-
- layerBox = wx.StaticBox (parent = self.addPanel, id = wx.ID_ANY,
- label = " %s " % (_("Layer description")))
- layerSizer = wx.StaticBoxSizer(layerBox, wx.VERTICAL)
-
- #
- # 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)),
- 'addCat':
- (wx.CheckBox(parent = self.addPanel, id = wx.ID_ANY,
- label = _("Insert record for each category into table")),
- None),
- }
-
- # 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)
-
- # tooltips
- self.addLayerWidgets['addCat'][0].SetToolTipString(_("You need to add categories "
- "by v.category module."))
- # table description
- tableBox = wx.StaticBox (parent = self.addPanel, id = wx.ID_ANY,
- label = " %s " % (_("Table description")))
- tableSizer = wx.StaticBoxSizer(tableBox, wx.VERTICAL)
- #
- # list of table widgets
- #
- keyCol = UserSettings.Get(group = 'atm', key = 'keycolumn', subkey = 'value')
- 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 = keyCol,
- 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)
-
- # data area
- dataSizer = wx.GridBagSizer(hgap = 5, vgap = 5)
- row = 0
- for key in ('layer', 'driver', 'database', 'table', 'key', 'addCat'):
- label, value = self.addLayerWidgets[key]
- if not value:
- span = (1, 2)
- else:
- span = (1, 1)
- dataSizer.Add(item = label,
- flag = wx.ALIGN_CENTER_VERTICAL, pos = (row, 0),
- span = span)
-
- if not value:
- row += 1
- continue
- if key == 'layer':
- style = wx.ALIGN_CENTER_VERTICAL | wx.ALIGN_LEFT
- else:
- style = wx.ALIGN_CENTER_VERTICAL | wx.EXPAND
-
- dataSizer.Add(item = value,
- flag = style, pos = (row, 1))
-
- row += 1
-
- dataSizer.AddGrowableCol(1)
- 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)
-
- # 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 = 3,
- flag = wx.ALL | wx.EXPAND,
- border = 3)
-
- pageSizer.Add(item = tableSizer,
- proportion = 2,
- flag = wx.TOP | wx.BOTTOM | wx.RIGHT | wx.EXPAND,
- border = 3)
-
- layerSizer.SetVirtualSizeHints(self.addPanel)
- self.addPanel.SetAutoLayout(True)
- self.addPanel.SetSizer(pageSizer)
- pageSizer.Fit(self.addPanel)
-
- def _createDeletePage(self):
- """!Delete layer"""
- self.deletePanel = wx.Panel(parent = self, id = wx.ID_ANY)
- self.AddPage(page = self.deletePanel, text = _("Remove layer"))
- label = wx.StaticText(parent = self.deletePanel, id = wx.ID_ANY,
- label = '%s:' % _("Layer to remove"))
- 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, _("&Remove 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.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 = 1)
- dataSizer.Add(item = self.deleteTable,
- proportion = 0,
- flag = wx.ALL | wx.EXPAND,
- border = 1)
- 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 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 key == '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 = []
- ret = RunCommand('db.tables',
- parent = self,
- read = True,
- flags = 'p',
- driver = driver,
- database = database)
-
- if ret is None:
- GError(parent = self,
- message = _("Unable to get list of tables.\n"
- "Please use db.connect to set database parameters."))
-
- return tables
-
- for table in ret.splitlines():
- tables.append(table)
-
- return tables
- def _getColumns(self, driver, database, table):
- """!Get list of column of given table"""
- columns = []
- ret = RunCommand('db.columns',
- parent = self,
- quiet = True,
- read = True,
- driver = driver,
- database = database,
- table = table)
-
- if ret == None:
- return columns
-
- for column in ret.splitlines():
- 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:
- GError(parent = self,
- message = _("Unable to create new table. "
- "Table name or key column name is missing."))
- return
- if table in self.addLayerWidgets['table'][1].GetItems():
- GError(parent = self,
- message = _("Unable to create new table. "
- "Table <%s> already exists in the database.") % table)
- return
-
- # create table
- sql = 'CREATE TABLE %s (%s INTEGER)' % (table, key)
- RunCommand('db.execute',
- quiet = True,
- parent = self,
- stdin = sql,
- input = '-',
- driver = driver,
- database = database)
-
- # 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():
- GError(parent = self,
- message = _("Unable to add new layer to vector map <%(vector)s>. "
- "Layer %(layer)d already exists.") % \
- {'vector' : self.mapDBInfo.map, 'layer' : layer})
- return
- # add new layer
- ret = RunCommand('v.db.connect',
- parent = self,
- quiet = True,
- map = self.mapDBInfo.map,
- driver = driver,
- database = database,
- table = table,
- key = key,
- layer = layer)
-
- # insert records into table if required
- if self.addLayerWidgets['addCat'][0].IsChecked():
- RunCommand('v.to.db',
- parent = self,
- quiet = True,
- map = self.mapDBInfo.map,
- layer = layer,
- qlayer = layer,
- option = 'cat',
- columns = key)
- if ret == 0:
- # update dialog (only for new layer)
- self.parentDialog.parentDbMgrBase.UpdateDialog(layer = layer)
- # update db info
- self.mapDBInfo = self.parentDialog.dbMgrData['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
- RunCommand('v.db.connect',
- parent = self,
- flags = 'd',
- map = self.mapDBInfo.map,
- layer = 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)
- RunCommand('db.execute',
- parent = self,
- stdin = sql,
- quiet = True,
- driver = driver,
- database = database)
-
- # update list of tables
- tableList = self.addLayerWidgets['table'][1]
- tableList.SetItems(self._getTables(driver, database))
- tableList.SetStringSelection(table)
-
- # update dialog
- self.parentDialog.parentDbMgrBase.UpdateDialog(layer = layer)
- # update db info
- self.mapDBInfo = self.parentDialog.dbMgrData['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
- RunCommand('v.db.connect',
- parent = self,
- quiet = True,
- flags = 'd',
- map = self.mapDBInfo.map,
- layer = layer)
- # add modified layer
- RunCommand('v.db.connect',
- quiet = True,
- map = self.mapDBInfo.map,
- driver = self.modifyLayerWidgets['driver'][1].GetStringSelection(),
- database = self.modifyLayerWidgets['database'][1].GetValue(),
- table = self.modifyLayerWidgets['table'][1].GetStringSelection(),
- key = self.modifyLayerWidgets['key'][1].GetStringSelection(),
- layer = int(layer))
-
- # update dialog (only for new layer)
- self.parentDialog.parentDbMgrBase.UpdateDialog(layer = layer)
- # update db info
- self.mapDBInfo = self.parentDialog.dbMgrData['mapDBInfo']
- event.Skip()
|