示例#1
0
def connect():
    #different connect code per tyoe of database
    if botsglobal.settings.DATABASE_ENGINE == 'sqlite3':
        #sqlite has some more fiddling; in separate file. Mainly because of some other method of parameter passing.
        if not os.path.isfile(botsglobal.settings.DATABASE_NAME):
            raise botslib.PanicError(
                _(u'Could not find database file for SQLite'))
        import botssqlite
        botsglobal.db = botssqlite.connect(
            database=botsglobal.settings.DATABASE_NAME)
    elif botsglobal.settings.DATABASE_ENGINE == 'mysql':
        import MySQLdb
        from MySQLdb import cursors
        botsglobal.db = MySQLdb.connect(
            host=botsglobal.settings.DATABASE_HOST,
            port=int(botsglobal.settings.DATABASE_PORT),
            db=botsglobal.settings.DATABASE_NAME,
            user=botsglobal.settings.DATABASE_USER,
            passwd=botsglobal.settings.DATABASE_PASSWORD,
            cursorclass=cursors.DictCursor,
            **botsglobal.settings.DATABASE_OPTIONS)
    elif botsglobal.settings.DATABASE_ENGINE == 'postgresql_psycopg2':
        import psycopg2
        import psycopg2.extensions
        import psycopg2.extras
        psycopg2.extensions.register_type(psycopg2.extensions.UNICODE)
        botsglobal.db = psycopg2.connect(
            'host=%s dbname=%s user=%s password=%s' %
            (botsglobal.settings.DATABASE_HOST,
             botsglobal.settings.DATABASE_NAME,
             botsglobal.settings.DATABASE_USER,
             botsglobal.settings.DATABASE_PASSWORD),
            connection_factory=psycopg2.extras.DictConnection)
        botsglobal.db.set_client_encoding('UNICODE')
示例#2
0
def connect():
    #different connect code per tyoe of database
    if botsglobal.settings.DATABASE_ENGINE == 'sqlite3':
        #sqlite has some more fiddling; in separate file. Mainly because of some other method of parameter passing.
        if not os.path.isfile(botsglobal.settings.DATABASE_NAME):
            raise botslib.PanicError(_(u'Could not find database file for SQLite'))
        import botssqlite
        botsglobal.db = botssqlite.connect(database = botsglobal.settings.DATABASE_NAME)
    elif botsglobal.settings.DATABASE_ENGINE == 'mysql':
        import MySQLdb
        from MySQLdb import cursors
        botsglobal.db = MySQLdb.connect(host=botsglobal.settings.DATABASE_HOST, 
                                        port=int(botsglobal.settings.DATABASE_PORT), 
                                        db=botsglobal.settings.DATABASE_NAME, 
                                        user=botsglobal.settings.DATABASE_USER, 
                                        passwd=botsglobal.settings.DATABASE_PASSWORD,
                                        cursorclass=cursors.DictCursor,
                                        **botsglobal.settings.DATABASE_OPTIONS)
    elif botsglobal.settings.DATABASE_ENGINE == 'postgresql_psycopg2':
        import psycopg2
        import psycopg2.extensions
        import psycopg2.extras
        psycopg2.extensions.register_type(psycopg2.extensions.UNICODE)
        botsglobal.db = psycopg2.connect( 'host=%s dbname=%s user=%s password=%s'%( botsglobal.settings.DATABASE_HOST,
                                                                                    botsglobal.settings.DATABASE_NAME,
                                                                                    botsglobal.settings.DATABASE_USER,
                                                                                    botsglobal.settings.DATABASE_PASSWORD),connection_factory=psycopg2.extras.DictConnection)
        botsglobal.db.set_client_encoding('UNICODE')
示例#3
0
def connect():
    ''' connect to database for non-django modules eg engine '''
    if botsglobal.settings.DATABASES['default']['ENGINE'] == 'django.db.backends.sqlite3':
        #sqlite has some more fiddling; in separate file. Mainly because of some other method of parameter passing.
        if not os.path.isfile(botsglobal.settings.DATABASES['default']['NAME']):
            raise botslib.PanicError(u'Could not find database file for SQLite')
        import botssqlite
        botsglobal.db = botssqlite.connect(database = botsglobal.settings.DATABASES['default']['NAME'])
    elif botsglobal.settings.DATABASES['default']['ENGINE'] == 'django.db.backends.mysql':
        import MySQLdb
        from MySQLdb import cursors
        botsglobal.db = MySQLdb.connect(host=botsglobal.settings.DATABASES['default']['HOST'],
                                        port=int(botsglobal.settings.DATABASES['default']['PORT']),
                                        db=botsglobal.settings.DATABASES['default']['NAME'],
                                        user=botsglobal.settings.DATABASES['default']['USER'],
                                        passwd=botsglobal.settings.DATABASES['default']['PASSWORD'],
                                        cursorclass=cursors.DictCursor,
                                        **botsglobal.settings.DATABASES['default']['OPTIONS'])
    elif botsglobal.settings.DATABASES['default']['ENGINE'] == 'django.db.backends.postgresql_psycopg2':
        import psycopg2
        import psycopg2.extensions
        import psycopg2.extras
        psycopg2.extensions.register_type(psycopg2.extensions.UNICODE)
        botsglobal.db = psycopg2.connect(host=botsglobal.settings.DATABASES['default']['HOST'],
                                        port=botsglobal.settings.DATABASES['default']['PORT'],
                                        database=botsglobal.settings.DATABASES['default']['NAME'],
                                        user=botsglobal.settings.DATABASES['default']['USER'],
                                        password=botsglobal.settings.DATABASES['default']['PASSWORD'],
                                        connection_factory=psycopg2.extras.DictConnection)
        botsglobal.db.set_client_encoding('UNICODE')
    else:
        raise botslib.PanicError(u'Unknown database engine "%(engine)s".',{'engine':botsglobal.settings.DATABASES['default']['ENGINE']})
示例#4
0
def connect():
    ''' connect to database for non-django modules eg engine '''
    if botsglobal.settings.DATABASES['default'][
            'ENGINE'] == 'django.db.backends.sqlite3':
        #sqlite has some more fiddling; in separate file. Mainly because of some other method of parameter passing.
        if not os.path.isfile(
                botsglobal.settings.DATABASES['default']['NAME']):
            raise botslib.PanicError(
                u'Could not find database file for SQLite')
        import botssqlite
        botsglobal.db = botssqlite.connect(
            database=botsglobal.settings.DATABASES['default']['NAME'])
    elif botsglobal.settings.DATABASES['default'][
            'ENGINE'] == 'django.db.backends.mysql':
        import MySQLdb
        from MySQLdb import cursors
        botsglobal.db = MySQLdb.connect(
            host=botsglobal.settings.DATABASES['default']['HOST'],
            port=int(botsglobal.settings.DATABASES['default']['PORT']),
            db=botsglobal.settings.DATABASES['default']['NAME'],
            user=botsglobal.settings.DATABASES['default']['USER'],
            passwd=botsglobal.settings.DATABASES['default']['PASSWORD'],
            cursorclass=cursors.DictCursor,
            **botsglobal.settings.DATABASES['default']['OPTIONS'])
    elif botsglobal.settings.DATABASES['default'][
            'ENGINE'] == 'django.db.backends.postgresql_psycopg2':
        import psycopg2
        import psycopg2.extensions
        import psycopg2.extras
        psycopg2.extensions.register_type(psycopg2.extensions.UNICODE)
        botsglobal.db = psycopg2.connect(
            host=botsglobal.settings.DATABASES['default']['HOST'],
            port=botsglobal.settings.DATABASES['default']['PORT'],
            database=botsglobal.settings.DATABASES['default']['NAME'],
            user=botsglobal.settings.DATABASES['default']['USER'],
            password=botsglobal.settings.DATABASES['default']['PASSWORD'],
            connection_factory=psycopg2.extras.DictConnection)
        botsglobal.db.set_client_encoding('UNICODE')
    else:
        raise botslib.PanicError(
            u'Unknown database engine "%(engine)s".',
            {'engine': botsglobal.settings.DATABASES['default']['ENGINE']})