-
Notifications
You must be signed in to change notification settings - Fork 0
/
tclogin.py
64 lines (50 loc) · 1.6 KB
/
tclogin.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
#coding: utf-8
'''
This module provides login checking for doctors, admin and acounter
'''
import functools
from flask import session, abort
from settings import all_doctors, ADMIN_PASSWORD, ACC_PASSWORD
class User (object):
def __init__(self, login, password, group):
self.login = login
self.password = password
self.group = group
def all_users():
doctors = [User(d.engname, d.password, 'doctor') for d in all_doctors()]
admin = User('admin', ADMIN_PASSWORD, 'admin')
accountant = User('tatiana', ACC_PASSWORD, 'accountant')
return doctors + [admin, accountant]
def _check_user(login, password):
for user in all_users():
if user.login == login and user.password == password:
return user
else:
return None
def user_wrapper(groups=None):
'''
Decorator to check if user loggined
'''
def decorator(function):
@functools.wraps(function)
def wrapper(*args, **kwargs):
if not 'user'in session:
abort(401)
user = _check_user(session['user'].login, session['user'].password)
if not user:
abort(401)
if groups and not user.group in groups:
abort(401)
return function(*args, **kwargs)
return wrapper
return decorator
def loggining(login, password):
'''
If login and password is available writes them into session and returns true else returns False
'''
user = _check_user(login, password)
if user:
session['user'] = user
return user
else:
return False