def reset_movie_history(date, rewrite='all', movie_id=None, **kwargs):
api = TraktTvApi()
media_type = 'movies'
if movie_id:
histories = [api.get_watched_history(type=media_type, id=movie_id)]
else:
all_watched = api.get_watched(type=media_type)
histories = []
for watched in all_watched:
histories.append(api.get_watched_history(type=media_type, id=watched['movie']['ids']['trakt']))
movies = []
to_delete = []
for history in histories:
if history:
sorted_history = sorted(history, lambda h: h['watched_at'])
if rewrite == 'earliest':
to_rewrite = [sorted_history[0]]
elif rewrite =='latest':
to_rewrite = [sorted_history[-1]]
else:
to_rewrite = sorted_history
for hist in to_rewrite:
movie_info = hist['movie']
movie_info['watched_at'] = date
movies.append(movie_info)
to_delete.append(hist['id'])
print "Deleting %d history items" % len(to_delete)
api.remove_from_history({'ids': to_delete})
print "Adding %d history items" % len(movies)
api.add_to_history({'movies': movies})
def auth_flow(**kwargs):
api = TraktTvApi()
secrets = load_secrets()
pin_url = PIN_URL % secrets[PIN_ID]
print "Head to %s to authenticate and enter your PIN below:" % pin_url
pin = raw_input("PIN: ")
while not pin_valid(pin):
pin = raw_input("PIN: ")
response = api.pin_request(pin)
with open(AUTH_PATH, "w") as auth_file:
expires = datetime.datetime.utcnow() + datetime.timedelta(seconds=response["expires_in"])
response["expires_on"] = expires.isoformat()
simplejson.dump(response, auth_file, indent=2)
print "Successfully authenticated."
