def createUser():
try:
global service, response
logger.info('Calling createUser')
data = request.json
user = User(username=data.get('username'),
user_type=data.get('user_type'),
name=data['user_details']['name'],
dob=data['user_details']['dob'],
email=data['user_details']['email'],
gender=data['user_details']['gender'],
street=data['user_details']['address']['street'],
province=data['user_details']['address']['province'],
country=data['user_details']['address']['country'],
phonenumber=data['user_details']['phonenumber'])
dashboard = Dashboard(
total_visits=data['dashboard']['total_visits'],
cancelled_visits=data['dashboard']['cancelled_visits'],
upcoming_visits=data['dashboard']['upcoming_visits'],
bookings_history=data.get('bookings_history'),
messages=data['dashboard']['messages'])
if (data.get('user_type') == 'store_owner'):
store = Store(
name=data['store_details']['name'],
establishedon=data['store_details']['establishedon'],
email=data['store_details']['email'],
size=data['store_details']['size'],
street=data['store_details']['street'],
province=data['store_details']['province'],
country=data['store_details']['country'],
phonenumber=data['store_details']['phonenumber'])
response = service.handleService(user, dashboard, store)
response = service.handleService(user, dashboard)
if (response != ""):
return 'User created'
else:
return 'User not created'
except (Exception) as e:
return e.response
def collect_raw_data(src_path,
src_extension,
dst_extension,
is_webshell=True,
check_format=None):
'''
transform file to train_set, with sha256sum
:param src_path: raw data file path
:param src_extension: string or list, raw data extension
:param dst_extension: string, dst file extension
:param check_format: check file format or not, eg. php, jsp, .net, py, rube
:return:
'''
if dst_extension not in global_extens:
raise Exception('dst extension error, must be one of %s' %
str(global_extens))
extens = src_extension if isinstance(src_extension,
list) else src_extension.split(' ')
cur_path = os.path.split(os.path.realpath(__file__))[0]
dst_path = os.path.abspath(
os.path.join(cur_path, os.path.pardir, 'train_set'))
fs = [
os.path.join(root, f) for (root, dirs, files) in os.walk(src_path)
for f in files
]
fs = [f for e in extens for f in fs if f.endswith(e)]
num = len(fs)
for f in fs:
sha256 = hashlib.sha256(open(f, 'rb').read()).hexdigest()
dst_file = os.path.join(dst_path, dst_extension[1:]+'_webshell', sha256+dst_extension) \
if is_webshell else os.path.join(dst_path, dst_extension[1:]+'_normal', sha256+dst_extension)
if os.path.isfile(dst_file):
num = num - 1
else:
shutil.copy(f, dst_file)
logger.info('add %d webshell/files' % num)
def handleService(self, user, dashboard, *store):
if (db_type == 'sqlLite'):
logger.info('sqlLite')
elif (db_type == 'postgresql'):
logger.info('postgresql')
elif (db_type == 'mysql'):
logger.info('mysql')
elif (db_type == 'mangodb'):
logger.info('mangodb')
elif (db_type == 'dynamodb'):
try:
logger.info('dynamodb')
global dao, dynamodb
response = dao.dataAccess(user, dashboard, dynamodb)
return response
except (Exception) as e:
raise
elif (db_type == 'rds'):
logger.info('')
elif (db_type == 'dynamodb'):
logger.info('dynamodb')
import os
from utility.logger import logger
from src.repositories.aws.dynamodb.dynamodb import Dynamodb
from src.dao import Dao
logger = logger()
db_type = os.getenv('DB_TYPE')
logger.info(db_type)
dao = None
dynamodb = None
class Service:
def __init__(self):
global dao, dynamodb
dao = Dao()
dynamodb = Dynamodb()
def handleService(self, user, dashboard, *store):
if (db_type == 'sqlLite'):
logger.info('sqlLite')
elif (db_type == 'postgresql'):
logger.info('postgresql')
elif (db_type == 'mysql'):
logger.info('mysql')
elif (db_type == 'mangodb'):
logger.info('mangodb')
elif (db_type == 'dynamodb'):
try:
logger.info('dynamodb')
def dataAccess(self, user, dashboard, repository, *store):
if (db_type == 'sqlLite'):
logger.info('sqlLite')
elif (db_type == 'postgresql'):
logger.info('postgresql')
elif (db_type == 'mysql'):
logger.info('mysql')
elif (db_type == 'mangodb'):
logger.info('mangodb')
elif (db_type == 'dynamodb'):
try:
logger.info('dynamodb')
item = {
'pk': getattr(user, 'username'),
'sk': getattr(user, 'user_type'),
'type': 'user',
'user_details': user.getUserDetails(),
'dashboard': dashboard.getDashboard()
}
if getattr(user, 'user_type') == "shopper":
booking_history = getattr(dashboard, 'bookings_history')
item.update({'booking_history': booking_history})
elif getattr(user, 'user_type') == "store_owner":
store_details = store.getStoreDetails()
messages = getattr(user, 'messages')
item.update({
'store_details': store_details,
'messages': messages
})
response = repository.newItem(item)
return response
except (Exception) as e:
raise
elif (db_type == 'rds'):
logger.info('')
elif (db_type == 'dynamodb'):
logger.info('dynamodb')
def uploadImage():
logger.info('Calling uploadImage')
return 'Hello, World!'
def getStores():
logger.info('Calling getStores')
return 'Hello, World!'
def getUsers():
logger.info('Calling getUsers')
return 'Hello, World!'
def updateRecord():
logger.info('Calling updateRecord')
return 'Hello, World!'
def deleteUser():
logger.info('Calling deleteUser')
return 'Hello, World!'
def fetchImage():
logger.info('Calling fetchImage')
return 'Hello, World!'
