from datetime import datetime

from resource.enums.region import Region
from resource.service.FeiShuService import FeiShuService
from resource.utils.utils import value_covert, bottom_right, convert_storage_unit
from service.RDSServicee import RDSService

rds_service = RDSService()
fei_shu_service = FeiShuService()

fei_shu_spread_sheet_token = "XzQGsheQzhk74rtknKacClASnTc"
fei_shu_sheet_id = "R50CB5"

tenant_access_token = fei_shu_service.get_tenant_access_token("cli_a89702999f3c900b", "47ewnaxRqJAvHYdUR8idHgfzfeqAu0Pz")

instance_type_map = {
    "Primary": "主实例",
    "Readonly": "只读实例",
    "Guard": "灾备实例",
    "Temp": "临时实例"
}

status_map = {
    "Creating": "创建中",
    "Running": "使用中",
    "Deleting": "删除中",
    "Rebooting": "重启中",
    "Stopping": "暂停中",
    "Stopped": "已暂停",
    "DBInstanceClassChanging": "升降级中",
    "TRANSING": "迁移中",
    "EngineVersionUpgrading": "迁移版本中",
    "TransingToOthers": "迁移数据到其他RDS中",
    "GuardDBInstanceCreating": "生产灾备实例中",
    "Restoring": "备份恢复中",
    "Importing": "数据导入中",
    "ImportingFromOthers": "从其他RDS实例导入数据中",
    "DBInstanceNetTypeChanging": "内外网切换中",
    "GuardSwitching": "容灾切换中",
    "INS_CLONING": "实例克隆中",
    "Released": "已释放实例"
}

storage_type_map = {
    "local_ssd": "高性能本地盘",
    "ephemeral_ssd": "性能本地盘",
    "cloud_ssd": "SSD 云盘",
    "general_essd": "高性能云盘",
    "cloud_essd0": "ESSD PL0 云盘",
    "cloud_essd": "ESSD PL1 云盘",
    "cloud_essd2": "ESSD PL2 云盘",
    "cloud_essd3": "ESSD PL3 云盘"
}

pay_type_map = {
    "Postpaid": "按量付费",
    "Prepaid": "包年包月",
    "SERVERLESS": "Serverless"
}

category_map = {
    "Basic": "基础系列",
    "HighAvailability": "高可用系列",
    "cluster": "MySQL 集群系列",
    "AlwaysOn": "SQL Server 集群系列",
    "Finance": "三节点企业系列",
    "Serverless_basic": "Serverless 基础系列"
}


def main():
    region_list = [Region.HANG_ZHOU, Region.HONG_KONG, Region.US_WEST, Region.SOUTHEAST]
    for region in region_list:
        all_instances = rds_service.get_all_instance(region=region)
        for instance in all_instances:
            dt = datetime.now().strftime('%Y-%m-%d')
            instance_id = instance.dbinstance_id
            instance_name = instance.dbinstance_description
            category = value_covert(instance.category, category_map)
            memory = convert_storage_unit(instance.dbinstance_memory, "mb", "gb")
            cpu = instance.dbinstance_cpu
            status = value_covert(instance.dbinstance_status, status_map)
            instance_type = value_covert(instance.dbinstance_type, instance_type_map)
            engine = instance.engine
            engine_version = instance.engine_version
            zone_id = instance.zone_id

            attributes = rds_service.get_instance_detail(region, instance_id)
            attribute = attributes[0]
            disk_used = convert_storage_unit(attribute.dbinstance_disk_used, "b", "gb")
            storage = attribute.dbinstance_storage
            storage_type = value_covert(attribute.dbinstance_storage_type, storage_type_map)
            pay_type = value_covert(attribute.pay_type, pay_type_map)

            values = [[dt, region.desc, zone_id, instance_id, instance_name, category, status, instance_type, pay_type, engine, engine_version, memory, cpu, disk_used, storage, storage_type]]
            start_col = "A2"
            end_col = bottom_right(start_col, values)
            fei_shu_service.spreadsheet_values_prepend(tenant_access_token, fei_shu_spread_sheet_token, f"{fei_shu_sheet_id}!{start_col}:{end_col}", values)


if __name__ == "__main__":
    main()