我有一个管理命令,如果没有明确给出文件路径,我想读取每个文件并执行该命令。如何将管理命令指向正确的目录并让它打开每个文件?
class Command(BaseCommand):
def add_arguments(self, parser):
parser.add_argument('csvfile', nargs='?', type=argparse.FileType('r'))
# This was set up for multi-tenant use later
parser.add_argument('company_name', nargs='?', type=str, default='Fake_Company')
def handle(self, *args, **options):
contact_ids = []
if options['csvfile'] == None:
csv_files = 'apps/user/csv_files' # this is the path to the directory
for file in csv_files:
with options['csvfile'] as csvfile:
csvreader = csv.DictReader(csvfile)
for row in csvreader:
contact_ids.append(row['\ufeff"Id"'])
print(row['\ufeff"Id"'])