Advertisement
Guest User

Dumb Script

a guest
Dec 29th, 2018
139
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.45 KB | None | 0 0
  1. import sys
  2. import logging
  3. import rds_config
  4. import pymysql
  5. #rds settings
  6. rds_host = "mysqlforlambdatest.cqg8l2snjl9h.us-east-1.rds.amazonaws.com"
  7. name = rds_config.db_username
  8. password = rds_config.db_password
  9. db_name = rds_config.db_name
  10.  
  11. logger = logging.getLogger()
  12. logging.basicConfig()
  13. logger = logging.getLogger('logger')
  14. logger.warning('The system may break down')
  15. logger.setLevel(logging.INFO)
  16.  
  17. try:
  18. conn = pymysql.connect(rds_host, user=name, passwd=password, db=db_name, connect_timeout=5)
  19. except:
  20. logger.error("ERROR: Unexpected error: Could not connect to MySql instance.")
  21. sys.exit()
  22.  
  23. logger.info("SUCCESS: Connection to RDS mysql instance succeeded")
  24. def handler(event, context):
  25. """
  26. This function fetches content from mysql RDS instance
  27. """
  28.  
  29. item_count = 0
  30.  
  31. with conn.cursor() as cur:
  32. cur.execute("create table Employee3 ( EmpID int NOT NULL, Name varchar(255) NOT NULL, PRIMARY KEY (EmpID))")
  33. cur.execute('insert into Employee3 (EmpID, Name) values(1, "Joe")')
  34. cur.execute('insert into Employee3 (EmpID, Name) values(2, "Bob")')
  35. cur.execute('insert into Employee3 (EmpID, Name) values(3, "Mary")')
  36. conn.commit()
  37. cur.execute("select * from Employee3")
  38. for row in cur:
  39. item_count += 1
  40. logger.info(row)
  41. #print(row)
  42. conn.commit()
  43.  
  44. return "Added %d items from RDS MySQL table" %(item_count)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement