uncloud/uncloud_pay/management/commands/import-vat-rates.py

51 lines
1.9 KiB
Python
Raw Normal View History

from django.core.management.base import BaseCommand
from uncloud_pay.models import VATRate
import logging
2020-10-08 17:54:04 +00:00
import urllib
import csv
import sys
import io
logger = logging.getLogger(__name__)
class Command(BaseCommand):
help = '''Imports VAT Rates. Assume vat rates of format https://github.com/kdeldycke/vat-rates/blob/master/vat_rates.csv'''
vat_url = "https://raw.githubusercontent.com/ungleich/vat-rates/main/vat_rates.csv"
2020-10-08 17:54:04 +00:00
def add_arguments(self, parser):
2020-10-08 17:54:04 +00:00
parser.add_argument('--vat-url', default=self.vat_url)
def handle(self, *args, **options):
2020-10-08 17:54:04 +00:00
vat_url = options['vat_url']
url_open = urllib.request.urlopen(vat_url)
# map to fileio using stringIO
csv_file = io.StringIO(url_open.read().decode('utf-8'))
reader = csv.DictReader(csv_file)
2020-10-08 17:54:04 +00:00
for row in reader:
if row["territory_codes"] and len(row["territory_codes"].splitlines()) > 1:
for code in row["territory_codes"].splitlines():
VATRate.objects.get_or_create(
starting_date=row["start_date"],
ending_date=row["stop_date"] if row["stop_date"] != "" else None,
territory_codes=code,
currency_code=row["currency_code"],
rate=row["rate"],
rate_type=row["rate_type"],
description=row["description"]
)
else:
VATRate.objects.get_or_create(
starting_date=row["start_date"],
ending_date=row["stop_date"] if row["stop_date"] != "" else None,
territory_codes=row["territory_codes"],
currency_code=row["currency_code"],
rate=row["rate"],
rate_type=row["rate_type"],
description=row["description"]
)
logger.info('All VAT Rates have been added!')