def generateAdmin():
teacher = generateTeacher(True)
teachers.append(teacher)
admin = Admin()
admin.personID = teacher.personID
admin.insert()
def getAdmin(self):
global g_Database
rows = g_Database.fetchAll('SELECT * FROM admin WHERE personID=' +
str(self.personID))
if (not len(rows)):
return None
admin = Admin()
admin.read_row(rows[0])
return admin
def get(self):
if not auth_admin():
self.response.out.write("Acesso negado")
return
email = cgi.escape(self.request.get('email')).decode("utf-8")
charid = int(cgi.escape(self.request.get('charid')))
charname = cgi.escape(self.request.get('charname')).decode("utf-8")
logging.info("==> %s" % charname)
genre = int(cgi.escape(self.request.get('genre')))
user = Admin(email=email, charId=charid, charName=charname, genre=genre)
user.put()
self.response.out.write("Cadastro efetuado com sucesso")
def create_account():
admin = Admin().from_json(json.dumps(g.data))
existed_username_count = Admin.objects(username=admin['username']).count()
if existed_username_count > 0:
message['message'] = 'Admin already existed'
return Response(json.dumps(message), status=400, mimetype='application/json')
admin.password = hashing_password(admin.password)
is_save = admin.save()
if is_save is None:
message['message'] = 'Failed to create new admin account'
return Response(json.dumps(message), status=400, mimetype='application/json')
message['message'] = 'Admin successfully created'
return Response(json.dumps(message), status=200, mimetype='application/json')
def get_one_admin(username):
admin = Admin.objects(username=username).exclude('id', 'password')
if admin.count() <= 0:
message['message'] = "Admin is not found"
return Response(json.dumps(message), status=404, mimetype='application/json')
admin = admin.first()
return Response(admin.to_json(), status=200, mimetype='application/json')
def get_admin():
"""Load admin into database."""
admin = Admin(username="******", password=1234)
db.session.add(admin)
db.session.commit()
def delete_account(username):
admin = Admin.objects(username=username).first()
if admin is None:
message['message'] = "Account does not exist"
return Response(json.dumps(message), status=400, mimetype='application/json')
admin.delete()
message['message'] = 'Account is successfully deleted'
return Response(json.dumps(message), status=200, mimetype='application/json')
def log_in(self):
username = flask.request.form.get("username")
password = flask.request.form.get("password")
if Admin.authenticate(self._db, username, password):
session["username"] = username
return redirect(url_for('admin_view'))
else:
return INVALID_CREDENTIALS
def add_superuser(self):
from model import Admin
admin = Admin()
admin.ADid = '3404cf38-c3cf-401f-8ba7-f8ce040f064f'
admin.ADnum = '123'
admin.ADpassword = '******'
admin.ADname = 'feng'
admin.ADlevel = 1
admin.ADheaderimg = 'https://timgsa.baidu.com/timg?image&quality=80&size=b9999_10000&sec=1540919391&di=91c1ae656341d5814e63280616ad8ade&imgtype=jpg&er=1&src=http%3A%2F%2Fimg.zcool.cn%2Fcommunity%2F0169d55548dff50000019ae9973427.jpg%401280w_1l_2o_100sh.jpg'
self.session.add(admin)
self.session.commit()
def register():
"""Register a user."""
email = request.form.get('email')
first_name = request.form.get('first_name')
last_name = request.form.get('last_name')
password = request.form.get('password')
db.session.add(Admin(email, first_name, last_name, hash(password)))
db.session.commit()
return redirect('/admin/login')
def login():
if request.method == 'POST':
try:
user = Admin.get(Admin.user == request.form.get('username'))
except DoesNotExist:
user = None
if user and pbkdf2_sha256.verify(request.form.get('password'),
user.password):
session['username'] = request.form['password']
return redirect(url_for('create'))
return render_template('login.jinja2',
error="Incorrect username or password")
return render_template('login.jinja2')
def admin_add_user(role, fields):
if role.lower() == 'student':
user = Student()
elif role.lower() == 'teacher':
user = Teacher()
elif role.lower() == 'admin':
user = Admin()
else:
raise APIError('Invalid role')
fields['password'] = hash_password(fields['password'])
for key, value in fields.items():
if not value:
continue
user.__setattr__(key, value)
db.session.add(user)
db.session.commit()
return user.dict
# This line will allow you "upgrade" an existing database by
# dropping all existing tables from it.
db.drop_tables([Donor, Donation, Admin])
#
db.create_tables([Donor, Donation, Admin])
alice = Donor.create(name="Alice")
# alice.save()
bob = Donor.create(name="Bob")
# bob.save()
charlie = Donor.create(name="Charlie")
# charlie.save()
donors = [alice, bob, charlie]
for x in range(30):
Donation.create(donor=random.choice(donors),
value=random.randint(100, 10000))
user1 = Admin.create(user="******", password=pbkdf2_sha256.hash("password"))
user2 = Admin.create(user="******", password=pbkdf2_sha256.hash("123456"))
# try:
# user = Admin.get(Admin.user == 'admin')
# except DoesNotExist:
# user = None
# print(user)
user = self.user_repo.find_user_by_email(login.email)
if user != None and user.check_password(login.password):
self.logged_user = user
return True
else:
self.logged_user = None
return False
def get_logged_user(self) -> Optional[User]:
return self.logged_user
if __name__ == '__main__':
users = [
Author('Ivan Petrov', '*****@*****.**', 'ivanp123'),
Admin('Admin Admin', '*****@*****.**', 'admin123', '35928976564'),
Admin('Nadezda Hristova', '*****@*****.**', 'nadia123',
'3592754632'),
]
user_repo = UserRepository(users)
print(f'Repository has {len(user_repo)} users.')
user_repo.add_user(
Author('Hrisitna Dimitrova', '*****@*****.**', 'hristina',
'expert'))
user_repo.add_user(
Author('Ivan Pavlov', '*****@*****.**', 'ivan123', 'professional'))
for user in user_repo:
print('+', user)
print(f'Repository has {len(user_repo)} users.')
# SA Configuration
engine = create_engine('sqlite:///admin.db')
db = Session(engine)
Base_admin.metadata.create_all(engine)
# Save Admin info into DB
username = input("Username: "******"" or username in admin_names:
username = input("Try another username: "******"Password: "******"" or password_check(password) != True:
password = input("Try another password: "******"Confirm your password: "******"" or confirm != password:
confirm = input("Try again: ")
p_hash = generate_password_hash(password)
new_admin = Admin(username=username, hashed_password=p_hash)
db.add(new_admin)
db.commit()
print("Admin created!")
db.close()
def initdb_command():
# Drop the database and create the new one
db.drop_all()
db.create_all()
# Hardcode the admin account
db.session.add(
Admin(username='******',
pw_hash=generate_password_hash('admin'),
email='*****@*****.**',
description='I am an admin ;)'))
# Hardcode `Human` student
db.session.add(
Student(username='******',
pw_hash=generate_password_hash('123456'),
email='*****@*****.**',
description='I am a student',
first_name='Human',
last_name='Namuh',
year=Year.SENIOR,
major='CS',
minor='LIFE',
gender=Gender.OTHER,
gpa=4.0,
work_status=WorkStatus.USCITIZEN,
clearance=Clearance.NONE,
search_status=SearchStatus.OPEN,
phone='4128933928'))
# Hardcode `Ariana` student
db.session.add(
Student(username='******',
pw_hash=generate_password_hash('ariana'),
email='*****@*****.**',
description='I am a student in CS',
first_name='ariana',
last_name='grande',
year=Year.JUNIOR,
major='CS',
minor='IS',
gender=Gender.OTHER,
gpa=3.88,
work_status=WorkStatus.USCITIZEN,
clearance=Clearance.NONE,
search_status=SearchStatus.OPEN,
phone='4129339992'))
# hardcode organization(s)
db.session.add(
Organization(username='******',
pw_hash=generate_password_hash('fresa'),
email='*****@*****.**',
description='Dance together!',
name='Fresa'))
db.session.add(
Organization(username='******',
pw_hash=generate_password_hash('ksa'),
email='*****@*****.**',
description='Koreans squads!',
name='KSA'))
db.session.add(
Organization(username='******',
pw_hash=generate_password_hash('asa'),
email='*****@*****.**',
description='Asian squads!',
name='ASA'))
db.session.add(
Organization(username='******',
pw_hash=generate_password_hash('wics'),
email='*****@*****.**',
description='Women in CS!',
name='WiCS'))
db.session.add(
Organization(username='******',
pw_hash=generate_password_hash('globaltie'),
email='*****@*****.**',
description='Global Tie!',
name='GlobalTie'))
# Hardcode company(s)
db.session.add(
Company(username='******',
pw_hash=generate_password_hash('google'),
email='*****@*****.**',
description='OH YESSSSS',
name='GOOGLE'))
db.session.commit()
print('Initialized the database')