Add main.py

main.py

@@ -0,0 +1,113 @@
+import mysql.connector
+import csv
+import smtplib
+from email.mime.text import MIMEText
+from email.mime.multipart import MIMEMultipart
+from email.mime.base import MIMEBase
+from email import encoders
+import os
+
+
+def connect_db():
+    try:
+        conn = mysql.connector.connect(
+            host='192.168.145.196',
+            user='python_sqlcsv_nosecure',
+            password='HBMqyUVM',
+            database='smsgw_control',
+            auth_plugin='mysql_native_password',
+            ssl_disabled=True
+        )
+        return conn
+    except mysql.connector.Error as err:
+        print(f"Error connecting to DB: {err}")
+        return None
+
+
+def run_query(conn, query):
+    cursor = conn.cursor()
+    cursor.execute(query)
+    rows = cursor.fetchall()
+    columns = [desc[0] for desc in cursor.description]
+    return rows, columns
+
+
+def write_csv(filename, rows, columns):
+    with open(filename, mode='w', newline='') as file:
+        writer = csv.writer(file)
+        writer.writerow(columns)
+        writer.writerows(rows)
+
+
+def send_email(sender_email, receiver_email, subject, body, filename, smtp_server, smtp_port, smtp_user, smtp_password):
+    message = MIMEMultipart()
+    message['From'] = sender_email
+    message['To'] = receiver_email
+    message['Subject'] = subject
+
+    message.attach(MIMEText(body, 'plain'))
+
+    with open(filename, "rb") as attachment:
+        part = MIMEBase('application', 'octet-stream')
+        part.set_payload(attachment.read())
+        encoders.encode_base64(part)
+        part.add_header('Content-Disposition', f"attachment; filename= {os.path.basename(filename)}")
+        message.attach(part)
+
+    try:
+        server = smtplib.SMTP(smtp_server, smtp_port)
+        server.starttls()
+        server.login(smtp_user, smtp_password)
+        text = message.as_string()
+        server.sendmail(sender_email, receiver_email, text)
+        server.quit()
+        print('Email sent')
+    except Exception as e:
+        print(f'Error sending email')
+
+if __name__ == "__main__":
+    # MySQL Connection and Query Info
+    conn = connect_db()
+    if conn:
+        query = """SELECT
+    mcc, mnc, route, price, currency
+FROM
+    smsgw_control.routingprices
+WHERE
+    clientid = '13271' AND route = 'M1'
+ORDER BY 1 ASC;"""
+
+        rows, columns = run_query(conn, query)
+        conn.close()
+
+        # Write to CSV
+        csv_filename = "SixFiveNetworksM1.csv"
+        write_csv(csv_filename, rows, columns)
+
+        # Email Information
+        sender_email = "donotreply@mexedia.mt"
+        receiver_email = "mark@42tele.com"
+        subject = "MEXEDIA PRICE CHANGES for SixFiveNetworks - M1 - SixFiveNe_DIR"
+        body = """
+Dear Valued Customer,
+
+We would like to inform you of the following price changes on your account.
+
+Please note that the changes are effective immediately, any messages sent from now on will be charged based on the new price.
+
+If you require any further assistance or clarification, please do not hesitate to contact your Account Manager or our Support Team on support.sms@mexedia.com and we'll gladly assist you further.
+
+Best Regards,
+Mexedia Team"""
+
+        smtp_server = "smtp.eu.mailgun.org"
+        smtp_port = 587
+        smtp_user = 'stps@monitoring.fortytwo.mt'
+        smtp_password = 'f18c48a7076c868bef91eb8f0ca59d33-9776af14-d2648a16'
+
+        # Send email with CSV attachment
+        send_email(
+            sender_email, receiver_email, subject, body, csv_filename,
+            smtp_server, smtp_port, smtp_user, smtp_password
+        )
+