base.py 32 KB

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