Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- class UserBook(Resource):
- def __init__(self):
- config = ConfigParser()
- config.read('postgres.ini')
- self.psql_connect = psycopg2.connect(host=config['PostgreSQL']['hostname'],
- user=config['PostgreSQL']['username'],
- password=config['PostgreSQL']['password'],
- dbname=config['PostgreSQL']['database'])
- self.psql_cur = self.psql_connect.cursor()
- def get(self, user_id):
- self.psql_cur.execute(f'select ub_username, ub_email from userbook where ub_id={user_id}')
- result = self.psql_cur.fetchall()
- return {'username': result[0][0],
- 'email': result[0][1]}
- def post(self, username, email=None):
- self.psql_cur.execute('insert into userbook(ub_username, ub_email) values (%s, %s) returning ub_id', (username, email))
- self.psql_connect.commit()
- result = self.psql_cur.fetchone()[0]
- return {'id': result}
- api.add_resource(UserBook, '/userbook/<user_id>', '/userbook/add/<username>') # Route_1
- if __name__ == '__main__':
- app.run(port='5002')
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement