1 import MySQLdb as mysql

     2 import sys

     3 

     4 def connect():

     5     try:

     6         import secrets

     7     except ImportError:

     8         print >> sys.stderr, 'Decrypt secrets.py.gpg or create a new copy!'

     9         sys.exit(1)

    10 

    11     if secrets.DATABASE_ENGINE != 'mysql':

    12         print >> sys.stderr, ('You are using a %s database' %

    13                               secrets.DATABASE_ENGINE)

    14         sys.exit(1)

    15 

    16     kwargs = {

    17         'charset': 'utf8',

    18         'use_unicode': True,

    19         }

    20     if secrets.DATABASE_USER:

    21         kwargs['user'] = secrets.DATABASE_USER

    22     if secrets.DATABASE_NAME:

    23         kwargs['db'] = secrets.DATABASE_NAME

    24     if secrets.DATABASE_PASSWORD:

    25         kwargs['passwd'] = secrets.DATABASE_PASSWORD

    26     if secrets.DATABASE_HOST.startswith('/'):

    27         kwargs['unix_socket'] = secrets.DATABASE_HOST

    28     elif secrets.DATABASE_HOST:

    29         kwargs['host'] = secrets.DATABASE_HOST

    30     if secrets.DATABASE_PORT:

    31         kwargs['port'] = int(secrets.DATABASE_PORT)

    32     return mysql.connect(**kwargs)