0

我有一个管理命令,如果没有明确给出文件路径,我想读取每个文件并执行该命令。如何将管理命令指向正确的目录并让它打开每个文件?

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"'])
4

0 回答 0