from ftplib import FTP

def fetch_weather_xml():
    ftp = FTP('ftp.bom.gov.au')
    ftp.login()

    files = {
        "IDQ11295.xml": "/anon/gen/fwo/IDQ11295.xml",  # Queensland
        "IDN11060.xml": "/anon/gen/fwo/IDN11060.xml",  # New South Wales
        "IDV10753.xml": "/anon/gen/fwo/IDV10753.xml",  # Victoria
    }

    for filename, path in files.items():
        print(f"📥 Downloading {filename} ...")
        with open(filename, 'wb') as f:
            ftp.retrbinary(f'RETR {path}', f.write)

    ftp.quit()
    print("✅ All weather XML files downloaded.")

fetch_weather_xml()

# SINGLE VERSION BELOW

# from ftplib import FTP

# def fetch_weather_xml():
    # ftp = FTP('ftp.bom.gov.au')
    # ftp.login()
    # with open('IDQ11295.xml', 'wb') as f:
        # ftp.retrbinary('RETR /anon/gen/fwo/IDQ11295.xml', f.write)
    # ftp.quit()

# fetch_weather_xml()