duolingo_client.py

#!/usr/bin/python3

import json
import os.path
import requests
import textwrap

BASE_URL = 'https://www.duolingo.com'
LOGIN_URL = BASE_URL + '/login'
URL = BASE_URL + '/2017-06-30/users/%s?fields=currentCourse,xpGains'
SESSION_URL = BASE_URL + '/2017-06-30/sessions'


def get_password():
  with open(os.path.expanduser('~/.duolingo-password')) as f:
    return f.read().strip()


class Duo:
  LESSONS = [1, 1, 2, 3, 5]
  LESSONS_TOTAL = []

  def __init__(self, username, password=get_password()):
    self.LESSONS_TOTAL.append(0)
    i = 0
    for _ in self.LESSONS:
      self.LESSONS_TOTAL.append(self.LESSONS_TOTAL[i] + self.LESSONS[i])
      i += 1

    login_response = requests.post(LOGIN_URL, data={'login': username, 'password': password})
    self._token = login_response.headers['jwt']
    user_id = json.loads(login_response.content.decode("utf-8"))['user_id']

    url = URL % user_id
    self._data = json.loads(requests.get(url, headers=self._get_headers()).content.decode("utf-8"))
    self.skills = []
    for row in self._data['currentCourse']['skills']:
      for skill in row:
        if skill.get('accessible', False):
          self.skills.append(skill)

  def _get_headers(self):
    return {'authorization': 'Bearer %s' % self._token}

  def get_data(self):
    return self._data

  def get_skills(self):
    return self.skills

  def get_xp_gains(self):
    return self._data['xpGains']

  def print_topics(self):
    course_total = 0
    course_finished = 0
    for skill in self.skills:
      name = skill['name']
      strength = int(skill['strength'] * 100)
      levels = skill['levels']
      lessons = skill['lessons']
      finished_levels = skill['finishedLevels']
      finished_lessons = skill['finishedLessons']
      if finished_levels == levels:
        skill_total = finished_lessons
        skill_finished = finished_lessons
      else:
        base_lessons = lessons / self.LESSONS[finished_levels]
        skill_total = base_lessons * self.LESSONS_TOTAL[levels]
        skill_finished = self.LESSONS_TOTAL[finished_levels] * base_lessons + finished_lessons

      course_total += skill_total
      course_finished += skill_finished

      print('%-40s %3d/%3d (%d) @ %d' % (name, skill_finished, skill_total, skill_total - skill_finished,
                                         strength))

    print('%-40s %3d/%3d (%d)' % ('Total', course_finished, course_total, course_total - course_finished))

  def get_tips(self):
    i = 0
    for skill in self.skills:
      explanation = skill.get('explanation', None)
      if explanation:
        i += 1
        title = explanation['title']
        print('Fetching %s' % title)
        filename = '%02d-%s.txt' % (i, title.replace(' ', '-'))
        url = explanation['url']
        tips = json.loads(requests.get(url).content)
        elements = tips['elements']
        with open(filename, 'w') as f:
          first = True
          for element in elements:
            if element['type'] == 'text':
              styled_string = element['element']['styledString']
              text = styled_string['text']
              styling = styled_string['styling']
              if not first and styling[0]['to'] == len(text):
                f.write('\n')
              f.write('%s\n' % textwrap.fill(text.encode('utf8')))
              first = False

  def get_session(self):
    data = {
      "fromLanguage": "en",
      "learningLanguage": "es",
      "challengeTypes": [
        "characterIntro",
        "characterMatch",
        "characterSelect",
        "dialogue",
        "form",
        "gapFill",
        # "judge",
        # "listen",
        "name",
        "listenComprehension",
        "listenTap",
        "readComprehension",
        "select",
        "selectPronunciation",
        "selectRecording",
        "selectTranscription",
        "tapCloze",
        "translate",
      ],
      "type": "GLOBAL_PRACTICE",
      "juicy": False
    }
    content = requests.post(SESSION_URL, headers=self._get_headers(), data=json.dumps(data)).content.decode("utf-8")
    session = json.loads(content)

    challenges = session['challenges']
    for i, challenge in enumerate(challenges):
      print('%d: %s from %s' % (i, challenge['type'], challenge['sourceLanguage']))

    # print(json.dumps(challenges[0], indent=2))


if __name__ == '__main__':
  duo = Duo('AlonAlbert')
  # duo.get_tips()
  print(len(duo.get_skills()))
  # print(json.dumps(duo.get_data(), indent=2))
  duo.get_session()
  # duo.print_topics()