def add_user():
db = open_db()
repo = UserRepo(db)
user_dict = json.loads(request.data)
user = models.User(str(uuid.uuid4()), user_dict['first_name'], user_dict['last_name'], str(time.time()))
user_inserted = repo.add_user(user)
close_db(db)
return "ok"
class BackupEndpointDataUseCase():
""" Use case for adding to the database de post data obtained from the reddit endpoint """
def __init__(self):
self.user_repo = UserRepo()
self.post_repo = PostRepo()
def execute(self, request):
response = Response()
try:
for post in [p['data'] for p in request.endpoint_data['data']['children']]:
user_id = self.user_repo.add_user(User(name = post['author'], ups = post['ups'], comments = post['num_comments']))
created_time = datetime.fromtimestamp(post['created_utc'], tz = pytz.timezone(request.timezone))
self.post_repo.add_post(Post(title = post['title'], ups = post['ups'], comments = post['num_comments'], created = created_time, author = user_id))
return response
except Exception as e:
response.add_exception_error(e)
return response