text-display/script/adduser.py

24 lines
714 B
Python
Raw Normal View History

2022-05-18 12:10:30 +00:00
import collections
from pymongo import MongoClient
from dotenv import load_dotenv
from os import getenv
import xxhash
import getpass
load_dotenv()
client=MongoClient(getenv("MONGO"))
db = client.get_database(getenv("DB"))
def add_user():
username = input('username ? ')
password = getpass.getpass('password ? ')
pass_retry = getpass.getpass('Re-type your password ? ')
if password != pass_retry:
print("not correct password.")
add_user()
hashed_pass = xxhash.xxh64(password, 5).hexdigest()
collection = db["users"]
print(f"username: {username}, password: {hashed_pass}")
collection.update_one({ "name": username}, { "$set": {"pass": hashed_pass}})
add_user()