diff --git a/dev-prg/DCComics/dccomicsinsert.py b/dev-prg/DCComics/dccomicsinsert.py new file mode 100644 index 0000000..683f37d --- /dev/null +++ b/dev-prg/DCComics/dccomicsinsert.py @@ -0,0 +1,43 @@ +import mysql.connector as db +from mysql.connector import Error + +class database: + def __init__(self, host, user, password, database): + self.host = host + self.user = user + self.password = password + self.database = database + self.connect() + + def connect(self): + try: + self.connection = db.connect(host=self.host, database=self.database, user=self.user, password=self.password) + if self.connection.is_connected(): + db_Info = self.connection.get_server_info() + print("Connected to MySQL Server version ", db_Info) + cursor = self.connection.cursor() + cursor.execute("select database();") + record = cursor.fetchone() + print("You're connected to database: ", record) + cursor.close() + except Error as e: + print("Error while connecting to MySQL", e) + exit + def query(self, query): + cursor = self.connection.cursor() + cursor.execute(query) + return cursor + def close(): + self.connection.close() + +conn = database("127.0.0.1", "comics", "123456789", 'comics') +if conn.connection.is_connected(): + charname = input("Character Name: ") + weapon = input("Weapon: ") + cr = conn.query("SELECT * FROM studio;") + for (id, name) in cr: + print("{}: {}".format(id, name)) + cr.close() + stnum = input("Studio Number:") + res.conn.query("INSERT INTO `charact` (`name`, `weapon`, `studio_id`) VALUES ('{}', '{}', '{}');".format(charname,weapon,stnum)) + res.close() \ No newline at end of file