base.py 39 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219
  1. """
  2. This packages includes all base classes to store basic information
  3. like id, name, mapset creation and modification time as well as sql
  4. serialization and de-serialization and the sql database interface.
  5. Usage:
  6. .. code-block:: python
  7. >>> import grass.temporal as tgis
  8. >>> tgis.init()
  9. >>> rbase = tgis.RasterBase(ident="soil@PERMANENT")
  10. >>> vbase = tgis.VectorBase(ident="soil:1@PERMANENT")
  11. >>> r3base = tgis.Raster3DBase(ident="soil@PERMANENT")
  12. >>> strdsbase = tgis.STRDSBase(ident="soil@PERMANENT")
  13. >>> stvdsbase = tgis.STVDSBase(ident="soil@PERMANENT")
  14. >>> str3dsbase = tgis.STR3DSBase(ident="soil@PERMANENT")
  15. (C) 2011-2013 by the GRASS Development Team
  16. This program is free software under the GNU General Public
  17. License (>=v2). Read the file COPYING that comes with GRASS
  18. for details.
  19. :author: Soeren Gebbert
  20. """
  21. from __future__ import print_function
  22. from datetime import datetime
  23. from .core import (
  24. get_tgis_message_interface,
  25. get_tgis_dbmi_paramstyle,
  26. SQLDatabaseInterfaceConnection,
  27. )
  28. ###############################################################################
  29. class DictSQLSerializer(object):
  30. def __init__(self):
  31. self.D = {}
  32. self.dbmi_paramstyle = get_tgis_dbmi_paramstyle()
  33. def serialize(self, type, table, where=None):
  34. """Convert the internal dictionary into a string of semicolon
  35. separated SQL statements The keys are the column names and
  36. the values are the row entries
  37. Usage:
  38. .. code-block:: python
  39. >>> init()
  40. >>> t = DictSQLSerializer()
  41. >>> t.D["id"] = "soil@PERMANENT"
  42. >>> t.D["name"] = "soil"
  43. >>> t.D["mapset"] = "PERMANENT"
  44. >>> t.D["creator"] = "soeren"
  45. >>> t.D["creation_time"] = datetime(2001,1,1)
  46. >>> t.D["modification_time"] = datetime(2001,1,1)
  47. >>> t.serialize(type="SELECT", table="raster_base")
  48. ('SELECT name , creator , creation_time , modification_time , mapset , id FROM raster_base ;\\n', ())
  49. >>> t.serialize(type="INSERT", table="raster_base")
  50. ('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'))
  51. >>> t.serialize(type="UPDATE", 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. >>> t.serialize(type="UPDATE ALL", table="raster_base")
  54. ('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'))
  55. :param type: must be SELECT. INSERT, UPDATE
  56. :param table: The name of the table to select, insert or update
  57. :param where: The optional where statement
  58. :return: a tuple containing the SQL string and the arguments
  59. """
  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 self.dbmi_paramstyle == "qmark":
  91. sql += "?"
  92. else:
  93. sql += "%s"
  94. else:
  95. if self.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 self.dbmi_paramstyle == "qmark":
  114. sql += " %s = ? " % key
  115. else:
  116. sql += " %s " % key
  117. sql += "= %s "
  118. else:
  119. if self.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 self.dbmi_paramstyle == "qmark":
  136. sql += " %s = ? " % key
  137. else:
  138. sql += " %s " % key
  139. sql += "= %s "
  140. else:
  141. if self.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. print(self.D)
  166. ###############################################################################
  167. class SQLDatabaseInterface(DictSQLSerializer):
  168. """This class represents the SQL database interface
  169. Functions to insert, select and update the internal
  170. structure of this class in the temporal database are implemented.
  171. This is the base class for raster, raster3d, vector and
  172. space time datasets data management classes:
  173. - Identification information (base)
  174. - Spatial extent
  175. - Temporal extent
  176. - Metadata
  177. Usage:
  178. .. code-block:: python
  179. >>> init()
  180. >>> t = SQLDatabaseInterface("raster", "soil@PERMANENT")
  181. >>> t.mapset = get_current_mapset()
  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. """
  207. def __init__(self, table=None, ident=None):
  208. """Constructor of this class
  209. :param table: The name of the table
  210. :param ident: The identifier (primary key) of this
  211. object in the database table
  212. """
  213. DictSQLSerializer.__init__(self)
  214. self.table = table # Name of the table, set in the subclass
  215. self.ident = ident
  216. self.msgr = get_tgis_message_interface()
  217. if self.ident and self.ident.find("@") >= 0:
  218. self.mapset = self.ident.split("@" "")[1]
  219. else:
  220. self.mapset = None
  221. def get_table_name(self):
  222. """Return the name of the table in which the internal
  223. data are inserted, updated or selected
  224. :return: The name of the table
  225. """
  226. return self.table
  227. def get_delete_statement(self):
  228. """Return the delete string
  229. :return: The DELETE string
  230. """
  231. return (
  232. "DELETE FROM "
  233. + self.get_table_name()
  234. + " WHERE id = '"
  235. + str(self.ident)
  236. + "';\n"
  237. )
  238. def delete(self, dbif=None):
  239. """Delete the entry of this object from the temporal database
  240. :param dbif: The database interface to be used,
  241. if None a temporary connection will be established
  242. """
  243. sql = self.get_delete_statement()
  244. # print(sql)
  245. if dbif:
  246. dbif.execute(sql, mapset=self.mapset)
  247. else:
  248. dbif = SQLDatabaseInterfaceConnection()
  249. dbif.connect()
  250. dbif.execute(sql, mapset=self.mapset)
  251. dbif.close()
  252. def get_is_in_db_statement(self):
  253. """Return the selection string that checks if this object is registered in the
  254. temporal database
  255. :return: The SELECT string
  256. """
  257. return (
  258. "SELECT id FROM "
  259. + self.get_table_name()
  260. + " WHERE id = '"
  261. + str(self.ident)
  262. + "';\n"
  263. )
  264. def is_in_db(self, dbif=None):
  265. """Check if this object is present in the temporal database
  266. :param dbif: The database interface to be used,
  267. if None a temporary connection will be established
  268. :return: True if this object is present in the temporal database,
  269. False otherwise
  270. """
  271. sql = self.get_is_in_db_statement()
  272. if dbif:
  273. dbif.execute(sql, mapset=self.mapset)
  274. row = dbif.fetchone(mapset=self.mapset)
  275. else:
  276. dbif = SQLDatabaseInterfaceConnection()
  277. dbif.connect()
  278. dbif.execute(sql, mapset=self.mapset)
  279. row = dbif.fetchone(mapset=self.mapset)
  280. dbif.close()
  281. # Nothing found
  282. if row is None:
  283. return False
  284. return True
  285. def get_select_statement(self):
  286. """Return the sql statement and the argument list in
  287. database specific style
  288. :return: The SELECT string
  289. """
  290. return self.serialize(
  291. "SELECT", self.get_table_name(), "WHERE id = '" + str(self.ident) + "'"
  292. )
  293. def get_select_statement_mogrified(self, dbif=None):
  294. """Return the select statement as mogrified string
  295. :param dbif: The database interface to be used,
  296. if None a temporary connection will be established
  297. :return: The SELECT string
  298. """
  299. if not dbif:
  300. dbif = SQLDatabaseInterfaceConnection()
  301. return dbif.mogrify_sql_statement(
  302. self.get_select_statement(), mapset=self.mapset
  303. )
  304. def select(self, dbif=None):
  305. """Select the content from the temporal database and store it
  306. in the internal dictionary structure
  307. :param dbif: The database interface to be used,
  308. if None a temporary connection will be established
  309. """
  310. sql, args = self.get_select_statement()
  311. # print(sql)
  312. # print(args)
  313. if dbif:
  314. if len(args) == 0:
  315. dbif.execute(sql, mapset=self.mapset)
  316. else:
  317. dbif.execute(sql, args, mapset=self.mapset)
  318. row = dbif.fetchone(mapset=self.mapset)
  319. else:
  320. dbif = SQLDatabaseInterfaceConnection()
  321. dbif.connect()
  322. if len(args) == 0:
  323. dbif.execute(sql, mapset=self.mapset)
  324. else:
  325. dbif.execute(sql, args, mapset=self.mapset)
  326. row = dbif.fetchone(mapset=self.mapset)
  327. dbif.close()
  328. # Nothing found
  329. if row is None:
  330. return False
  331. if len(row) > 0:
  332. self.deserialize(row)
  333. else:
  334. self.msgr.fatal(_("Object not found in the temporal database"))
  335. return True
  336. def get_insert_statement(self):
  337. """Return the sql statement and the argument
  338. list in database specific style
  339. :return: The INSERT string"""
  340. return self.serialize("INSERT", self.get_table_name())
  341. def get_insert_statement_mogrified(self, dbif=None):
  342. """Return the insert statement as mogrified string
  343. :param dbif: The database interface to be used,
  344. if None a temporary connection will be established
  345. :return: The INSERT string
  346. """
  347. if not dbif:
  348. dbif = SQLDatabaseInterfaceConnection()
  349. return dbif.mogrify_sql_statement(
  350. self.get_insert_statement(), mapset=self.mapset
  351. )
  352. def insert(self, dbif=None):
  353. """Serialize the content of this object and store it in the temporal
  354. database using the internal identifier
  355. :param dbif: The database interface to be used,
  356. if None a temporary connection will be established
  357. """
  358. sql, args = self.get_insert_statement()
  359. # print(sql)
  360. # print(args)
  361. if dbif:
  362. dbif.execute(sql, args, mapset=self.mapset)
  363. else:
  364. dbif = SQLDatabaseInterfaceConnection()
  365. dbif.connect()
  366. dbif.execute(sql, args, mapset=self.mapset)
  367. dbif.close()
  368. def get_update_statement(self, ident=None):
  369. """Return the sql statement and the argument list
  370. in database specific style
  371. :param ident: The identifier to be updated, useful for renaming
  372. :return: The UPDATE string
  373. """
  374. if ident:
  375. return self.serialize(
  376. "UPDATE", self.get_table_name(), "WHERE id = '" + str(ident) + "'"
  377. )
  378. else:
  379. return self.serialize(
  380. "UPDATE", self.get_table_name(), "WHERE id = '" + str(self.ident) + "'"
  381. )
  382. def get_update_statement_mogrified(self, dbif=None, ident=None):
  383. """Return the update statement as mogrified string
  384. :param dbif: The database interface to be used,
  385. if None a temporary connection will be established
  386. :param ident: The identifier to be updated, useful for renaming
  387. :return: The UPDATE string
  388. """
  389. if not dbif:
  390. dbif = SQLDatabaseInterfaceConnection()
  391. return dbif.mogrify_sql_statement(
  392. self.get_update_statement(ident), mapset=self.mapset
  393. )
  394. def update(self, dbif=None, ident=None):
  395. """Serialize the content of this object and update it in the temporal
  396. database using the internal identifier
  397. Only object entries which are exists (not None) are updated
  398. :param dbif: The database interface to be used,
  399. if None a temporary connection will be established
  400. :param ident: The identifier to be updated, useful for renaming
  401. """
  402. if self.ident is None:
  403. self.msgr.fatal(_("Missing identifier"))
  404. sql, args = self.get_update_statement(ident)
  405. # print(sql)
  406. # print(args)
  407. if dbif:
  408. dbif.execute(sql, args, mapset=self.mapset)
  409. else:
  410. dbif = SQLDatabaseInterfaceConnection()
  411. dbif.connect()
  412. dbif.execute(sql, args, mapset=self.mapset)
  413. dbif.close()
  414. def get_update_all_statement(self, ident=None):
  415. """Return the sql statement and the argument
  416. list in database specific style
  417. :param ident: The identifier to be updated, useful for renaming
  418. :return: The UPDATE string
  419. """
  420. if ident:
  421. return self.serialize(
  422. "UPDATE ALL", self.get_table_name(), "WHERE id = '" + str(ident) + "'"
  423. )
  424. else:
  425. return self.serialize(
  426. "UPDATE ALL",
  427. self.get_table_name(),
  428. "WHERE id = '" + str(self.ident) + "'",
  429. )
  430. def get_update_all_statement_mogrified(self, dbif=None, ident=None):
  431. """Return the update all statement as mogrified string
  432. :param dbif: The database interface to be used,
  433. if None a temporary connection will be established
  434. :param ident: The identifier to be updated, useful for renaming
  435. :return: The UPDATE string
  436. """
  437. if not dbif:
  438. dbif = SQLDatabaseInterfaceConnection()
  439. return dbif.mogrify_sql_statement(
  440. self.get_update_all_statement(ident), mapset=self.mapset
  441. )
  442. def update_all(self, dbif=None, ident=None):
  443. """Serialize the content of this object, including None objects,
  444. and update it in the temporal database using the internal identifier
  445. :param dbif: The database interface to be used,
  446. if None a temporary connection will be established
  447. :param ident: The identifier to be updated, useful for renaming
  448. """
  449. if self.ident is None:
  450. self.msgr.fatal(_("Missing identifier"))
  451. sql, args = self.get_update_all_statement(ident)
  452. # print(sql)
  453. # print(args)
  454. if dbif:
  455. dbif.execute(sql, args, mapset=self.mapset)
  456. else:
  457. dbif = SQLDatabaseInterfaceConnection()
  458. dbif.connect()
  459. dbif.execute(sql, args, mapset=self.mapset)
  460. dbif.close()
  461. ###############################################################################
  462. class DatasetBase(SQLDatabaseInterface):
  463. """This is the base class for all maps and spacetime datasets storing
  464. basic identification information
  465. Usage:
  466. .. code-block:: python
  467. >>> init()
  468. >>> t = DatasetBase("raster", "soil@PERMANENT", creator="soeren", ctime=datetime(2001,1,1), ttype="absolute")
  469. >>> t.id
  470. 'soil@PERMANENT'
  471. >>> t.name
  472. 'soil'
  473. >>> t.mapset
  474. 'PERMANENT'
  475. >>> t.creator
  476. 'soeren'
  477. >>> t.ctime
  478. datetime.datetime(2001, 1, 1, 0, 0)
  479. >>> t.ttype
  480. 'absolute'
  481. >>> t.print_info()
  482. +-------------------- Basic information -------------------------------------+
  483. | Id: ........................ soil@PERMANENT
  484. | Name: ...................... soil
  485. | Mapset: .................... PERMANENT
  486. | Creator: ................... soeren
  487. | Temporal type: ............. absolute
  488. | Creation time: ............. 2001-01-01 00:00:00
  489. >>> t.print_shell_info()
  490. id=soil@PERMANENT
  491. name=soil
  492. mapset=PERMANENT
  493. creator=soeren
  494. temporal_type=absolute
  495. creation_time='2001-01-01 00:00:00'
  496. """
  497. def __init__(
  498. self,
  499. table=None,
  500. ident=None,
  501. name=None,
  502. mapset=None,
  503. creator=None,
  504. ctime=None,
  505. ttype=None,
  506. ):
  507. """Constructor
  508. :param table: The name of the temporal database table
  509. that should be used to store the values
  510. :param ident: The unique identifier must be a combination of
  511. the dataset name, layer name and the mapset
  512. "name@mapset" or "name:layer@mapset"
  513. used as as primary key in the temporal database
  514. :param name: The name of the map or dataset
  515. :param mapset: The name of the mapset
  516. :param creator: The name of the creator
  517. :param ctime: The creation datetime object
  518. :param ttype: The temporal type
  519. - "absolute" Identifier for absolute time
  520. - "relative" Identifier for relative time
  521. """
  522. SQLDatabaseInterface.__init__(self, table, ident)
  523. self.set_id(ident)
  524. if ident is not None and name is None and mapset is None:
  525. if ident.find("@") >= 0:
  526. name, mapset = ident.split("@")
  527. if name.find(":") >= 0:
  528. name, layer = ident.split(":")
  529. self.set_name(name)
  530. self.set_mapset(mapset)
  531. self.set_creator(creator)
  532. self.set_ctime(ctime)
  533. self.set_ttype(ttype)
  534. def set_id(self, ident):
  535. """Convenient method to set the unique identifier (primary key)
  536. :param ident: The unique identifier must be a combination
  537. of the dataset name, layer name and the mapset
  538. "name@mapset" or "name:layer@mapset"
  539. """
  540. self.ident = ident
  541. self.D["id"] = ident
  542. name = ""
  543. if ident is not None:
  544. if ident.find("@") >= 0:
  545. name, mapset = ident.split("@")
  546. self.set_mapset(mapset)
  547. self.set_name(name)
  548. else:
  549. self.msgr.fatal(_("Wrong identifier, the mapset is missing"))
  550. if name.find(":") >= 0:
  551. name, layer = ident.split(":")
  552. self.set_layer(layer)
  553. self.set_name(name)
  554. def set_name(self, name):
  555. """Set the name of the dataset
  556. :param name: The name of the dataset
  557. """
  558. self.D["name"] = name
  559. def set_mapset(self, mapset):
  560. """Set the mapset of the dataset
  561. :param mapset: The name of the mapset in which this dataset is stored
  562. """
  563. self.D["mapset"] = mapset
  564. def set_layer(self, layer):
  565. """Convenient method to set the layer of the map (part of primary key)
  566. Layer are supported for vector maps
  567. :param layer: The layer of the map
  568. """
  569. self.D["layer"] = layer
  570. def set_creator(self, creator):
  571. """Set the creator of the dataset
  572. :param creator: The name of the creator
  573. """
  574. self.D["creator"] = creator
  575. def set_ctime(self, ctime=None):
  576. """Set the creation time of the dataset,
  577. if nothing set the current time is used
  578. :param ctime: The current time of type datetime
  579. """
  580. if ctime is None:
  581. self.D["creation_time"] = datetime.today()
  582. else:
  583. self.D["creation_time"] = ctime
  584. def set_ttype(self, ttype):
  585. """Set the temporal type of the dataset: absolute or relative,
  586. if nothing set absolute time will assumed
  587. :param ttype: The temporal type of the dataset "absolute or relative"
  588. """
  589. if ttype is None or (ttype != "absolute" and ttype != "relative"):
  590. self.D["temporal_type"] = "absolute"
  591. else:
  592. self.D["temporal_type"] = ttype
  593. def get_id(self):
  594. """Convenient method to get the unique identifier (primary key)
  595. :return: None if not found
  596. """
  597. if "id" in self.D:
  598. return self.D["id"]
  599. else:
  600. return None
  601. def get_map_id(self):
  602. """Convenient method to get the unique map identifier
  603. without layer information
  604. :return: the name of the vector map as "name@mapset"
  605. or None in case the id was not set
  606. """
  607. if self.id:
  608. if self.id.find(":") >= 0:
  609. # Remove the layer identifier from the id
  610. return self.id.split("@")[0].split(":")[0] + "@" + self.id.split("@")[1]
  611. else:
  612. return self.id
  613. else:
  614. return None
  615. def get_layer(self):
  616. """Convenient method to get the layer of the map (part of primary key)
  617. Layer are currently supported for vector maps
  618. :return: None if not found
  619. """
  620. if "layer" in self.D:
  621. return self.D["layer"]
  622. else:
  623. return None
  624. def get_name(self):
  625. """Get the name of the dataset
  626. :return: None if not found"""
  627. if "name" in self.D:
  628. return self.D["name"]
  629. else:
  630. return None
  631. def get_mapset(self):
  632. """Get the name of mapset of this dataset
  633. :return: None if not found"""
  634. if "mapset" in self.D:
  635. return self.D["mapset"]
  636. else:
  637. return None
  638. def get_creator(self):
  639. """Get the creator of the dataset
  640. :return: None if not found"""
  641. if "creator" in self.D:
  642. return self.D["creator"]
  643. else:
  644. return None
  645. def get_ctime(self):
  646. """Get the creation time of the dataset, datatype is datetime
  647. :return: None if not found"""
  648. if "creation_time" in self.D:
  649. return self.D["creation_time"]
  650. else:
  651. return None
  652. def get_ttype(self):
  653. """Get the temporal type of the map
  654. :return: None if not found"""
  655. if "temporal_type" in self.D:
  656. return self.D["temporal_type"]
  657. else:
  658. return None
  659. # Properties of this class
  660. id = property(fget=get_id, fset=set_id)
  661. map_id = property(fget=get_map_id, fset=None)
  662. name = property(fget=get_name, fset=set_name)
  663. mapset = property(fget=get_mapset, fset=set_mapset)
  664. ctime = property(fget=get_ctime, fset=set_ctime)
  665. ttype = property(fget=get_ttype, fset=set_ttype)
  666. creator = property(fget=get_creator, fset=set_creator)
  667. def print_info(self):
  668. """Print information about this class in human readable style"""
  669. # 0123456789012345678901234567890
  670. print(
  671. " +-------------------- Basic information -------------------------------------+"
  672. )
  673. print(" | Id: ........................ " + str(self.get_id()))
  674. print(" | Name: ...................... " + str(self.get_name()))
  675. print(" | Mapset: .................... " + str(self.get_mapset()))
  676. if self.get_layer():
  677. print(" | Layer:...................... " + str(self.get_layer()))
  678. print(" | Creator: ................... " + str(self.get_creator()))
  679. print(" | Temporal type: ............. " + str(self.get_ttype()))
  680. print(" | Creation time: ............. " + str(self.get_ctime()))
  681. def print_shell_info(self):
  682. """Print information about this class in shell style"""
  683. print("id=" + str(self.get_id()))
  684. print("name=" + str(self.get_name()))
  685. print("mapset=" + str(self.get_mapset()))
  686. if self.get_layer():
  687. print("layer=" + str(self.get_layer()))
  688. print("creator=" + str(self.get_creator()))
  689. print("temporal_type=" + str(self.get_ttype()))
  690. print("creation_time='{}'".format(str(self.get_ctime())))
  691. ###############################################################################
  692. class RasterBase(DatasetBase):
  693. """Time stamped raster map base information class"""
  694. def __init__(
  695. self,
  696. ident=None,
  697. name=None,
  698. mapset=None,
  699. creator=None,
  700. creation_time=None,
  701. temporal_type=None,
  702. ):
  703. DatasetBase.__init__(
  704. self,
  705. "raster_base",
  706. ident,
  707. name,
  708. mapset,
  709. creator,
  710. creation_time,
  711. temporal_type,
  712. )
  713. class Raster3DBase(DatasetBase):
  714. """Time stamped 3D raster map base information class"""
  715. def __init__(
  716. self,
  717. ident=None,
  718. name=None,
  719. mapset=None,
  720. creator=None,
  721. creation_time=None,
  722. temporal_type=None,
  723. ):
  724. DatasetBase.__init__(
  725. self,
  726. "raster3d_base",
  727. ident,
  728. name,
  729. mapset,
  730. creator,
  731. creation_time,
  732. temporal_type,
  733. )
  734. class VectorBase(DatasetBase):
  735. """Time stamped vector map base information class"""
  736. def __init__(
  737. self,
  738. ident=None,
  739. name=None,
  740. mapset=None,
  741. layer=None,
  742. creator=None,
  743. creation_time=None,
  744. temporal_type=None,
  745. ):
  746. DatasetBase.__init__(
  747. self,
  748. "vector_base",
  749. ident,
  750. name,
  751. mapset,
  752. creator,
  753. creation_time,
  754. temporal_type,
  755. )
  756. self.set_id(ident)
  757. if ident is not None and name is None and mapset is None:
  758. if ident.find("@") >= 0:
  759. name, mapset = ident.split("@")
  760. if layer is None:
  761. if name.find(":") >= 0:
  762. name, layer = name.split(":")
  763. self.set_name(name)
  764. self.set_mapset(mapset)
  765. # Layer currently only in use by vector maps
  766. self.set_layer(layer)
  767. ###############################################################################
  768. class STDSBase(DatasetBase):
  769. """Base class for space time datasets
  770. This class adds the semantic type member variable to the dataset
  771. base class.
  772. Usage:
  773. .. code-block:: python
  774. >>> init()
  775. >>> t = STDSBase("stds", "soil@PERMANENT", semantic_type="average", creator="soeren", ctime=datetime(2001,1,1), ttype="absolute", mtime=datetime(2001,1,1))
  776. >>> t.semantic_type
  777. 'average'
  778. >>> t.print_info()
  779. +-------------------- Basic information -------------------------------------+
  780. | Id: ........................ soil@PERMANENT
  781. | Name: ...................... soil
  782. | Mapset: .................... PERMANENT
  783. | Creator: ................... soeren
  784. | Temporal type: ............. absolute
  785. | Creation time: ............. 2001-01-01 00:00:00
  786. | Modification time:.......... 2001-01-01 00:00:00
  787. | Semantic type:.............. average
  788. >>> t.print_shell_info()
  789. id=soil@PERMANENT
  790. name=soil
  791. mapset=PERMANENT
  792. creator=soeren
  793. temporal_type=absolute
  794. creation_time='2001-01-01 00:00:00'
  795. modification_time='2001-01-01 00:00:00'
  796. semantic_type=average
  797. """
  798. def __init__(
  799. self,
  800. table=None,
  801. ident=None,
  802. name=None,
  803. mapset=None,
  804. semantic_type=None,
  805. creator=None,
  806. ctime=None,
  807. ttype=None,
  808. mtime=None,
  809. ):
  810. DatasetBase.__init__(self, table, ident, name, mapset, creator, ctime, ttype)
  811. self.set_semantic_type(semantic_type)
  812. self.set_mtime(mtime)
  813. def set_semantic_type(self, semantic_type):
  814. """Set the semantic type of the space time dataset"""
  815. self.D["semantic_type"] = semantic_type
  816. def set_mtime(self, mtime=None):
  817. """Set the modification time of the space time dataset, if nothing set
  818. the current time is used
  819. """
  820. if mtime is None:
  821. self.D["modification_time"] = datetime.now()
  822. else:
  823. self.D["modification_time"] = mtime
  824. def get_semantic_type(self):
  825. """Get the semantic type of the space time dataset
  826. :return: None if not found
  827. """
  828. if "semantic_type" in self.D:
  829. return self.D["semantic_type"]
  830. else:
  831. return None
  832. def get_mtime(self):
  833. """Get the modification time of the space time dataset, datatype is
  834. datetime
  835. :return: None if not found
  836. """
  837. if "modification_time" in self.D:
  838. return self.D["modification_time"]
  839. else:
  840. return None
  841. semantic_type = property(fget=get_semantic_type, fset=set_semantic_type)
  842. def print_info(self):
  843. """Print information about this class in human readable style"""
  844. DatasetBase.print_info(self)
  845. # 0123456789012345678901234567890
  846. print(" | Modification time:.......... " + str(self.get_mtime()))
  847. print(" | Semantic type:.............. " + str(self.get_semantic_type()))
  848. def print_shell_info(self):
  849. """Print information about this class in shell style"""
  850. DatasetBase.print_shell_info(self)
  851. print("modification_time='{}'".format(str(self.get_mtime())))
  852. print("semantic_type=" + str(self.get_semantic_type()))
  853. ###############################################################################
  854. class STRDSBase(STDSBase):
  855. """Space time raster dataset base information class"""
  856. def __init__(
  857. self,
  858. ident=None,
  859. name=None,
  860. mapset=None,
  861. semantic_type=None,
  862. creator=None,
  863. ctime=None,
  864. ttype=None,
  865. ):
  866. STDSBase.__init__(
  867. self,
  868. "strds_base",
  869. ident,
  870. name,
  871. mapset,
  872. semantic_type,
  873. creator,
  874. ctime,
  875. ttype,
  876. )
  877. class STR3DSBase(STDSBase):
  878. """Space time 3D raster dataset base information class"""
  879. def __init__(
  880. self,
  881. ident=None,
  882. name=None,
  883. mapset=None,
  884. semantic_type=None,
  885. creator=None,
  886. ctime=None,
  887. ttype=None,
  888. ):
  889. STDSBase.__init__(
  890. self,
  891. "str3ds_base",
  892. ident,
  893. name,
  894. mapset,
  895. semantic_type,
  896. creator,
  897. ctime,
  898. ttype,
  899. )
  900. class STVDSBase(STDSBase):
  901. """Space time vector dataset base information class"""
  902. def __init__(
  903. self,
  904. ident=None,
  905. name=None,
  906. mapset=None,
  907. semantic_type=None,
  908. creator=None,
  909. ctime=None,
  910. ttype=None,
  911. ):
  912. STDSBase.__init__(
  913. self,
  914. "stvds_base",
  915. ident,
  916. name,
  917. mapset,
  918. semantic_type,
  919. creator,
  920. ctime,
  921. ttype,
  922. )
  923. ###############################################################################
  924. class AbstractSTDSRegister(SQLDatabaseInterface):
  925. """This is the base class for all maps to store the space time datasets
  926. as comma separated string in which they are registered
  927. Usage:
  928. .. code-block:: python
  929. >>> init()
  930. >>> t = AbstractSTDSRegister("raster", "soil@PERMANENT", "A@P,B@P,C@P")
  931. >>> t.id
  932. 'soil@PERMANENT'
  933. >>> t.registered_stds
  934. 'A@P,B@P,C@P'
  935. """
  936. def __init__(self, table=None, ident=None, registered_stds=None):
  937. """Constructor
  938. :param table: The name of the temporal database table
  939. that should be used to store the values
  940. :param ident: The unique identifier must be a combination of
  941. the dataset name, layer name and the mapset
  942. "name@mapset" or "name:layer@mapset"
  943. used as as primary key in the temporal database
  944. :param stds: A comma separated list of space time dataset ids
  945. """
  946. SQLDatabaseInterface.__init__(self, table, ident)
  947. self.set_id(ident)
  948. self.set_registered_stds(registered_stds)
  949. def set_id(self, ident):
  950. """Convenient method to set the unique identifier (primary key)
  951. :param ident: The unique identifier must be a combination
  952. of the dataset name, layer name and the mapset
  953. "name@mapset" or "name:layer@mapset"
  954. """
  955. self.ident = ident
  956. self.D["id"] = ident
  957. def set_registered_stds(self, registered_stds):
  958. """Get the comma separated list of space time datasets ids
  959. in which this map is registered
  960. :param registered_stds: A comma separated list of space time
  961. dataset ids in which this map is registered
  962. """
  963. self.D["registered_stds"] = registered_stds
  964. def get_id(self):
  965. """Convenient method to get the unique identifier (primary key)
  966. :return: None if not found
  967. """
  968. if "id" in self.D:
  969. return self.D["id"]
  970. else:
  971. return None
  972. def get_registered_stds(self):
  973. """Get the comma separated list of space time datasets ids
  974. in which this map is registered
  975. :return: None if not found
  976. """
  977. if "registered_stds" in self.D:
  978. return self.D["registered_stds"]
  979. else:
  980. return None
  981. # Properties of this class
  982. id = property(fget=get_id, fset=set_id)
  983. registered_stds = property(fget=get_registered_stds, fset=set_registered_stds)
  984. ###############################################################################
  985. class RasterSTDSRegister(AbstractSTDSRegister):
  986. """Time stamped raster map base information class"""
  987. def __init__(self, ident=None, registered_stds=None):
  988. AbstractSTDSRegister.__init__(
  989. self, "raster_stds_register", ident, registered_stds
  990. )
  991. class Raster3DSTDSRegister(AbstractSTDSRegister):
  992. """Time stamped 3D raster map base information class"""
  993. def __init__(self, ident=None, registered_stds=None):
  994. AbstractSTDSRegister.__init__(
  995. self, "raster3d_stds_register", ident, registered_stds
  996. )
  997. class VectorSTDSRegister(AbstractSTDSRegister):
  998. """Time stamped vector map base information class"""
  999. def __init__(self, ident=None, registered_stds=None):
  1000. AbstractSTDSRegister.__init__(
  1001. self, "vector_stds_register", ident, registered_stds
  1002. )
  1003. ###############################################################################
  1004. if __name__ == "__main__":
  1005. import doctest
  1006. doctest.testmod()