Advertisement
Guest User

Untitled

a guest
Nov 28th, 2018
137
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 1.19 KB | None | 0 0
  1. class UserBook(Resource):
  2.  
  3.     def __init__(self):
  4.         config = ConfigParser()
  5.         config.read('postgres.ini')
  6.         self.psql_connect = psycopg2.connect(host=config['PostgreSQL']['hostname'],
  7.                                              user=config['PostgreSQL']['username'],
  8.                                              password=config['PostgreSQL']['password'],
  9.                                              dbname=config['PostgreSQL']['database'])
  10.         self.psql_cur = self.psql_connect.cursor()
  11.  
  12.     def get(self, user_id):
  13.         self.psql_cur.execute(f'select ub_username, ub_email from userbook where ub_id={user_id}')
  14.         result = self.psql_cur.fetchall()
  15.         return {'username': result[0][0],
  16.                 'email': result[0][1]}
  17.  
  18.     def post(self, username, email=None):
  19.         self.psql_cur.execute('insert into userbook(ub_username, ub_email) values (%s, %s) returning ub_id', (username, email))
  20.         self.psql_connect.commit()
  21.         result = self.psql_cur.fetchone()[0]
  22.         return {'id': result}
  23.  
  24. api.add_resource(UserBook, '/userbook/<user_id>', '/userbook/add/<username>')  # Route_1
  25.  
  26.  
  27. if __name__ == '__main__':
  28.     app.run(port='5002')
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement