alx/procat2
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

send_locally_feed.py: simplify csv dir logic

Browse Source
This commit is contained in:
Seth Ladygo
2020-02-19 12:55:25 -08:00
parent b6fbb3f9de
commit 908793d077
1 changed files with 5 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
procat2/management/commands/send_locally_feed.py
View File

@ -2,6 +2,7 @@ import os
import csv import csv
import tempfile import tempfile
from pathlib import Path from pathlib import Path
from os.path import basename, dirname, realpath, abspath
from django.core.management.base import BaseCommand, CommandError from django.core.management.base import BaseCommand, CommandError
from django.conf import settings from django.conf import settings
@ -31,18 +32,14 @@ class Command(BaseCommand):
return UPC.objects.filter(material__in=materials).distinct('upc') return UPC.objects.filter(material__in=materials).distinct('upc')
def csv_dir(self):
return os.path.join(settings.ASSET_DIR, 'locally')
def csv_path(self, season, region): def csv_path(self, season, region):
csv_dir = os.path.join(settings.ASSET_DIR, 'locally')
filename = f'Keen_{season}_{region}.csv'.replace(' ', '_') filename = f'Keen_{season}_{region}.csv'.replace(' ', '_')
#mkstemp(suffix=None, prefix=None, dir=None, text=False) return os.path.join(csv_dir, filename)
return os.path.join(self.csv_dir(), filename)
def write_csv(self, path, upcs, season): def write_csv(self, path, upcs, season):
Path(self.csv_dir()).mkdir(parents=True, exist_ok=True) Path(dirname(path)).mkdir(parents=True, exist_ok=True)
with open(path, 'w', newline='') as csvfile: with open(path, 'w', newline='') as csvfile:
writer = csv.writer(csvfile) #, delimiter=',', quotechar='|', quoting=csv.QUOTE_MINIMAL) writer = csv.writer(csvfile) #, delimiter=',', quotechar='|', quoting=csv.QUOTE_MINIMAL)
writer.writerow(self.upc_headers()) writer.writerow(self.upc_headers())