import pandas as pd
import numpy as np
from datetime import datetime
from datetime import date
import logging, os
from sraPy.connections.mysql import sraController
from prefect.blocks.system import *
import boto3
import json

LOG_LEVEL = 'INFO'
LOG_FILE_NAME = os.path.basename(__file__)

class FidelityAllocations:
    @staticmethod
    def run_alloc():
        logging.getLogger(LOG_FILE_NAME).setLevel(LOG_LEVEL)
        <http://logging.info|logging.info>('Connecting to SRSE and SRA instances...')
        <http://logging.info|logging.info>(f'{os.getcwd()}')


        sraDatabaseSecretName = String.load("sra-database")
        srseDatabaseSecretName = String.load("srse-database")

from datetime import datetime
from prefect import task, Flow
#from pydantic import BaseModel
from main.prefect.utility.email import SRAMailer
from main.reports.allocations.fidelity_allocations import FidelityAllocations
from prefect.blocks.system import Secret
from prefect.blocks.notifications import SlackWebhook
from prefect.orion.schemas.states import Completed, Failed

# --------------------------------------------------------------------------------------------------
#class Custodian(BaseModel):
#    name: str


@task(name = 'Fidelity Allocations',
      retries = 2,
      retry_delay_seconds = 60)
def allocations_process():
    logger = get_run_logger()
    excel_file = FidelityAllocations.run_alloc()