예제 #1
0
def prepare_db(mongo_client):
    """Adds user `admin` and makes the `username` an unique index.
    """
    db = mongo_client.user_database
    users = db.users
    users.ensure_index([("username", ASCENDING)], unique=True)
    #unique:true, dropDups : true
    users.insert({'username': '******', 'saltedpw': pwh('123')}, safe=True)          
"""                                                          

if __name__=='__main__':
    if not(len(sys.argv)==2) or not(sys.argv[-1]=='GO'):
        print desc
        sys.exit(0)
    mongo_client = MongoClient()
    db = mongo_client.user_database
    users = db.users

    # Delete all users
    users.drop()
    prepare_db(mongo_client)
    users.ensure_index([("username", ASCENDING)], unique=True)
    #unique:true, dropDups : true
    users.insert({'username': '******', 'saltedpw':pwh('123')}, safe=True)
    users.insert({'username': '******', 'saltedpw':pwh('123')}, safe=True)
    users.insert({'username': '******', 'saltedpw':pwh('321')}, safe=True)
    users.insert({'username': '******', 'saltedpw':pwh('321')}, safe=True)
    try:
        users.insert({'username': '******', 'saltedpw':pwh('321')}, safe=True)
        print "Error: It seems that I can submit dublicates to the db."
    except DuplicateKeyError, e:
        print "Ok, dublicates not allowed in db."
    print
    print "These are the entries now in the user database"
    usr_sel = users.find()
    for user in usr_sel:
        print user

    python db_demo_populate.py GO
"""

if __name__ == '__main__':
    if not (len(sys.argv) == 2) or not (sys.argv[-1] == 'GO'):
        print desc
        sys.exit(0)
    mongo_client = MongoClient()
    db = mongo_client.user_database
    users = db.users

    # Delete all users
    users.drop()
    prepare_db(mongo_client)
    users.ensure_index([("username", ASCENDING)], unique=True)
    #unique:true, dropDups : true
    users.insert({'username': '******', 'saltedpw': pwh('123')}, safe=True)
    users.insert({'username': '******', 'saltedpw': pwh('123')}, safe=True)
    users.insert({'username': '******', 'saltedpw': pwh('321')}, safe=True)
    users.insert({'username': '******', 'saltedpw': pwh('321')}, safe=True)
    try:
        users.insert({'username': '******', 'saltedpw': pwh('321')}, safe=True)
        print "Error: It seems that I can submit dublicates to the db."
    except DuplicateKeyError, e:
        print "Ok, dublicates not allowed in db."
    print
    print "These are the entries now in the user database"
    usr_sel = users.find()
    for user in usr_sel:
        print user