base.py 33 KB

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