base.py 38 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049
  1. """!@package grass.temporal
  2. @brief GRASS Python scripting module (temporal GIS functions)
  3. Temporal GIS base classes to be used in other
  4. Python temporal gis packages.
  5. This packages includes all base classes to store basic information
  6. like id, name, mapset creation and modification time as well as sql
  7. serialization and de-serialization and the sql database interface.
  8. Usage:
  9. @code
  10. >>> import grass.temporal as tgis
  11. >>> tgis.init()
  12. >>> rbase = tgis.RasterBase(ident="soil@PERMANENT")
  13. >>> vbase = tgis.VectorBase(ident="soil:1@PERMANENT")
  14. >>> r3base = tgis.Raster3DBase(ident="soil@PERMANENT")
  15. >>> strdsbase = tgis.STRDSBase(ident="soil@PERMANENT")
  16. >>> stvdsbase = tgis.STVDSBase(ident="soil@PERMANENT")
  17. >>> str3dsbase = tgis.STR3DSBase(ident="soil@PERMANENT")
  18. @endcode
  19. (C) 2011-2013 by the GRASS Development Team
  20. This program is free software under the GNU General Public
  21. License (>=v2). Read the file COPYING that comes with GRASS
  22. for details.
  23. @author Soeren Gebbert
  24. """
  25. from datetime import datetime, date, time, timedelta
  26. from core import *
  27. ###############################################################################
  28. class DictSQLSerializer(object):
  29. def __init__(self):
  30. self.D = {}
  31. def serialize(self, type, table, where=None):
  32. """!Convert the internal dictionary into a string of semicolon
  33. separated SQL statements The keys are the column names and
  34. the values are the row entries
  35. Usage:
  36. \code
  37. >>> init()
  38. >>> t = DictSQLSerializer()
  39. >>> t.D["id"] = "soil@PERMANENT"
  40. >>> t.D["name"] = "soil"
  41. >>> t.D["mapset"] = "PERMANENT"
  42. >>> t.D["creator"] = "soeren"
  43. >>> t.D["creation_time"] = datetime(2001,1,1)
  44. >>> t.D["modification_time"] = datetime(2001,1,1)
  45. >>> t.serialize(type="SELECT", table="raster_base")
  46. ('SELECT name , creator , creation_time , modification_time , mapset , id FROM raster_base ;\\n', ())
  47. >>> t.serialize(type="INSERT", table="raster_base")
  48. ('INSERT INTO raster_base ( name ,creator ,creation_time ,modification_time ,mapset ,id ) VALUES (? ,? ,? ,? ,? ,?) ;\\n', ('soil', 'soeren', datetime.datetime(2001, 1, 1, 0, 0), datetime.datetime(2001, 1, 1, 0, 0), 'PERMANENT', 'soil@PERMANENT'))
  49. >>> t.serialize(type="UPDATE", table="raster_base")
  50. ('UPDATE raster_base SET name = ? ,creator = ? ,creation_time = ? ,modification_time = ? ,mapset = ? ,id = ? ;\\n', ('soil', 'soeren', datetime.datetime(2001, 1, 1, 0, 0), datetime.datetime(2001, 1, 1, 0, 0), 'PERMANENT', 'soil@PERMANENT'))
  51. >>> t.serialize(type="UPDATE ALL", table="raster_base")
  52. ('UPDATE raster_base SET name = ? ,creator = ? ,creation_time = ? ,modification_time = ? ,mapset = ? ,id = ? ;\\n', ('soil', 'soeren', datetime.datetime(2001, 1, 1, 0, 0), datetime.datetime(2001, 1, 1, 0, 0), 'PERMANENT', 'soil@PERMANENT'))
  53. @param type must be SELECT. INSERT, UPDATE
  54. @param table The name of the table to select, insert or update
  55. @param where The optional where statement
  56. @return a tuple containing the SQL string and the arguments
  57. \endcode
  58. """
  59. dbif = SQLDatabaseInterfaceConnection()
  60. sql = ""
  61. args = []
  62. # Create ordered select statement
  63. if type == "SELECT":
  64. sql += 'SELECT '
  65. count = 0
  66. for key in self.D.keys():
  67. if count == 0:
  68. sql += ' %s ' % key
  69. else:
  70. sql += ' , %s ' % key
  71. count += 1
  72. sql += ' FROM ' + table + ' '
  73. if where:
  74. sql += where
  75. sql += ";\n"
  76. # Create insert statement
  77. if type == "INSERT":
  78. count = 0
  79. sql += 'INSERT INTO ' + table + ' ('
  80. for key in self.D.keys():
  81. if count == 0:
  82. sql += ' %s ' % key
  83. else:
  84. sql += ' ,%s ' % key
  85. count += 1
  86. count = 0
  87. sql += ') VALUES ('
  88. for key in self.D.keys():
  89. if count == 0:
  90. if dbif.dbmi.paramstyle == "qmark":
  91. sql += '?'
  92. else:
  93. sql += '%s'
  94. else:
  95. if dbif.dbmi.paramstyle == "qmark":
  96. sql += ' ,?'
  97. else:
  98. sql += ' ,%s'
  99. count += 1
  100. args.append(self.D[key])
  101. sql += ') '
  102. if where:
  103. sql += where
  104. sql += ";\n"
  105. # Create update statement for existing entries
  106. if type == "UPDATE":
  107. count = 0
  108. sql += 'UPDATE ' + table + ' SET '
  109. for key in self.D.keys():
  110. # Update only entries which are not None
  111. if self.D[key] is not None:
  112. if count == 0:
  113. if dbif.dbmi.paramstyle == "qmark":
  114. sql += ' %s = ? ' % key
  115. else:
  116. sql += ' %s ' % key
  117. sql += '= %s '
  118. else:
  119. if dbif.dbmi.paramstyle == "qmark":
  120. sql += ' ,%s = ? ' % key
  121. else:
  122. sql += ' ,%s ' % key
  123. sql += '= %s '
  124. count += 1
  125. args.append(self.D[key])
  126. if where:
  127. sql += where
  128. sql += ";\n"
  129. # Create update statement for all entries
  130. if type == "UPDATE ALL":
  131. count = 0
  132. sql += 'UPDATE ' + table + ' SET '
  133. for key in self.D.keys():
  134. if count == 0:
  135. if dbif.dbmi.paramstyle == "qmark":
  136. sql += ' %s = ? ' % key
  137. else:
  138. sql += ' %s ' % key
  139. sql += '= %s '
  140. else:
  141. if dbif.dbmi.paramstyle == "qmark":
  142. sql += ' ,%s = ? ' % key
  143. else:
  144. sql += ' ,%s ' % key
  145. sql += '= %s '
  146. count += 1
  147. args.append(self.D[key])
  148. if where:
  149. sql += where
  150. sql += ";\n"
  151. return sql, tuple(args)
  152. def deserialize(self, row):
  153. """!Convert the content of the dbmi dictionary like row into the
  154. internal dictionary
  155. @param row The dictionary like row to store in the internal dict
  156. """
  157. self.D = {}
  158. for key in row.keys():
  159. self.D[key] = row[key]
  160. def clear(self):
  161. """!Initialize the internal storage"""
  162. self.D = {}
  163. def print_self(self):
  164. """!Print the content of the internal dictionary to stdout
  165. """
  166. print self.D
  167. ###############################################################################
  168. class SQLDatabaseInterface(DictSQLSerializer):
  169. """!This class represents the SQL database interface
  170. Functions to insert, select and update the internal
  171. structure of this class in the temporal database are implemented.
  172. This is the base class for raster, raster3d, vector and
  173. space time datasets data management classes:
  174. - Identification information (base)
  175. - Spatial extent
  176. - Temporal extent
  177. - Metadata
  178. Usage:
  179. \code
  180. >>> init()
  181. >>> t = SQLDatabaseInterface("raster", "soil@PERMANENT")
  182. >>> t.D["name"] = "soil"
  183. >>> t.D["mapset"] = "PERMANENT"
  184. >>> t.D["creator"] = "soeren"
  185. >>> t.D["creation_time"] = datetime(2001,1,1)
  186. >>> t.get_delete_statement()
  187. "DELETE FROM raster WHERE id = 'soil@PERMANENT';\\n"
  188. >>> t.get_is_in_db_statement()
  189. "SELECT id FROM raster WHERE id = 'soil@PERMANENT';\\n"
  190. >>> t.get_select_statement()
  191. ("SELECT creation_time , mapset , name , creator FROM raster WHERE id = 'soil@PERMANENT';\\n", ())
  192. >>> t.get_select_statement_mogrified()
  193. "SELECT creation_time , mapset , name , creator FROM raster WHERE id = 'soil@PERMANENT';\\n"
  194. >>> t.get_insert_statement()
  195. ('INSERT INTO raster ( creation_time ,mapset ,name ,creator ) VALUES (? ,? ,? ,?) ;\\n', (datetime.datetime(2001, 1, 1, 0, 0), 'PERMANENT', 'soil', 'soeren'))
  196. >>> t.get_insert_statement_mogrified()
  197. "INSERT INTO raster ( creation_time ,mapset ,name ,creator ) VALUES ('2001-01-01 00:00:00' ,'PERMANENT' ,'soil' ,'soeren') ;\\n"
  198. >>> t.get_update_statement()
  199. ("UPDATE raster SET creation_time = ? ,mapset = ? ,name = ? ,creator = ? WHERE id = 'soil@PERMANENT';\\n", (datetime.datetime(2001, 1, 1, 0, 0), 'PERMANENT', 'soil', 'soeren'))
  200. >>> t.get_update_statement_mogrified()
  201. "UPDATE raster SET creation_time = '2001-01-01 00:00:00' ,mapset = 'PERMANENT' ,name = 'soil' ,creator = 'soeren' WHERE id = 'soil@PERMANENT';\\n"
  202. >>> t.get_update_all_statement()
  203. ("UPDATE raster SET creation_time = ? ,mapset = ? ,name = ? ,creator = ? WHERE id = 'soil@PERMANENT';\\n", (datetime.datetime(2001, 1, 1, 0, 0), 'PERMANENT', 'soil', 'soeren'))
  204. >>> t.get_update_all_statement_mogrified()
  205. "UPDATE raster SET creation_time = '2001-01-01 00:00:00' ,mapset = 'PERMANENT' ,name = 'soil' ,creator = 'soeren' WHERE id = 'soil@PERMANENT';\\n"
  206. \endcode
  207. """
  208. def __init__(self, table=None, ident=None):
  209. """!Constructor of this class
  210. @param table The name of the table
  211. @param ident The identifier (primary key) of this
  212. object in the database table
  213. """
  214. DictSQLSerializer.__init__(self)
  215. self.table = table # Name of the table, set in the subclass
  216. self.ident = ident
  217. self.msgr = get_tgis_message_interface()
  218. def get_table_name(self):
  219. """!Return the name of the table in which the internal
  220. data are inserted, updated or selected
  221. @return The name of the table
  222. """
  223. return self.table
  224. def get_delete_statement(self):
  225. """!Return the delete string
  226. @return The DELETE string
  227. """
  228. return "DELETE FROM " + self.get_table_name() + \
  229. " WHERE id = \'" + str(self.ident) + "\';\n"
  230. def delete(self, dbif=None):
  231. """!Delete the entry of this object from the temporal database
  232. @param dbif The database interface to be used,
  233. if None a temporary connection will be established
  234. """
  235. sql = self.get_delete_statement()
  236. #print sql
  237. if dbif:
  238. dbif.cursor.execute(sql)
  239. else:
  240. dbif = SQLDatabaseInterfaceConnection()
  241. dbif.connect()
  242. dbif.cursor.execute(sql)
  243. dbif.close()
  244. def get_is_in_db_statement(self):
  245. """!Return the selection string that checks if this object is registered in the
  246. temporal database
  247. @return The SELECT string
  248. """
  249. return "SELECT id FROM " + self.get_table_name() + \
  250. " WHERE id = \'" + str(self.ident) + "\';\n"
  251. def is_in_db(self, dbif=None):
  252. """!Check if this object is present in the temporal database
  253. @param dbif The database interface to be used,
  254. if None a temporary connection will be established
  255. @return True if this object is present in the temporal database, False otherwise
  256. """
  257. sql = self.get_is_in_db_statement()
  258. #print sql
  259. if dbif:
  260. dbif.cursor.execute(sql)
  261. row = dbif.cursor.fetchone()
  262. else:
  263. dbif = SQLDatabaseInterfaceConnection()
  264. dbif.connect()
  265. dbif.cursor.execute(sql)
  266. row = dbif.cursor.fetchone()
  267. dbif.close()
  268. # Nothing found
  269. if row is None:
  270. return False
  271. return True
  272. def get_select_statement(self):
  273. """!Return the sql statement and the argument list in
  274. database specific style
  275. @return The SELECT string
  276. """
  277. return self.serialize("SELECT", self.get_table_name(),
  278. "WHERE id = \'" + str(self.ident) + "\'")
  279. def get_select_statement_mogrified(self, dbif=None):
  280. """!Return the select statement as mogrified string
  281. @param dbif The database interface to be used,
  282. if None a temporary connection will be established
  283. @return The SELECT string
  284. """
  285. if not dbif:
  286. dbif = SQLDatabaseInterfaceConnection()
  287. return dbif.mogrify_sql_statement(self.get_select_statement())
  288. def select(self, dbif=None):
  289. """!Select the content from the temporal database and store it
  290. in the internal dictionary structure
  291. @param dbif The database interface to be used,
  292. if None a temporary connection will be established
  293. """
  294. sql, args = self.get_select_statement()
  295. #print sql
  296. #print args
  297. if dbif:
  298. if len(args) == 0:
  299. dbif.cursor.execute(sql)
  300. else:
  301. dbif.cursor.execute(sql, args)
  302. row = dbif.cursor.fetchone()
  303. else:
  304. dbif = SQLDatabaseInterfaceConnection()
  305. dbif.connect()
  306. if len(args) == 0:
  307. dbif.cursor.execute(sql)
  308. else:
  309. dbif.cursor.execute(sql, args)
  310. row = dbif.cursor.fetchone()
  311. dbif.close()
  312. # Nothing found
  313. if row is None:
  314. return False
  315. if len(row) > 0:
  316. self.deserialize(row)
  317. else:
  318. self.msgr.fatal(_("Object <%s> not found in the temporal database")
  319. % self.get_id())
  320. return True
  321. def get_insert_statement(self):
  322. """!Return the sql statement and the argument
  323. list in database specific style
  324. @return The INSERT string"""
  325. return self.serialize("INSERT", self.get_table_name())
  326. def get_insert_statement_mogrified(self, dbif=None):
  327. """!Return the insert statement as mogrified string
  328. @param dbif The database interface to be used,
  329. if None a temporary connection will be established
  330. @return The INSERT string
  331. """
  332. if not dbif:
  333. dbif = SQLDatabaseInterfaceConnection()
  334. return dbif.mogrify_sql_statement(self.get_insert_statement())
  335. def insert(self, dbif=None):
  336. """!Serialize the content of this object and store it in the temporal
  337. database using the internal identifier
  338. @param dbif The database interface to be used,
  339. if None a temporary connection will be established
  340. """
  341. sql, args = self.get_insert_statement()
  342. #print sql
  343. #print args
  344. if dbif:
  345. dbif.cursor.execute(sql, args)
  346. else:
  347. dbif = SQLDatabaseInterfaceConnection()
  348. dbif.connect()
  349. dbif.cursor.execute(sql, args)
  350. dbif.close()
  351. def get_update_statement(self, ident=None):
  352. """!Return the sql statement and the argument list
  353. in database specific style
  354. @param ident The identifier to be updated, useful for renaming
  355. @return The UPDATE string
  356. """
  357. if ident:
  358. return self.serialize("UPDATE", self.get_table_name(),
  359. "WHERE id = \'" + str(ident) + "\'")
  360. else:
  361. return self.serialize("UPDATE", self.get_table_name(),
  362. "WHERE id = \'" + str(self.ident) + "\'")
  363. def get_update_statement_mogrified(self, dbif=None, ident=None):
  364. """!Return the update statement as mogrified string
  365. @param dbif The database interface to be used,
  366. if None a temporary connection will be established
  367. @param ident The identifier to be updated, useful for renaming
  368. @return The UPDATE string
  369. """
  370. if not dbif:
  371. dbif = SQLDatabaseInterfaceConnection()
  372. return dbif.mogrify_sql_statement(self.get_update_statement(ident))
  373. def update(self, dbif=None, ident=None):
  374. """!Serialize the content of this object and update it in the temporal
  375. database using the internal identifier
  376. Only object entries which are exists (not None) are updated
  377. @param dbif The database interface to be used,
  378. if None a temporary connection will be established
  379. @param ident The identifier to be updated, useful for renaming
  380. """
  381. if self.ident is None:
  382. self.msgr.fatal(_("Missing identifer"))
  383. sql, args = self.get_update_statement(ident)
  384. #print sql
  385. #print args
  386. if dbif:
  387. dbif.cursor.execute(sql, args)
  388. else:
  389. dbif = SQLDatabaseInterfaceConnection()
  390. dbif.connect()
  391. dbif.cursor.execute(sql, args)
  392. dbif.close()
  393. def get_update_all_statement(self, ident=None):
  394. """!Return the sql statement and the argument
  395. list in database specific style
  396. @param ident The identifier to be updated, useful for renaming
  397. @return The UPDATE string
  398. """
  399. if ident:
  400. return self.serialize("UPDATE ALL", self.get_table_name(),
  401. "WHERE id = \'" + str(ident) + "\'")
  402. else:
  403. return self.serialize("UPDATE ALL", self.get_table_name(),
  404. "WHERE id = \'" + str(self.ident) + "\'")
  405. def get_update_all_statement_mogrified(self, dbif=None, ident=None):
  406. """!Return the update all statement as mogrified string
  407. @param dbif The database interface to be used,
  408. if None a temporary connection will be established
  409. @param ident The identifier to be updated, useful for renaming
  410. @return The UPDATE string
  411. """
  412. if not dbif:
  413. dbif = SQLDatabaseInterfaceConnection()
  414. return dbif.mogrify_sql_statement(self.get_update_all_statement(ident))
  415. def update_all(self, dbif=None, ident=None):
  416. """!Serialize the content of this object, including None objects,
  417. and update it in the temporal database using the internal identifier
  418. @param dbif The database interface to be used,
  419. if None a temporary connection will be established
  420. @param ident The identifier to be updated, useful for renaming
  421. """
  422. if self.ident is None:
  423. self.msgr.fatal(_("Missing identifer"))
  424. sql, args = self.get_update_all_statement(ident)
  425. #print sql
  426. #print args
  427. if dbif:
  428. dbif.cursor.execute(sql, args)
  429. else:
  430. dbif = SQLDatabaseInterfaceConnection()
  431. dbif.connect()
  432. dbif.cursor.execute(sql, args)
  433. dbif.close()
  434. ###############################################################################
  435. class DatasetBase(SQLDatabaseInterface):
  436. """!This is the base class for all maps and spacetime datasets storing
  437. basic identification information
  438. Usage:
  439. \code
  440. >>> init()
  441. >>> t = DatasetBase("raster", "soil@PERMANENT", creator="soeren", ctime=datetime(2001,1,1), ttype="absolute")
  442. >>> t.id
  443. 'soil@PERMANENT'
  444. >>> t.name
  445. 'soil'
  446. >>> t.mapset
  447. 'PERMANENT'
  448. >>> t.creator
  449. 'soeren'
  450. >>> t.ctime
  451. datetime.datetime(2001, 1, 1, 0, 0)
  452. >>> t.ttype
  453. 'absolute'
  454. >>> t.print_info()
  455. +-------------------- Basic information -------------------------------------+
  456. | Id: ........................ soil@PERMANENT
  457. | Name: ...................... soil
  458. | Mapset: .................... PERMANENT
  459. | Creator: ................... soeren
  460. | Temporal type: ............. absolute
  461. | Creation time: ............. 2001-01-01 00:00:00
  462. >>> t.print_shell_info()
  463. id=soil@PERMANENT
  464. name=soil
  465. mapset=PERMANENT
  466. creator=soeren
  467. temporal_type=absolute
  468. creation_time=2001-01-01 00:00:00
  469. \endcode
  470. """
  471. def __init__(self, table=None, ident=None, name=None, mapset=None,
  472. creator=None, ctime=None,ttype=None):
  473. """!Constructor
  474. @param table The name of the temporal database table
  475. that should be used to store the values
  476. @param ident The unique identifier must be a combination of
  477. the dataset name, layer name and the mapset
  478. "name@mapset" or "name:layer@mapset"
  479. used as as primary key in the temporal database
  480. @param name The name of the map or dataset
  481. @param mapset The name of the mapset
  482. @param creator The name of the creator
  483. @param ctime The creation datetime object
  484. @param ttype The temporal type
  485. - "absolute" Identifier for absolute time
  486. - "relative" Identifier for relative time
  487. """
  488. SQLDatabaseInterface.__init__(self, table, ident)
  489. self.set_id(ident)
  490. if ident is not None and name is None and mapset is None:
  491. if ident.find("@") >= 0:
  492. name, mapset = ident.split("@")
  493. if name.find(":") >= 0:
  494. name, layer = ident.split(":")
  495. self.set_name(name)
  496. self.set_mapset(mapset)
  497. self.set_creator(creator)
  498. self.set_ctime(ctime)
  499. self.set_ttype(ttype)
  500. def set_id(self, ident):
  501. """!Convenient method to set the unique identifier (primary key)
  502. @param ident The unique identifier must be a combination
  503. of the dataset name, layer name and the mapset
  504. "name@mapset" or "name:layer@mapset"
  505. """
  506. self.ident = ident
  507. self.D["id"] = ident
  508. if ident is not None:
  509. if ident.find("@") >= 0:
  510. name, mapset = ident.split("@")
  511. self.set_mapset(mapset)
  512. self.set_name(name)
  513. else:
  514. self.msgr.fatal(_("Wrong identifier, the mapset is missing"))
  515. if name.find(":") >= 0:
  516. name, layer = ident.split(":")
  517. self.set_layer(layer)
  518. self.set_name(name)
  519. def set_name(self, name):
  520. """!Set the name of the dataset
  521. @param name The name of the dataset
  522. """
  523. self.D["name"] = name
  524. def set_mapset(self, mapset):
  525. """!Set the mapset of the dataset
  526. @param mapset The name of the mapset in which this dataset is stored
  527. """
  528. self.D["mapset"] = mapset
  529. def set_layer(self, layer):
  530. """!Convenient method to set the layer of the map (part of primary key)
  531. Layer are supported for vector maps
  532. @param layer The layer of the map
  533. """
  534. self.D["layer"] = layer
  535. def set_creator(self, creator):
  536. """!Set the creator of the dataset
  537. @param creator The name of the creator
  538. """
  539. self.D["creator"] = creator
  540. def set_ctime(self, ctime=None):
  541. """!Set the creation time of the dataset,
  542. if nothing set the current time is used
  543. @param ctime The current time of type datetime
  544. """
  545. if ctime is None:
  546. self.D["creation_time"] = datetime.today()
  547. else:
  548. self.D["creation_time"] = ctime
  549. def set_ttype(self, ttype):
  550. """!Set the temporal type of the dataset: absolute or relative,
  551. if nothing set absolute time will assumed
  552. @param ttype The temporal type of the dataset "absolute or relative"
  553. """
  554. if ttype is None or (ttype != "absolute" and ttype != "relative"):
  555. self.D["temporal_type"] = "absolute"
  556. else:
  557. self.D["temporal_type"] = ttype
  558. def get_id(self):
  559. """!Convenient method to get the unique identifier (primary key)
  560. @return None if not found
  561. """
  562. if "id" in self.D:
  563. return self.D["id"]
  564. else:
  565. return None
  566. def get_map_id(self):
  567. """!Convenient method to get the unique map identifier
  568. without layer information
  569. @return the name of the vector map as "name@mapset"
  570. or None in case the id was not set
  571. """
  572. if self.id:
  573. if self.id.find(":") >= 0:
  574. # Remove the layer identifier from the id
  575. return self.id.split("@")[0].split(":")[0] + "@" + \
  576. self.id.split("@")[1]
  577. else:
  578. return self.id
  579. else:
  580. return None
  581. def get_layer(self):
  582. """!Convenient method to get the layer of the map (part of primary key)
  583. Layer are currently supported for vector maps
  584. @return None if not found
  585. """
  586. if "layer" in self.D:
  587. return self.D["layer"]
  588. else:
  589. return None
  590. def get_name(self):
  591. """!Get the name of the dataset
  592. @return None if not found"""
  593. if "name" in self.D:
  594. return self.D["name"]
  595. else:
  596. return None
  597. def get_mapset(self):
  598. """!Get the name of mapset of this dataset
  599. @return None if not found"""
  600. if "mapset" in self.D:
  601. return self.D["mapset"]
  602. else:
  603. return None
  604. def get_creator(self):
  605. """!Get the creator of the dataset
  606. @return None if not found"""
  607. if "creator" in self.D:
  608. return self.D["creator"]
  609. else:
  610. return None
  611. def get_ctime(self):
  612. """!Get the creation time of the dataset, datatype is datetime
  613. @return None if not found"""
  614. if "creation_time" in self.D:
  615. return self.D["creation_time"]
  616. else:
  617. return None
  618. def get_ttype(self):
  619. """!Get the temporal type of the map
  620. @return None if not found"""
  621. if "temporal_type" in self.D:
  622. return self.D["temporal_type"]
  623. else:
  624. return None
  625. # Properties of this class
  626. id = property(fget=get_id, fset=set_id)
  627. map_id = property(fget=get_map_id, fset=None)
  628. name = property(fget=get_name, fset=set_name)
  629. mapset = property(fget=get_mapset, fset=set_mapset)
  630. ctime = property(fget=get_ctime, fset=set_ctime)
  631. ttype = property(fget=get_ttype, fset=set_ttype)
  632. creator = property(fget=get_creator, fset=set_creator)
  633. def print_info(self):
  634. """!Print information about this class in human readable style"""
  635. # 0123456789012345678901234567890
  636. print " +-------------------- Basic information -------------------------------------+"
  637. print " | Id: ........................ " + str(self.get_id())
  638. print " | Name: ...................... " + str(self.get_name())
  639. print " | Mapset: .................... " + str(self.get_mapset())
  640. if self.get_layer():
  641. print " | Layer:...................... " + str(self.get_layer())
  642. print " | Creator: ................... " + str(self.get_creator())
  643. print " | Temporal type: ............. " + str(self.get_ttype())
  644. print " | Creation time: ............. " + str(self.get_ctime())
  645. def print_shell_info(self):
  646. """!Print information about this class in shell style"""
  647. print "id=" + str(self.get_id())
  648. print "name=" + str(self.get_name())
  649. print "mapset=" + str(self.get_mapset())
  650. if self.get_layer():
  651. print "layer=" + str(self.get_layer())
  652. print "creator=" + str(self.get_creator())
  653. print "temporal_type=" + str(self.get_ttype())
  654. print "creation_time=" + str(self.get_ctime())
  655. ###############################################################################
  656. class RasterBase(DatasetBase):
  657. """!Time stamped raster map base information class"""
  658. def __init__(self, ident=None, name=None, mapset=None, creator=None,
  659. creation_time=None, temporal_type=None):
  660. DatasetBase.__init__(self, "raster_base", ident, name, mapset,
  661. creator, creation_time, temporal_type)
  662. class Raster3DBase(DatasetBase):
  663. """!Time stamped 3D raster map base information class"""
  664. def __init__(self, ident=None, name=None, mapset=None, creator=None,
  665. creation_time=None, temporal_type=None,):
  666. DatasetBase.__init__(self, "raster3d_base", ident, name,
  667. mapset, creator, creation_time,
  668. temporal_type)
  669. class VectorBase(DatasetBase):
  670. """!Time stamped vector map base information class"""
  671. def __init__(self, ident=None, name=None, mapset=None, layer=None,
  672. creator=None, creation_time=None, temporal_type=None):
  673. DatasetBase.__init__(self, "vector_base", ident, name, mapset,
  674. creator, creation_time, temporal_type)
  675. self.set_id(ident)
  676. if ident is not None and name is None and mapset is None:
  677. if ident.find("@") >= 0:
  678. name, mapset = ident.split("@")
  679. if layer is None:
  680. if name.find(":") >= 0:
  681. name, layer = name.split(":")
  682. self.set_name(name)
  683. self.set_mapset(mapset)
  684. # Layer currently only in use by vector maps
  685. self.set_layer(layer)
  686. ###############################################################################
  687. class STDSBase(DatasetBase):
  688. """!Base class for space time datasets
  689. This class adds the semantic type member variable to the dataset
  690. base class.
  691. Usage:
  692. \code
  693. >>> init()
  694. >>> t = STDSBase("stds", "soil@PERMANENT", semantic_type="average", creator="soeren", ctime=datetime(2001,1,1), ttype="absolute", mtime=datetime(2001,1,1))
  695. >>> t.semantic_type
  696. 'average'
  697. >>> t.print_info()
  698. +-------------------- Basic information -------------------------------------+
  699. | Id: ........................ soil@PERMANENT
  700. | Name: ...................... soil
  701. | Mapset: .................... PERMANENT
  702. | Creator: ................... soeren
  703. | Temporal type: ............. absolute
  704. | Creation time: ............. 2001-01-01 00:00:00
  705. | Modification time:.......... 2001-01-01 00:00:00
  706. | Semantic type:.............. average
  707. >>> t.print_shell_info()
  708. id=soil@PERMANENT
  709. name=soil
  710. mapset=PERMANENT
  711. creator=soeren
  712. temporal_type=absolute
  713. creation_time=2001-01-01 00:00:00
  714. modification_time=2001-01-01 00:00:00
  715. semantic_type=average
  716. \endcode
  717. """
  718. def __init__(self, table=None, ident=None, name=None, mapset=None,
  719. semantic_type=None, creator=None, ctime=None,
  720. ttype=None, mtime=None):
  721. DatasetBase.__init__(self, table, ident, name, mapset, creator,
  722. ctime, ttype)
  723. self.set_semantic_type(semantic_type)
  724. self.set_mtime(mtime)
  725. def set_semantic_type(self, semantic_type):
  726. """!Set the semantic type of the space time dataset"""
  727. self.D["semantic_type"] = semantic_type
  728. def set_mtime(self, mtime=None):
  729. """!Set the modification time of the space time dataset, if nothing set the current time is used"""
  730. if mtime == None:
  731. self.D["modification_time"] = datetime.now()
  732. else:
  733. self.D["modification_time"] = mtime
  734. def get_semantic_type(self):
  735. """!Get the semantic type of the space time dataset
  736. @return None if not found"""
  737. if "semantic_type" in self.D:
  738. return self.D["semantic_type"]
  739. else:
  740. return None
  741. def get_mtime(self):
  742. """!Get the modification time of the space time dataset, datatype is datetime
  743. @return None if not found"""
  744. if self.D.has_key("modification_time"):
  745. return self.D["modification_time"]
  746. else:
  747. return None
  748. semantic_type = property(fget=get_semantic_type, fset=set_semantic_type)
  749. def print_info(self):
  750. """!Print information about this class in human readable style"""
  751. DatasetBase.print_info(self)
  752. # 0123456789012345678901234567890
  753. print " | Modification time:.......... " + str(self.get_mtime())
  754. print " | Semantic type:.............. " + str(
  755. self.get_semantic_type())
  756. def print_shell_info(self):
  757. """!Print information about this class in shell style"""
  758. DatasetBase.print_shell_info(self)
  759. print "modification_time=" + str(self.get_mtime())
  760. print "semantic_type=" + str(self.get_semantic_type())
  761. ###############################################################################
  762. class STRDSBase(STDSBase):
  763. """!Space time raster dataset base information class"""
  764. def __init__(self, ident=None, name=None, mapset=None,
  765. semantic_type=None, creator=None, ctime=None,
  766. ttype=None):
  767. STDSBase.__init__(self, "strds_base", ident, name, mapset,
  768. semantic_type, creator, ctime,
  769. ttype)
  770. class STR3DSBase(STDSBase):
  771. """!Space time 3D raster dataset base information class"""
  772. def __init__(self, ident=None, name=None, mapset=None,
  773. semantic_type=None, creator=None, ctime=None,
  774. ttype=None):
  775. STDSBase.__init__(self, "str3ds_base", ident, name, mapset,
  776. semantic_type, creator, ctime,
  777. ttype)
  778. class STVDSBase(STDSBase):
  779. """!Space time vector dataset base information class"""
  780. def __init__(self, ident=None, name=None, mapset=None,
  781. semantic_type=None, creator=None, ctime=None,
  782. ttype=None):
  783. STDSBase.__init__(self, "stvds_base", ident, name, mapset,
  784. semantic_type, creator, ctime,
  785. ttype)
  786. ###############################################################################
  787. class AbstractSTDSRegister(SQLDatabaseInterface):
  788. """!This is the base class for all maps to store the space time datasets
  789. as comma separated string in which they are registered
  790. Usage:
  791. \code
  792. >>> init()
  793. >>> t = AbstractSTDSRegister("raster", "soil@PERMANENT", "A@P,B@P,C@P")
  794. >>> t.id
  795. 'soil@PERMANENT'
  796. >>> t.registered_stds
  797. 'A@P,B@P,C@P'
  798. \endcode
  799. """
  800. def __init__(self, table=None, ident=None, registered_stds=None):
  801. """!Constructor
  802. @param table The name of the temporal database table
  803. that should be used to store the values
  804. @param ident The unique identifier must be a combination of
  805. the dataset name, layer name and the mapset
  806. "name@mapset" or "name:layer@mapset"
  807. used as as primary key in the temporal database
  808. @param stds A comma separted list of space time dataset ids
  809. """
  810. SQLDatabaseInterface.__init__(self, table, ident)
  811. self.set_id(ident)
  812. self.set_registered_stds(registered_stds)
  813. def set_id(self, ident):
  814. """!Convenient method to set the unique identifier (primary key)
  815. @param ident The unique identifier must be a combination
  816. of the dataset name, layer name and the mapset
  817. "name@mapset" or "name:layer@mapset"
  818. """
  819. self.ident = ident
  820. self.D["id"] = ident
  821. def set_registered_stds(self, registered_stds):
  822. """!Get the comma separated list of space time datasets ids
  823. in which this map is registered
  824. @param registered_stds A comma separated list of space time dataset ids
  825. in which this map is registered
  826. """
  827. self.D["registered_stds"] = registered_stds
  828. def get_id(self):
  829. """!Convenient method to get the unique identifier (primary key)
  830. @return None if not found
  831. """
  832. if "id" in self.D:
  833. return self.D["id"]
  834. else:
  835. return None
  836. def get_registered_stds(self):
  837. """!Get the comma separated list of space time datasets ids
  838. in which this map is registered
  839. @return None if not found"""
  840. if "registered_stds" in self.D:
  841. return self.D["registered_stds"]
  842. else:
  843. return None
  844. # Properties of this class
  845. id = property(fget=get_id, fset=set_id)
  846. registered_stds = property(fget=get_registered_stds, fset=set_registered_stds)
  847. ###############################################################################
  848. class RasterSTDSRegister(AbstractSTDSRegister):
  849. """!Time stamped raster map base information class"""
  850. def __init__(self, ident=None, registered_stds=None):
  851. AbstractSTDSRegister.__init__(self, "raster_stds_register", ident, registered_stds)
  852. class Raster3DSTDSRegister(AbstractSTDSRegister):
  853. """!Time stamped 3D raster map base information class"""
  854. def __init__(self, ident=None, registered_stds=None):
  855. AbstractSTDSRegister.__init__(self, "raster3d_stds_register", ident, registered_stds)
  856. class VectorSTDSRegister(AbstractSTDSRegister):
  857. """!Time stamped vector map base information class"""
  858. def __init__(self, ident=None, registered_stds=None):
  859. AbstractSTDSRegister.__init__(self, "vector_stds_register", ident, registered_stds)
  860. ###############################################################################
  861. if __name__ == "__main__":
  862. import doctest
  863. doctest.testmod()