Add management command to import csv and update template

main/management/commands/import_fuelpurchases.py

@@ -0,0 +1,89 @@
+import csv
+import datetime
+import re
+from decimal import Decimal
+
+from django.core.management.base import BaseCommand
+
+from main.models import FuelPurchase
+
+
+class Command(BaseCommand):
+    help = "Import fuel purchases from CSV file"
+
+    # Valintakenttien sanakirjat nimistä arvoiksi
+    GAS_STATION_MAP = {
+        'Neste': 1,
+        'Teboil': 2,
+        'ABC': 3,
+        'Shell': 4,
+        'St1': 5,
+        'SEO': 6,
+        'ysi5': 7,
+        '-': 0,
+    }
+
+    OCTANE_MAP = {
+        '95': 95,
+        '98': 98,
+    }
+
+    CAR_MAP = {
+        '_': 0,
+        'Renault': 1,
+        'Nissan': 2,
+        'Smart': 3,
+    }
+
+    def add_arguments(self, parser):
+        parser.add_argument('csvfile', type=str, help='Path to the CSV file')
+
+    def parse_decimal(self, value_str):
+        # Poista kaikki ei-numerot/pilkku/piste ja korvaa pilkut pisteeksi
+        # esim. "10,00 €" -> "10.00"
+        value_str = value_str.strip()
+        # poista valuuttamerkit ja välilyönnit
+        value_str = re.sub(r'[^\d,.-]', '', value_str)
+        value_str = value_str.replace(',', '.')
+        return Decimal(value_str)
+
+    def parse_date(self, date_str):
+        # Muoto: 08.02.2015
+        return datetime.datetime.strptime(date_str, "%d.%m.%Y").date()
+
+    def handle(self, *args, **options):
+        file_path = options['csvfile']
+        with open(file_path, newline='', encoding='utf-8') as csvfile:
+            reader = csv.DictReader(csvfile, delimiter=';')
+            count = 0
+            for row in reader:
+                try:
+                    purchase_date = self.parse_date(row['Päivämäärä'])
+                    total_cost = self.parse_decimal(row['Hinta €'])
+                    price_per_litre = self.parse_decimal(row['Litrahinta €'])
+                    amount_litres = self.parse_decimal(row['Litraa'])
+
+                    gas_station_name = row['Asema'].strip()
+                    gas_station = self.GAS_STATION_MAP.get(gas_station_name, 0)
+
+                    octane_name = row['Polttoaine'].strip()
+                    octane = self.OCTANE_MAP.get(octane_name, 95)
+
+                    car_name = row['Auto'].strip()
+                    car = self.CAR_MAP.get(car_name, 0)
+
+                    obj, created = FuelPurchase.objects.get_or_create(
+                        purchase_date=purchase_date,
+                        gas_station=gas_station,
+                        octane=octane,
+                        car=car,
+                        total_cost=total_cost,
+                        price_per_litre=price_per_litre,
+                        amount_litres=amount_litres
+                    )
+                    if created:
+                        count += 1
+                except Exception as e:
+                    self.stderr.write(f"Error processing row {row}: {e}")
+
+            self.stdout.write(self.style.SUCCESS(f"Imported {count} fuel purchase records"))