#============================================================================ # This file is part of Pwman3. # # Pwman3 is free software; you can redistribute it and/or modify # it under the terms of the GNU General Public License, version 2 # as published by the Free Software Foundation; # # Pwman3 is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with Pwman3; if not, write to the Free Software # Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA #============================================================================ # Copyright (C) 2012 Oz Nahum #============================================================================ # Copyright (C) 2006 Ivan Kelly #============================================================================ """ 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 #from pwman.data.drivers import osqlite 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 # TODO: implement version checks for other supported DBs. 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": # pragma: no cover try: from pwman.data.drivers import postgresql db = postgresql.PostgresqlDatabase() except ImportError: raise DatabaseException("python-pygresql not installed") elif dbtype == "MySQL": # pragma: no cover 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