if __name__ == "__main__":
# Argparse
parser = argparse.ArgumentParser()
parser.add_argument("--report", action="store_true")
args = parser.parse_args()
report = args.report
# Env file
load_dotenv()
db_user = os.getenv("POSTGRES_USER", os.getenv("DB_USER"))
db_pwd = os.getenv("POSTGRES_PASSWORD", os.getenv("DB_PWD"))
db_name = os.getenv("POSTGRES_DB", os.getenv("DB_NAME"))
provider = os.getenv("PROVIDER", "postgresql")
port = os.getenv("port", "5432")
# Cleaning & inserting
fm = FileManager(db_user, db_pwd, db_name, provider, port)
users = fm.clean_users()
ads = fm.clean_ads()
referrals = fm.clean_referrals()
ads_transaction = fm.clean_ads_transaction()
# Report
if report:
rm = ReportManager(users=users,
ads=ads,
referrals=referrals,
ads_transaction=ads_transaction)
rm.process()