class Controller(object): def __init__(self): self._connection = PostgresConnection().get_connection() self._cursor = self._connection.cursor(cursor_factory=DictCursor) self._create_tables() self._code_usages_blank_model = CodeUsagesBlankModel(self._connection) self._notarius_model = NotariusModel(self._connection) self._blank_model = BlankModel(self._connection) self._usages_register_model = UsagesRegisterModel(self._connection) self._users_model = UsersModel(self._connection) self._journal_actions_model = JournalActionsModel(self._connection) self._verifications_register_model = VerificationsRegisterModel( self._connection) self._users_model.generate_data(5000) self._notarius_model.generate_data(5000) self._blank_model.generate_data(5000) self._usages_register_model.generate_data(5000) self._journal_actions_model.generate_data(5000) self._verifications_register_model.generate_data(5000) self._code_usages_blank_model.generate_data(5000) # self._users_model.generate_data(100) @property def connection(self): return self._connection def _create_tables(self): file_path = path.join(path.dirname(path.abspath(__file__)), '../create_tables.sql') with open(file_path, 'r') as f: sql = f.read() self._cursor.execute(sql) self._connection.commit()
def generate(): if roles_required(["admin", "registrar"]) == 400: return jsonify({"msg": "no access"}), 400 if not request.is_json: return jsonify({"msg": "Missing JSON in request"}), 400 try: num = request.json['number'] users_model = UsersModel(PostgresConnection().get_connection()) notarius_model = NotariusModel(PostgresConnection().get_connection()) blank_model = BlankModel(PostgresConnection().get_connection()) usages_register_model = UsagesRegisterModel( PostgresConnection().get_connection()) journal_actions_model = JournalActionsModel( PostgresConnection().get_connection()) verifications_register_model = VerificationsRegisterModel( PostgresConnection().get_connection()) code_usages_blank_model = CodeUsagesBlankModel( PostgresConnection().get_connection()) users_model.generate_data(num) notarius_model.generate_data(num) blank_model.generate_data(num) usages_register_model.generate_data(num) journal_actions_model.generate_data(num) verifications_register_model.generate_data(num) code_usages_blank_model.generate_data(num) except Exception as e: return jsonify({"msg": str(e)}), 400 return jsonify({"msg": "data generated"}), 200