equal
deleted
inserted
replaced
|
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) |