import json
import psycopg2
import psycopg2.extras
import random

SETTINGS_JSON = 'settings.json'

RANDOM_SOURCE_STR = 'abcdefghijklmnopqrstuvwxyz0123456789'


def getRandomSfid(prefix):
    return prefix + ''.join([random.choice(RANDOM_SOURCE_STR) for x in xrange(15)])

settings = json.load(open(SETTINGS_JSON))

conn = psycopg2.connect(
    host=settings['host'],
    port=settings['port'],
    database=settings['database'],
    user=settings['user'],
    password=settings['password'])
cur = conn.cursor(cursor_factory=psycopg2.extras.DictCursor)

for object in settings['objects']:
    cur.execute('SELECT id, sfid FROM "{0}" WHERE sfid IS NULL OR sfid = \'\''.format(object['name']))

    targets = [{'id':row['id'], 'sfid': getRandomSfid(object['prefix'])} for row in cur]
    for target in targets:
        cur.execute('UPDATE "{0}" SET sfid=%s WHERE id=%s'.format(object['name']), [target['sfid'], target['id']])
    print(object['name'] + ':' + str(targets))

conn.commit()
cur.close()
conn.close()