def _admin_episodejokes(episode_code): from flask import request details = request.form.get('details', None) if request.method == 'POST': episode_joke_id = request.form.get('episode_joke_id', None) ej = EpisodeJoke.get(id=int(episode_joke_id)) ej.details = details ej.save() return '%s' % ej.id if request.method == 'PUT': joke_code = request.form.get('joke_code', None) joke_type = request.form.get('type', None) joke = Joke.get(code=int(joke_code)) episode = Episode.get(code=episode_code) code = 's%se%sj%s' % (str(episode.season).zfill(2), str( episode.episode).zfill(2), joke.code) context = {} context['ej'] = EpisodeJoke(joke=joke, episode=episode, joke_type=joke_type, details=details, code=code) context['ej'].save() return render_template('_episodejoke_form_row.html', **context)
def _parse_episodejokes(sheet): """ Parses joke sheet. Imports episodejokes. Will not update. """ episodes = list(Episode.select()) for row in sheet: for episode in episodes: joke = Joke.get(Joke.code == row['code']) if episode.code == 's02e03': episode.title = '\xc2\xa1Amigos!' if episode.code == 's01e13': episode.title = 'Beef Consomm\xc3\xa9' if row[episode.title] in ['1', 'f', 'b']: ej_dict = {} ej_dict['joke'] = joke ej_dict['episode'] = episode ej_dict['joke_type'] = row[episode.title] ej_dict['code'] = '%sj%s' % (ej_dict['episode'].code, ej_dict['joke'].code) try: EpisodeJoke.get(EpisodeJoke.code == ej_dict['code']) except EpisodeJoke.DoesNotExist: EpisodeJoke(**ej_dict).save()