73 lines
2.1 KiB
Python
73 lines
2.1 KiB
Python
import mysql.connector
|
|
from mysql.connector import Error
|
|
import config
|
|
import openpyxl
|
|
import os
|
|
import re
|
|
import ast
|
|
from datetime import datetime
|
|
|
|
|
|
class ContractorInfo:
|
|
ID = ""
|
|
contInfo = None
|
|
def __init__(self, id):
|
|
self.ID = id
|
|
print(id)
|
|
self.fetchData()
|
|
|
|
def fetchData(self):
|
|
try:
|
|
connection = config.get_db_connection()
|
|
cursor = connection.cursor(dictionary=True, buffered=True)
|
|
cursor.callproc('GetContractorInfoById', [self.ID])
|
|
for result in cursor.stored_results():
|
|
self.contInfo = result.fetchone()
|
|
|
|
print(self.contInfo,flush=True)
|
|
finally:
|
|
cursor.close()
|
|
connection.close()
|
|
|
|
def fetchalldata(self):
|
|
|
|
try:
|
|
connection = config.get_db_connection()
|
|
cursor = connection.cursor(dictionary=True, buffered=True)
|
|
print("here", flush=True)
|
|
|
|
|
|
# ---------------- Hold Types ----------------
|
|
cursor = connection.cursor(dictionary=True)
|
|
|
|
cursor.callproc('GetHoldTypesByContractor', [self.ID])
|
|
|
|
hold_types = []
|
|
for result in cursor.stored_results():
|
|
hold_types = result.fetchall()
|
|
hold_type_map = {ht['hold_type_id']: ht['hold_type'] for ht in hold_types}
|
|
|
|
# ---------------- Invoices ----------------
|
|
cursor = connection.cursor(dictionary=True)
|
|
|
|
cursor.callproc('GetInvoicesByContractor', [self.ID])
|
|
|
|
invoices = []
|
|
for result in cursor.stored_results():
|
|
invoices = result.fetchall()
|
|
|
|
# Remove duplicate invoices
|
|
invoice_ids_seen = set()
|
|
unique_invoices = []
|
|
for inv in invoices:
|
|
if inv["Invoice_Id"] not in invoice_ids_seen:
|
|
invoice_ids_seen.add(inv["Invoice_Id"])
|
|
unique_invoices.append(inv)
|
|
invoices = unique_invoices
|
|
|
|
finally:
|
|
cursor.close()
|
|
connection.close()
|
|
|
|
|