| Index: sitescripts/urlfixer/bin/forceDomains.py |
| =================================================================== |
| --- a/sitescripts/urlfixer/bin/forceDomains.py |
| +++ b/sitescripts/urlfixer/bin/forceDomains.py |
| @@ -18,38 +18,38 @@ |
| import MySQLdb |
| from sitescripts.utils import get_config, setupStderr |
| -""" |
| +''' |
| This script adds domains supplied on the command line to the "correct domains" |
| list permanently. This is useful for less popular domains that are commonly |
| affected by false positives. |
| -""" |
| +''' |
| def forceDomains(domains): |
| db = _get_db() |
| for domain in domains: |
| cursor = db.cursor(MySQLdb.cursors.DictCursor) |
| - cursor.execute("""INSERT INTO domains(domain, forceinclusion) VALUES (%s, 1) |
| - ON DUPLICATE KEY UPDATE forceinclusion = 1""", domain) |
| + cursor.execute('''INSERT INTO domains(domain, forceinclusion) VALUES (%s, 1) |
| + ON DUPLICATE KEY UPDATE forceinclusion = 1''', domain) |
| db.commit() |
| def _get_db(): |
| - database = get_config().get("urlfixer", "database") |
| - dbuser = get_config().get("urlfixer", "dbuser") |
| - dbpasswd = get_config().get("urlfixer", "dbpassword") |
| - if os.name == "nt": |
| + database = get_config().get('urlfixer', 'database') |
| + dbuser = get_config().get('urlfixer', 'dbuser') |
| + dbpasswd = get_config().get('urlfixer', 'dbpassword') |
| + if os.name == 'nt': |
| return MySQLdb.connect(user=dbuser, passwd=dbpasswd, db=database, |
| - use_unicode=True, charset="utf8", named_pipe=True) |
| + use_unicode=True, charset='utf8', named_pipe=True) |
| else: |
| return MySQLdb.connect(user=dbuser, passwd=dbpasswd, db=database, |
| - use_unicode=True, charset="utf8") |
| + use_unicode=True, charset='utf8') |
| if __name__ == '__main__': |
| setupStderr() |
| if len(sys.argv) <= 1: |
| - print >>sys.stderr, "Please specify the domain names as command line parameters" |
| + print >>sys.stderr, 'Please specify the domain names as command line parameters' |
| sys.exit(1) |
| forceDomains(sys.argv[1:]) |