82 lines
2.2 KiB
Python
82 lines
2.2 KiB
Python
import os
|
|
import time
|
|
from scaffold.core.data.select import select_data
|
|
from scaffold.core.data.insert import insert_data
|
|
from scaffold.core.data.update import update_data
|
|
from scaffold.core.data.delete import delete_data
|
|
from scaffold.core.data.sql import query_builder
|
|
|
|
query_builder.query_path = os.path.abspath('./data/sql/')
|
|
|
|
|
|
class create(insert_data):
|
|
table = 'users'
|
|
required = {'email', 'password', 'username', 'first_name', 'last_name', 'created'}
|
|
columns = {'email', 'password', 'username', 'first_name', 'last_name', 'created'}
|
|
columns_optional = {'profile_image'}
|
|
|
|
def calculated_data(self):
|
|
return {'created': time.strftime('%Y-%m-%d %H:%M:%S')}
|
|
|
|
def set(self, data):
|
|
data['created'] = time.strftime('%Y-%m-%d %H:%M:%S')
|
|
super(create, self).set(data)
|
|
|
|
class update_last_login(update_data):
|
|
#~ table = 'users'
|
|
debug = True
|
|
query_str = "update `users` set `last_login`=now() where id=%(user_id)s"
|
|
required = {'user_id'}
|
|
#~ columns = {'id'}
|
|
columns_where = {}
|
|
|
|
class delete_password_reset(delete_data):
|
|
"""clean up expired password resets"""
|
|
table = 'user_password_reset'
|
|
sql_where = 'where DATE_ADD(created, INTERVAL 1 HOUR) < now()'
|
|
required = {}
|
|
|
|
|
|
class create_password_reset(insert_data):
|
|
table = 'user_password_reset'
|
|
required = {'user_id', 'reset_code'}
|
|
|
|
class get_user_by_reset_code(select_data):
|
|
required = {'reset_code'}
|
|
query_file = 'get_user_password_reset.sql'
|
|
columns_where = ['reset_code']
|
|
|
|
class change_password(update_data):
|
|
table = 'users'
|
|
required = {'id', 'password'}
|
|
columns_where = ['password']
|
|
sql_where = 'id=%(id)s'
|
|
|
|
|
|
class get_users(select_data):
|
|
required = {}
|
|
query_file = 'get_users.sql'
|
|
|
|
|
|
class get_user_details(select_data):
|
|
#~ debug = True
|
|
required = {'id'}
|
|
query_file = 'get_user_detail.sql'
|
|
columns_where = {'users.id'}
|
|
|
|
|
|
class get_by_email(select_data):
|
|
required = {'email'}
|
|
query_file = 'get_users.sql'
|
|
columns_where = {'email'}
|
|
|
|
class get_by_username(select_data):
|
|
required = {'email'}
|
|
query_file = 'get_user_credentials.sql'
|
|
columns_where = {'email'}
|
|
|
|
class authorize(select_data):
|
|
required = {'id'}
|
|
query_file = 'get_user_credentials.sql'
|
|
columns_where = {'id'}
|