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