ideas.py

""" Idea related functions. """

import json
import users
import sqlite3
import datetime as dt
from collections import namedtuple

from urllib.parse import urlencode
from urllib.request import urlopen

import db

GISTS_URL = "https://api.github.com/users/{0}/gists?"

Idea = namedtuple('Idea', 'gistid, username, description, created_at, tweetid')


def add(user, gist):
	""" Create a new idea. """

	idea = Idea(gist['id'], user.username, gist['description'],
				gist['created_at'], None)
	sql = "INSERT INTO ideas VALUES (?, ?, ?, ?, ?);"
	try:
		db.write(db.con, sql, idea)
		return idea
	except sqlite3.IntegrityError:
		return None


def get_untweeted():
	""" Return ideas that haven't been tweeted yet. """

	sql = "SELECT * FROM ideas WHERE tweetid IS NULL;"
	fetchall = db.read(db.con, sql, None)
	return map(Idea._make, fetchall)

def get_tweeted(time):
	"""Return ideas that have been tweeted out. """

	sql = "SELECT * FROM ideas WHERE tweetid IS NOT NULL AND FROM_DAYS(TO_DAYS(created_at)) <= (?) ;"
	fetchall = db.read(db.con, sql, time)
	return map(Idea._make, fetchall)



def set_tweetid(idea, tid):
	""" Set the tweetid of the idea to the passed value. """

	sql = "UPDATE ideas SET tweetid = (?) WHERE gistid = (?);"
	db.write(db.con, sql, (tid, idea.gistid))


def fetch_gists(user):
	""" Fetch gists from github. """

	url = GISTS_URL.format(user.username)
	params = {
		'client_id': db.config["Github"]["Client_Id"],
		'client_secret': db.config["Github"]["Client_Secret"]
	}

	if user.last_fetch:
		lf = dt.datetime.strptime(user.last_fetch, "%Y-%m-%d %H:%M:%S.%f")
		params['since'] = dt.datetime.isoformat(lf)

	try:
		data = urlopen(url + urlencode(params)).read().decode('utf-8')
	except:
		data = []
	else:
		users.update_last_fetch(user)

	return json.loads(data)