1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782278327842785278627872788278927902791279227932794279527962797279827992800280128022803280428052806280728082809281028112812281328142815281628172818281928202821282228232824282528262827282828292830283128322833283428352836283728382839284028412842284328442845284628472848284928502851285228532854285528562857285828592860286128622863286428652866286728682869287028712872287328742875287628772878287928802881288228832884288528862887288828892890289128922893289428952896289728982899290029012902290329042905290629072908290929102911291229132914291529162917291829192920292129222923292429252926292729282929293029312932293329342935293629372938293929402941294229432944294529462947294829492950295129522953295429552956295729582959296029612962296329642965296629672968296929702971297229732974297529762977297829792980298129822983298429852986298729882989299029912992299329942995299629972998299930003001300230033004300530063007300830093010301130123013301430153016301730183019302030213022302330243025302630273028302930303031303230333034303530363037303830393040304130423043304430453046304730483049305030513052305330543055305630573058305930603061306230633064306530663067306830693070307130723073307430753076307730783079308030813082308330843085308630873088308930903091309230933094309530963097309830993100310131023103310431053106310731083109311031113112311331143115311631173118311931203121312231233124312531263127312831293130313131323133313431353136313731383139314031413142314331443145314631473148314931503151315231533154315531563157315831593160316131623163316431653166316731683169317031713172317331743175317631773178317931803181318231833184318531863187318831893190319131923193319431953196319731983199320032013202320332043205320632073208320932103211321232133214321532163217321832193220322132223223322432253226322732283229323032313232323332343235323632373238323932403241324232433244324532463247324832493250325132523253325432553256325732583259326032613262326332643265326632673268326932703271327232733274327532763277327832793280328132823283328432853286 |
- """!
- @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__":
- sys.path.append(os.path.join(os.getenv('GISBASE'), 'etc', 'gui', 'wxpython'))
- 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()
-
- 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):
- """!Update list according new mapDBInfo description"""
- self.mapDBInfo = mapDBInfo
- self.LoadData(self.layer)
- 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
-
- fs = UserSettings.Get(group = 'atm', key = 'fieldSeparator', subkey = 'value')
- # read data
- # FIXME: Max. number of rows, while the GUI is still usable
- # stdout can be very large, do not use PIPE, redirect to temp file
- # TODO: more effective way should be implemented...
- outFile = tempfile.NamedTemporaryFile(mode = 'w+b')
- cmdParams = dict(quiet = True,
- parent = self,
- flags = 'c')
- if sys.platform != "win32":
- cmdParams.update(dict(sep = fs))
- if sql:
- cmdParams.update(dict(sql = sql,
- output = outFile.name,
- overwrite = True))
- ret = RunCommand('db.select',
- **cmdParams)
- else:
- cmdParams.update(dict(map = self.mapDBInfo.map,
- layer = layer,
- where = where,
- stdout = outFile))
- 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>. "
- "Try to change field separator in GUI Settings, "
- "Attributes tab, Data browser section.") % \
- {'table' : tableName })
- self.columns = {} # because of IsEmpty method
- return
- self.AddDataRow(i, record, columns, keyId)
- i += 1
- if i >= 100000:
- self.log.write(_("Limit 100000 records."))
- break
-
- self.SetItemCount(i)
-
- i = 0
- for col in columns:
- width = self.columns[col]['length'] * 6 # FIXME
- if width < 60:
- width = 60
- if width > 300:
- width = 300
- self.SetColumnWidth(col = i, width = width)
- i += 1
-
- self.SendSizeEvent()
-
- 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, log = 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.cmdLog = log # self.parent.goutput
- # 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.cmdLog = self.parentDbMgrBase.cmdLog
- 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:
- sqlFile = tempfile.NamedTemporaryFile(mode = "wt")
- 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.file.write(sql.encode(enc) + ';')
- else:
- sqlFile.file.write(sql + ';')
- sqlFile.file.write(os.linesep)
- sqlFile.file.flush()
- 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 = sqlFile.name,
- driver = driver,
- database = database)
-
- # 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)
- # attribute data
- sqlBox = wx.StaticBox(parent = panel, 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 = panel, 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)
- pageSizer.Add(item = sqlSizer,
- 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'] (value)
- else: # -> string
- tlist.itemDataMap[item][idx] = values[i]
- except:
- 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(self.dbMgrData['mapDBInfo'])
-
- 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(float = True)
- 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'),
- log = self.cmdLog,
- 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()
-
- if UserSettings.Get(group = 'atm', key = 'askOnDeleteRec', subkey = 'enabled'):
- deleteDialog = wx.MessageBox(parent = self,
- message = _("Selected column '%s' will PERMANENTLY removed "
- "from table. Do you want to drop the column?") % \
- (tlist.GetItemText(item)),
- caption = _("Drop column(s)"),
- style = wx.YES_NO | wx.CENTRE)
- if deleteDialog != wx.YES:
- return False
-
- 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.AddGrowableCol(2)
- flexSizer.Add(item = label,
- flag = wx.ALIGN_CENTER_VERTICAL)
- flexSizer.Add(item = self.deleteLayer,
- flag = wx.ALIGN_CENTER_VERTICAL)
- dataSizer.Add(item = flexSizer,
- proportion = 0,
- flag = wx.ALL | wx.EXPAND,
- border = 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()
|