60 lines
1.5 KiB
Python
60 lines
1.5 KiB
Python
import os
|
|
import mysql.connector as con
|
|
from mysql.connector.errors import InterfaceError
|
|
|
|
GitHubWorkflow = True
|
|
|
|
try:
|
|
print(os.environ['GITHUB_ACTIONS'])
|
|
except:
|
|
GitHubWorkflow = False
|
|
|
|
def returnValue(key):
|
|
return os.environ[key]
|
|
|
|
if GitHubWorkflow:
|
|
host = returnValue("CURIE_HOST")
|
|
db = returnValue("CURIE_DB")
|
|
user = returnValue("CURIE_USER")
|
|
port = returnValue("CURIE_PORT")
|
|
password = returnValue("CURIE_PASSWORD")
|
|
fromaddr = returnValue("CURIE_EMAIL")
|
|
emailServer = returnValue("CURIE_EMAIL_SERVER")
|
|
emailPort = returnValue("CURIE_EMAIL_PORT")
|
|
emailPassword = returnValue("CURIE_EMAIL_PASSWORD")
|
|
else:
|
|
import configparser
|
|
config = configparser.ConfigParser()
|
|
config.read('config.ini')
|
|
try:
|
|
config['DATABASE']
|
|
except KeyError:
|
|
config.read("../config.ini")
|
|
v = config['DATABASE']
|
|
|
|
host = v['HOST']
|
|
db = v['NAME']
|
|
user = v['USER']
|
|
password = v['PASSWORD']
|
|
port = v['PORT']
|
|
fromaddr = config['SMTP']['EMAIL']
|
|
emailServer = config['SMTP']['SERVER']
|
|
emailPort = config['SMTP']['PORT']
|
|
emailPassword = config['SMTP']['PASSWORD']
|
|
|
|
|
|
try:
|
|
mycon = con.connect(host=host,user=user,password=password,port=port,database=db)
|
|
except InterfaceError:
|
|
print("Could not connect to the database")
|
|
sys.exit(1)
|
|
|
|
mycursor = mycon.cursor()
|
|
|
|
print("Removing Test Query")
|
|
jobID = "l9xo2isr98oepcia"
|
|
q = 'delete from curieweb where id="%s"' % (jobID)
|
|
mycursor.execute(q)
|
|
mycon.commit()
|
|
|
|
print("Database working perfectly") |