12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879 |
- """MySQL Database implementation."""
- from pwman.data.database import Database, __DB_FORMAT__
- import pymysql as mysql
- mysql.install_as_MySQLdb()
- class MySQLDatabase(Database):
- @classmethod
- def check_db_version(cls, dburi):
- port = 3306
- credentials, host = dburi.netloc.split('@')
- user, passwd = credentials.split(':')
- if ':' in host:
- host, port = host.split(':')
- port = int(port)
- con = mysql.connect(host=host, port=port, user=user, passwd=passwd,
- db=dburi.path.lstrip('/'))
- cur = con.cursor()
- try:
- cur.execute("SELECT VERSION FROM DBVERSION")
- version = cur.fetchone()
- cur.close()
- con.close()
- return version[-1]
- except mysql.ProgrammingError:
- con.rollback()
- return str(__DB_FORMAT__)
- def __init__(self, mysqluri, dbformat=__DB_FORMAT__):
- self.dburi = mysqluri
- self.dbversion = dbformat
- self._sub = "%s"
- self._list_nodes_sql = "SELECT NODEID FROM LOOKUP WHERE TAGID = %s "
- self._add_node_sql = ("INSERT INTO NODE(USERNAME, PASSWORD, URL, "
- "NOTES) "
- "VALUES(%s, %s, %s, %s)")
- self._insert_tag_sql = "INSERT INTO TAG(DATA) VALUES(%s)"
- self.ProgrammingError = mysql.ProgrammingError
- def _open(self):
- port = 3306
- credentials, host = self.dburi.netloc.split('@')
- user, passwd = credentials.split(':')
- if ':' in host:
- host, port = host.split(':')
- port = int(port)
- self._con = mysql.connect(host=host, port=port, user=user,
- passwd=passwd,
- db=self.dburi.path.lstrip('/'))
- self._cur = self._con.cursor()
- self._create_tables()
|