after12am / dress Goto Github PK

class DataSource(object):

    def __new__(cls, *args, **kwargs):
        if hasattr(cls, 'instance'):
            return cls.instance
        return None

    @classmethod
    def select(cls, datasource):
        cls.datasource = datasource

    @classmethod
    def connect(cls, options):
        if cls.datasource == 'Database/MySQL':
            cls.instance = MySQL(host=options.host, user=options.user, \
                password=options.password, charset=options.charset, database=options.database)
        if cls.datasource == 'Database/PostgreSQL':
            cls.instance = PostgreSQL(host=options.host, user=options.user, \
                password=options.password, database=options.database, port=options.port)
        if cls.datasource == 'Database/SQLite3':
            cls.instance = SQLite3(options.database)
        if hasattr(cls, 'instance') and cls.instance:
            cls.instance.connect()
            return True
        return False

    @classmethod
    def close(cls):
        if hasattr(cls, 'instance') and cls.instance:
            cls.instance.close()

def select(datasource):
    DataSource.select(datasource)

def connect(options):
    return DataSource.connect(options)

def close():
    DataSource.close()

def get_instance():
    return DataSource()

parser.add_option(
    "--datasource", 
    dest="datasource",
    help="Supported datasource. valid options are mysql, sqlite3, postgres",
    type="string",
    default=None
)

parser.add_option(
    "--database", 
    dest="database",
    help="Supported datasource. valid options are mysql, sqlite3, postgres",
    type="string",
    default=None
)