Пример #1
0
import sys

import init_env
import settings
import utils
from db_client import Advertiser, DBClient

if __name__ == "__main__":
    try:
        assert len(sys.argv) == 4
    except:
        print "usage : %s [username] [password] [email]"%sys.argv[0]
        exit(-1)

    name = sys.argv[1]
    pwd = utils.gen_secret(sys.argv[2])
    email = sys.argv[3]

    db = DBClient(settings.db_user, settings.db_password,
        settings.db_host, settings.db_name)
    l = len(db.select_all(Advertiser))

    user = Advertiser(name=name, password=pwd, email=email)
    db.do_save(user)
    assert len(db.select_all(Advertiser)) == l + 1