Spaces:
Sleeping
Sleeping
File size: 1,246 Bytes
b5f6a08 |
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 |
from dataclasses import dataclass
import pandas as pd
import os
@dataclass
class Users:
db_file: str = 'users.csv'
current_path: str = os.path.abspath(os.path.dirname(__file__))
def get_db(self):
try:
db = pd.read_csv(os.path.join(self.current_path, self.db_file))
except FileNotFoundError:
db = self.create_db()
return db
def create_db(self):
db = pd.DataFrame(columns=['login', 'password'])
db = db.append({'login': 'admin', 'password': 'password'}, ignore_index=True)
db.to_csv(os.path.join(self.current_path, self.db_file), index=False)
return db
def exists(self, login: str, password: str) -> bool:
db = self.get_db()
if db.loc[db.login == login].shape[0] != 0:
if db.loc[db.login == login, 'password'].values[0] == password:
return True
return False
def is_admin(self, login: str, password: str) -> bool:
db = self.get_db()
if db.loc[db.login == login, 'login'].values[0] == 'admin' and \
db.loc[db.login == login, 'password'].values[0] == password:
return True
else:
return False
|