everywordjojo/main.py at master · paixlukee/everywordjojo · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
# Copyright (c) Brendan McCarthy

import os
import webapp2
import ConfigParser
import jinja2
import string
import tweepy
import logging
import urllib
import re
import sys

from tweepy import *
from google.appengine.api import users
from google.appengine.ext import db

JINJA_ENVIRONMENT = jinja2.Environment(
	loader     = jinja2.FileSystemLoader(os.path.dirname(__file__)),
	extensions = ['jinja2.ext.autoescape'],
	autoescape = False)

class Index(db.Model):
	index = db.IntegerProperty()

def tweet(status):
	config = ConfigParser.RawConfigParser()
	config.read('settings.cfg')

	# http://dev.twitter.com/apps/myappid
	CONSUMER_KEY = config.get('API Information', 'CONSUMER_KEY')
	CONSUMER_SECRET = config.get('API Information', 'CONSUMER_SECRET')
	# http://dev.twitter.com/apps/myappid/my_token
	ACCESS_TOKEN_KEY = config.get('API Information', 'ACCESS_TOKEN_KEY')
	ACCESS_TOKEN_SECRET = config.get('API Information', 'ACCESS_TOKEN_SECRET')

	auth = tweepy.OAuthHandler(CONSUMER_KEY, CONSUMER_SECRET)
	auth.set_access_token(ACCESS_TOKEN_KEY, ACCESS_TOKEN_SECRET)
	api = tweepy.API(auth,retry_count=2, retry_delay=1,
                         retry_errors=set([401, 404, 500, 503]))
	result = api.update_status(status)

def get_current_line(index):
	with open("words.txt") as source_fh:
		for i in range(index+1):
			status_str = source_fh.readline().strip()
		return status_str + " is a JoJo Reference"

class MainHandler(webapp2.RequestHandler):
	def get(self):
		user = users.get_current_user()
		disabled = ""

		if users.get_current_user():
			url = users.create_logout_url(self.request.uri)
			url_linktext = 'Hello, ' + user.nickname() + '. Logout'
			if user.nickname() == "paixlukee": #or user.nickname() == "test@example.com":
				disabled = ""
			else:
				disabled = "disabled"
		else:
			url = users.create_login_url(self.request.uri)
			url_linktext = 'Hello, guest. Login'
			disabled = "disabled"

		template_values = {
			'url': url,
			'url_linktext': url_linktext,
			'disabled': disabled,
		}

		template = JINJA_ENVIRONMENT.get_template('index.html')
		self.response.write(template.render(template_values))

class TweetHandler(webapp2.RequestHandler):
	def get(self):
		indexList = db.GqlQuery("SELECT * " +
    							"FROM Index ")
		indexNum = indexList[0].index
		tweet_text = get_current_line(indexNum)

		try:
			tweet(tweet_text)
			logging.info(tweet_text)
			indexNum += 1
		except TweepError as e:
			logging.exception(e)

		db.delete(db.Query())

		newIndex = Index()
		newIndex.index = indexNum
		newIndex.put()

		# self.redirect('/')

class MakeHandler(webapp2.RequestHandler):
	def get(self):
		try:
		  init = int(self.request.get("nr"))
		except:
	          init = 0
		index = Index()
		index.index = init
		index.put()

		self.redirect('/')

class TestHandler(webapp2.RequestHandler):
	def get(self):
		tweet('fuck test')

		self.redirect('/')

app = webapp2.WSGIApplication([
    ('/', MainHandler),
    ('/tweet', TweetHandler),
    ('/make', MakeHandler),
    ('/test', TestHandler),
], debug=True)