webapp/app/visa_free_config_fetchers/usa_fetcher.py

# -*- coding: utf-8 -*-
from app.visa_free_config_fetchers.base_fetcher import BaseFetcher

class USAFetcher(BaseFetcher):
    def __init__(self):
        super(USAFetcher, self).__init__("usa")

    def fetch_free_config(self, origin, target, visa_type, vac_code=None):
        origin_clean = origin.lower().strip()
        target_clean = target.lower().strip()
        vac_clean = (vac_code or "").lower().strip()

        missions = self.metadata.get("missions", {})
        mission_key = "{}_{}".format(origin_clean, target_clean)
        if vac_clean and "{}_{}_{}".format(origin_clean, vac_clean, target_clean) in missions:
            mission_key = "{}_{}_{}".format(origin_clean, vac_clean, target_clean)

        mission_data = missions.get(mission_key, {})
        if not mission_data:
            mission_data = {
                "usa_url": "https://www.usvisascheduling.com",
                "location": vac_clean.upper() if vac_clean else ""
            }

        return mission_data