Guest User

Untitled

a guest
Nov 22nd, 2018
181
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.76 KB | None | 0 0
  1. import boto3
  2. import psycopg2
  3. import os
  4.  
  5. ssm = boto3.client('ssm',region_name='eu-central-1')
  6. db_password_parameter = ssm.get_parameter(Name=f"/ENV/PROJECT/main/db_master_password", WithDecryption=True)
  7. db_url_parameter = ssm.get_parameter(Name=f"/ENV/PROJECT/main/db_url", WithDecryption=True)
  8. db_password = db_password_parameter['Parameter']['Value']
  9. db_url = db_url_parameter['Parameter']['Value']
  10.  
  11. conn = psycopg2.connect(dbname="postgres", user="USER", password=db_password, host=db_url)
  12. cur = conn.cursor()
  13.  
  14. cur.execute("SELECT datname FROM pg_database;")
  15. print(cur.fetchall())
  16.  
  17. conn.set_isolation_level(0)
  18. try:
  19. cur.execute("drop DATABASE PROJECT;")
  20. except:
  21. print('problem create')
  22.  
  23. try:
  24. cur.execute("CREATE DATABASE PROJECT;")
  25. except:
  26. print('problem drop')
Add Comment
Please, Sign In to add comment