123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081 |
- """
- Factory to create Database instances
- A Generic interface for all DB engines.
- Usage:
- import pwman.data.factory as DBFactory
- db = DBFactory.create(params)
- db.open()
- .....
- """
- from pwman.data.database import DatabaseException
- from pwman.data.drivers import sqlite
- class FactoryException(Exception):
- def __init__(self, message):
- self.message
- def __str__(self):
- return self.message
- def check_db_version(type):
- if type == "SQLite":
- ver = sqlite.check_db_version()
- try:
- return float(ver.strip("\'"))
- except ValueError:
- return 0.3
-
- def create(dbtype, version=None, filename=None):
- """
- create(params) -> Database
- Create a Database instance.
- 'type' can only be 'SQLite' at the moment
- """
- if dbtype == "SQLite":
- from pwman.data.drivers import sqlite
- if filename:
- db = sqlite.SQLiteDatabaseNewForm(filename)
- elif version >= 0.4:
- db = sqlite.SQLiteDatabaseNewForm()
- elif dbtype == "Postgresql":
- try:
- from pwman.data.drivers import postgresql
- db = postgresql.PostgresqlDatabase()
- except ImportError:
- raise DatabaseException("python-pygresql not installed")
- elif dbtype == "MySQL":
- try:
- from pwman.data.drivers import mysql
- db = mysql.MySQLDatabase()
- except ImportError:
- raise DatabaseException("python-mysqldb not installed")
- else:
- raise DatabaseException("Unknown database type specified")
- return db
|