def get_user(cls, user_id: int = None, email: str = None) -> User:
if user_id:
user_raw = UserDBManager.get_user(user_id)
return User(*user_raw) if user_raw else None
elif email:
user_raw = UserDBManager.get_user_by_email(email)
return User(*user_raw) if user_raw else None
return None
def main():
db_session.global_init("db/blogs.db")
user = User()
user.surname = "Scott"
user.name = "Ridley"
user.age = "21"
user.position = "captain"
user.speciality = "research engineer"
user.address = "module_1"
user.email = "*****@*****.**"
db_sess = db_session.create_session()
db_sess.add(user)
user = User()
user.surname = "Surn"
user.name = "Na"
user.age = "123"
user.position = "cook"
user.speciality = "aboltus"
user.address = "55.5, 66.6"
user.email = "*****@*****.**"
db_sess = db_session.create_session()
db_sess.add(user)
user = User()
user.surname = "Sun"
user.name = "Ame"
user.age = "277"
user.position = "assistant"
user.speciality = "aboltus"
user.address = "module_0"
user.email = "*****@*****.**"
db_sess = db_session.create_session()
db_sess.add(user)
user = User()
user.surname = "Urn"
user.name = "Nam"
user.age = "A12"
user.position = "assistant"
user.speciality = "aboltus"
user.address = "module_777"
user.email = "*****@*****.**"
db_sess = db_session.create_session()
db_sess.add(user)
db_sess.commit()
app.run()
async def login_user(user_id: str, password: str) -> (Optional[User], int):
url = "https://auth-d.edirepository.org/auth/login/pasta"
dn = f"uid={user_id},o=EDI,dc=edirepository,dc=org"
authorization = (base64.b64encode(
f"{dn}:{password}".encode("utf-8"))).decode("utf-8")
headers = {"authorization": "Basic " + authorization}
async with aiohttp.ClientSession() as session:
async with session.get(url=url, headers=headers) as resp:
s = resp.status
if s == status.HTTP_200_OK:
cookies = resp.cookies
auth_token = cookies["auth-token"]
common_name = user_id
return User(user_id, common_name, auth_token), s
elif s == status.HTTP_400_BAD_REQUEST:
msg = "Basic Authorization header not sent in request"
logger.error(msg)
return None, s
elif s == status.HTTP_401_UNAUTHORIZED:
return None, s
elif s == status.HTTP_418_IM_A_TEAPOT:
return None, s
else:
msg = "Unrecognized error occurred - response status code: {s}"
logger.error(msg)
return None, s
def user(name: str):
name = escape(name or "")
user_object = User(name)
if not user_object.exists:
return abort(404)
return render_template("profile.html", title=name, user=user_object)
def delete_user(username) -> User:
user = User()
del_user = User.objects(username=username)
del_user.delete()
del_user = User.objects(username=username)
return del_user
def new_user(token):
global tokens, users
response = requests.post('https://account.keeer.net/api/auth/query_kiuid',
data={
'token':
token,
'sign':
_sign(token,
config['server']['keeer_secret_key'])
},
headers={'User-agent': 'Chrome/78.0.3904.70'})
if response.status_code != 200:
raise Exception('Response error<%s>:%s' %
(response.status_code, response.text))
return False
response = response.json()
if response['status'] == 0:
kiuid = response['result']
if kiuid not in users:
user = User.from_file(kiuid)
user = user if user != None else User(kiuid)
users.update({kiuid: user})
dump_users_to_file()
if token not in tokens:
tokens.update({token: kiuid})
dump_tokens_to_file()
return True
return False
def create_user(name: str, email: str) -> User:
user = User()
user.name = name
user.email = email
user.save()
return user
def register():
message_role = role_user()
form_role = RoleUserForm()
form = RegisterForm()
if form.validate_on_submit():
if form.password.data != form.password_again.data:
return render_template('register.html',
title='Регистрация',
form=form,
message="Пароли не совпадают")
db_sess = db_session.create_session()
if db_sess.query(User).filter(User.email == form.email.data).first():
return render_template('register.html',
title='Регистрация',
form=form,
message="Такой пользователь уже есть",
message_role=message_role,
form_role=form_role)
user = User(
name=form.name.data,
surname=form.surname.data,
age=form.age.data,
email=form.email.data,
role=1,
)
user.set_password(form.password.data)
db_sess.add(user)
db_sess.commit()
return redirect('/login')
return render_template('register.html',
title='Регистрация',
form=form,
message_role=message_role,
form_role=form_role)
def register():
form = RegisterForm()
if form.validate_on_submit():
if form.password.data != form.password2.data:
return render_template('register.html',
title='Регистрация',
form=form,
message="Пароли не совпадают")
session = db_session.create_session()
if session.query(User).filter(User.email == form.email.data).first():
return render_template('register.html',
title='Регистрация',
form=form,
message="Такой пользователь уже есть")
user = User()
user.rank = form.rank.data
user.name = form.name.data
user.email = form.email.data
user.nick = form.nick.data
user.set_password(form.password.data)
user.owned = ''
session.add(user)
session.commit()
return redirect('/login')
return render_template("register.html", form=form, title='Регистрация')
def reqister():
form = RegisterForm()
if form.validate_on_submit():
if form.password.data != form.password_again.data:
return render_template('register.html',
title='Регистрация',
form=form,
message="Пароли не совпадают")
db_sess = db_session.create_session()
if db_sess.query(User).filter(User.email == form.email.data).first():
return render_template('register.html',
title='Регистрация',
form=form,
message="Такой пользователь уже есть")
user = User(name=form.name.data, email=form.email.data, is_admin=False)
user.set_password(form.password.data)
db_sess.add(user)
db_sess.commit()
f = open('log/log.txt', 'a')
data = time.ctime(time.time())
f.write(f'register {data} | {user.name}\n')
f.close()
return redirect('/login')
return render_template('register.html', title='Регистрация', form=form)
def register():
form = RegisterForm()
if form.validate_on_submit():
if form.password.data != form.password_again.data:
return render_template('user_reg.html',
title='Registration',
form=form,
message="Passwords are not the same")
session = db_session.create_session()
if session.query(User).filter(User.login == form.login.data).first():
return render_template('user_reg.html',
title='Registration',
form=form,
message="This user already exists")
act_id = hash_password(form.login.data)
user = User(name=form.name.data,
login=form.login.data,
age=form.age.data,
email=form.email.data,
is_active=False,
act_id=act_id,
likes=0,
videos=0)
user.set_password(form.password.data)
session.add(user)
session.commit()
send_mail(form.email.data, act_id)
return render_template(
'activation.html',
message=
"A verify message was sent to your email address. Please follow the link in the message"
)
return render_template('user_reg.html', title='Registration', form=form)
def login():
if request.method == "GET":
return render_template("auth/login.html", title="Login")
username = escape(request.form.get("name", ""))
password = escape(request.form.get("password", ""))
captcha = request.form.get("g-recaptcha-response", False)
user_object = User(username)
logCaptcha(captcha)
if assertConditions(
user_object, request.form.get("remember", True),
zip([
captcha, user_object.exists,
check_password(user_object, password)
], [
"Please complete the reCAPTCHA.", "Invalid username provided.",
"Username or password is incorrect."
])):
return redirect(url_for("main.home"))
else:
return render_template("auth/login.html",
title="Login",
username=username,
password=password)
def signup():
"""signup page"""
if current_user.is_authenticated:
return redirect(f'/id{current_user.id}')
form = AuthorizeForm()
if form.validate_on_submit():
# if password and control password are equal
# and username are not taken, add the user to data base
if form.password.data == form.password_control.data:
if not db_sess.query(User).filter(
User.username == form.username.data).first():
user = User()
user.username = form.username.data
user.create_password_hash(form.password.data)
db_sess.add(user)
db_sess.commit()
return redirect('/login')
return render_template('signup.html',
form=form,
message="Такое имя пользователя уже занято")
return render_template('signup.html',
form=form,
message="Пароли не совпадают")
return render_template('signup.html', form=form,
message="") # do not remove 'message' argument here
def register():
form = RegisterForm()
if form.validate_on_submit():
if form.password.data != form.password_again.data:
return render_template('register.html',
title='Регистрация',
form=form,
message="Пароли не совпадают")
db_sess = db_session.create_session()
if db_sess.query(User).filter(User.email == form.email.data).first():
return render_template(
'register.html',
title='Регистрация',
form=form,
message="Пользователь с таким Email уже зарегистрирован!")
user = User()
if not db_sess.query(User).count():
user.is_admin = True
user.email = form.email.data
user.set_password(form.password.data)
db_sess.add(user)
db_sess.commit()
login_user(user, remember=True)
return redirect("/")
return render_template('register.html', form=form)
def register():
form = RegisterForm()
if form.validate_on_submit():
if form.password.data != form.password_again.data:
params = {
'title': 'Регистрация',
'form': form,
'message': 'Пароли не совпадают'
}
return render_template('register.html', **params)
db_sess = db_session.create_session()
if db_sess.query(User).filter(User.email == form.email.data).first():
params = {
'title': 'Регистрация',
'form': form,
'message': 'Пользователь с такой почтой уже существует'
}
return render_template('register.html', **params)
user = User()
user.username = form.username.data
user.email = form.email.data
user.set_password(form.password.data)
db_sess.add(user)
db_sess.commit()
login_user(user, remember=True)
return redirect('/')
params = {
'title': 'Регистрация',
'form': form,
}
return render_template('register.html', **params)
def clk_save(self):
user = User()
user.setUserType(self.type.get())
user.setUserStatus(int(self.status.get()))
user.setName(self.name.get())
user.setEmail(self.uemail.get())
user.setContact(self.ucontact.get())
user.setAddress(self.uaddress.get(1.0, 'end'))
user.setGender(int(self.gender.get()))
if (self.addeditflag == 'add'):
user.setUserName(self.uname.get())
user.setPassword('anamika')
if (UMS_services.add(user)):
messagebox.showinfo(
'create new user',
'Added successfully with password anamika')
else:
messagebox.showerror('create new user', 'Unable to add.')
elif (self.addeditflag == 'edit'):
self.uid.config(state='normal')
user.setUserId(int(self.uid.get()))
self.uid.config(state='disabled')
if (UMS_services.update(user)):
messagebox.showinfo('update', 'Updated Sucessfully')
else:
messagebox.showerror('update', 'Unable to update')
self.s.config(state='disabled')
self.a.config(state='normal')
self.e.config(state='normal')
self.userlist = UMS_services.view()
if (self.addeditflag == 'add'):
self.curr_index = len(self.userlist) - 1
self.addeditflag = 'view'
self.showRecord()
def create_test_user(name, email, hashed_pw, active_status):
u = User()
u.name = name
u.email = email
u.hashed_pw = hashed_pw
u.is_active = active_status
return u
def email_verification():
try:
form = EmailVerificationForm()
if form.validate_on_submit():
db_sess = db_session.create_session()
if session['Verification Code'] == form.code.data:
user = User(
name=session['User Nickname'],
email=session['User Email'],
)
user.set_password(session['User Password'])
db_sess.add(user)
db_sess.commit()
logging.info('Added a new User')
return redirect('/login')
else:
return render_template('email_verification.html',
title='Подтверждение',
form=form,
message='Неверный код подтверждения')
return render_template('email_verification.html',
title='Подтверждение',
form=form)
except Exception:
return render_template('error.html')
def getcounter():
user = User()
counter = 1
firstuser = User.objects().order_by('-_id').first()
if firstuser:
counter = (firstuser._id) + 1
return counter
def register_user(data) -> User:
user = User()
user._id = UserModel.getcounter()
user.email = data['email']
user.username = data['username']
user.password = encrypt_password(data['password'])
user.role = data['role']
user.save()
return user
def main():
db_session.global_init("db/mars.db")
cap = User()
cap.surname = 'Scott'
cap.name = 'Ridley'
cap.age = 21
cap.position = 'captain'
cap.speciality = 'research engineer'
cap.address = 'module_1'
cap.email = '*****@*****.**'
db_sess = db_session.create_session()
db_sess.add(cap)
user1 = User()
user1.surname = 'Black'
user1.name = 'Adam'
user1.age = 19
user1.position = 'user'
user1.speciality = 'doctor'
user1.address = 'module_3'
user1.email = '*****@*****.**'
db_sess.add(user1)
user2 = User()
user2.surname = 'Green'
user2.name = 'Jereme'
user2.age = 25
user2.position = 'user'
user2.speciality = 'scientist'
user2.address = 'module_1'
user2.email = '*****@*****.**'
db_sess.add(user2)
user3 = User()
user3.surname = 'Braun'
user3.name = 'Simon'
user3.age = 22
user3.position = 'user'
user3.speciality = 'cooker'
user3.address = 'module_2'
user3.email = '*****@*****.**'
db_sess.add(user3)
db_sess.commit()
def post(self):
session = db_session.create_session()
args = parser1.parse_args()
user = User(
nickname=args['nickname'],
status_id=args['status_id']
)
user.set_password(args['password'])
session.add(user)
session.commit()
def create_users():
s = create_session()
user = User(
nickname='admin',
status_id=1,
)
user.set_password('admin')
s.add(user)
for i in range(1, 11):
nick = f'user{i}'
user = User(
nickname=nick,
status_id=2,
)
user.set_password(nick)
s.add(user)
s.commit()
def post(self):
args = parser.parse_args()
session = db_session.create_session()
user = User(
surname=args['username'],
email=args['email'],
)
user.set_password(args['hashed_password'])
session.add(user)
session.commit()
return jsonify({'success': 'OK'})
def post(self):
args = parser.parse_args()
session = db_session.create_session()
user = User()
User.id = args['id'],
User.surname = args['surname'],
User.name = args['name'],
User.age = args['age'],
session.add(user)
session.commit()
return jsonify({'success': 'OK'})
async def create_account(name: str, email: str, password: str) -> User:
user = User()
user.email = email
user.name = name
user.hash_password = crypto.hash(password, rounds=172_434)
async with db_session.create_async_session() as session:
session.add(user)
await session.commit()
return user
def add_users(user_id, nickname, prompt, last_update):
user = User()
user.user_id = user_id
user.nickname = nickname
user.prompt = prompt
user.last_update = last_update
user.cor_answ = 0
user.all_questions = 0
session = db_session.create_session()
session.add(user)
session.commit()
def register():
# Загрузка формы регистрации
form = RegisterForm()
store_settings = get_store_settings()
store_settings['title'] = 'Регистрация'
# Проверка правильности введённых данных
if form.validate_on_submit():
if form.password.data != form.password_again.data:
return render_template('register.html',
form=form,
message="Пароли не совпадают",
**store_settings)
if check_password(form.password.data):
return render_template('register.html',
form=form,
message=check_password(form.password.data),
**store_settings)
if not form.age.data.isnumeric():
return render_template('register.html',
form=form,
message="Неверный возраст",
**store_settings)
db_sess = db_session.create_session()
if db_sess.query(User).filter(User.email == form.email.data).first():
return render_template('register.html',
form=form,
message="Такой пользователь уже есть",
**store_settings)
# Создание нового пользователя и занесение в базу данных
user = User(name=form.name.data,
email=form.email.data,
surname=form.surname.data,
age=int(form.age.data),
address=form.address.data,
got_bonus=0)
user.set_password(form.password.data)
db_sess.add(user)
db_sess.commit()
# Создание json-файла с данными нового пользователя: счёт, корзина и заказы
with open(f'accounts/user_{user.id}.json', 'w',
encoding='utf-8') as jsonfile:
data = {
'shopping_cart': {
'items': [],
'summary': {}
},
'orders': {},
'currencies': dict()
}
for i in db_sess.query(Currency).all():
data['currencies'][i.id] = 0
json.dump(data, jsonfile)
return redirect("/")
return render_template('register.html', form=form, **store_settings)
def student(message):
if message.chat.id in admin:
markup = data.buttons.button_admin()
else:
markup = data.buttons.button_days()
msg = bot.send_message(message.chat.id, "Спасибо, " + message.chat.first_name, reply_markup=markup)
user = User()
user.message(msg)
user.getNumberGroup(message.text)
db_worker = SQLStudents(database_name)
db_worker.insert_students(user)
db_worker.close()
def signup():
if request.method == "GET":
return render_template("auth/signup.html", title="Sign Up")
email = escape(request.form.get("email", ""))
username = escape(request.form.get("name", ""))
password = escape(request.form.get("password", ""))
captcha = request.form.get("g-recaptcha-response", False)
user_object = User(username)
logCaptcha(captcha)
if assertConditions(
user_object, request.form.get("remember", True),
zip([
request.form.get("terms", False), captcha,
not user_object.exists,
Checker(username).is_valid,
validate_email(email,
check_regex=True,
use_blacklist=True,
check_mx=False,
debug=False),
len(
list(
user_object.database.where("email", "==",
email).stream())) == 0
], [
"Please accept the Terms of Service.",
"Please complete the reCAPTCHA.",
"Username is already in use.", "Username not allowed.",
"Invalid email.", "Email is already in use."
])):
user_object.update(
team="Solo",
email=email,
submitted=False,
password=hash_password(password),
created_date=datetime.datetime.now(),
image_url=
"https://cdn.pixabay.com/photo/2017/11/10/05/48/user-2935527_1280.png",
)
current_members = solo_team["members"]
current_members.append(username)
solo_team.update(members=current_members)
return redirect(url_for("main.home"))
else:
return render_template("auth/signup.html",
title="Sign Up",
email=email,
username=username,
password=password)
