importkits.py 文件源码

def handle(self, *args, **opts):

        user = None
        if opts['username'] is not None:
            user = User.objects.get(username=opts['username'])
        elif opts['userid'] is not None:
            user = User.objects.get(id=opts['userid'])
        else:
            raise CommandError('Either username or userid is required')

        if opts['clear']:
            Run.objects.all().delete()
            Adapter.objects.all().delete()
            Kit.objects.all().delete()

        with open(opts['csvfile'], 'r') as f:
            if opts['format'] == 'chaim':
                self.import_chaim_csv(user, f)
            else:
                self.import_csv(user, f)