mygpo/maintenance/management/commands/import-episode-actions.py

   1 from datetime import datetime
   2 import os.path
   3
   4 from django.core.management.base import BaseCommand
   5
   6 from mygpo.users.models import User
   7 from mygpo.api.advanced import update_episodes
   8 from mygpo.core.json import json
   9 from mygpo.utils import progress
  10
  11
  12 class Command(BaseCommand):
  13     """Imports episode actions from a file named <userid>-<something>"""
  14
  15     def handle(self, *args, **options):
  16
  17         path = args[0]
  18
  19         for filename in os.listdir(path):
  20             user_id, _ = filename.split('-', 1)
  21             filename = os.path.join(path, filename)
  22             self.import_file(user_id, filename)
  23             print
  24
  25
  26     def import_file(self, user_id, filename):
  27
  28         progress(0, 100, filename)
  29
  30         with open(filename, 'r') as f:
  31             actions = json.load(f)
  32
  33         progress(0, len(actions), filename)
  34
  35         user = User.get(user_id)
  36
  37         now = datetime.now()
  38
  39         batch_size = 100
  40
  41         count = len(actions) / batch_size
  42
  43         for low in range(0, len(actions), batch_size):
  44             high = low+batch_size
  45             batch = actions[low:high]
  46
  47             update_episodes(user, batch, now, None)
  48
  49             progress(high, len(actions), filename)