archive
/
achievementsaggregator


			
							123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160
							import cgi
from Scraper import scraper

from google.appengine.ext import webapp
from google.appengine.api import users
from google.appengine.ext.webapp.util import run_wsgi_app
from google.appengine.ext import db

class AchievementSource(db.Model):
    name = db.StringProperty()
    url = db.LinkProperty()

class UserAccount(db.Model):
    user = db.UserProperty()
    source = db.ReferenceProperty(reference_class=AchievementSource)
    credentials = db.StringProperty()
    added = db.DateTimeProperty(auto_now_add=True)
    updated = db.DateTimeProperty()

class Achievement(db.Model):
    name = db.StringProperty()
    image = db.LinkProperty()
    description = db.StringProperty()
    source = db.ReferenceProperty(reference_class=AchievementSource)

class AwardedAchievement(db.Model):
    achievement = db.ReferenceProperty(reference_class=Achievement)
    user = db.UserProperty()
    awarded = db.DateTimeProperty()
    discovered = db.DateTimeProperty(auto_now_add=True)

class MainPage(webapp.RequestHandler):
    def get(self):
        user = users.get_current_user()

        if not user:
            self.redirect(users.create_login_url(self.request.uri))
            return

        self.show_header()

        if users.is_current_user_admin():
            self.show_admin_form()

        self.show_sources()

        self.show_footer()

        self.response.out.write(scraper().scrape_spore("csmith87"))

    def show_header(self):
        self.response.out.write("""
          <html>
            <head>
              <title>Achievements Aggregator</title>
            </head>
            <body>""")

    def show_footer(self):
        self.response.out.write("""
            </body>
          </html>""")

    def show_admin_form(self):
        self.response.out.write("""
          <h1>Admin - Sources</h1>
          <h2>Add</h2>
          <form action="/admin/addsource" method="post">
           <label>Name: <input type="text" name="name"/></label>
           <label>URL: <input type="text" name="url"/></label>
           <input type="submit" value="Add"/>
          </form>
          <h2>View</h2>
          <table>
           <tr><th>Source</th><th>URL</th></tr>
           """)

        for source in AchievementSource.all():
            self.response.out.write("<tr><td>")
            self.response.out.write(cgi.escape(source.name))
            self.response.out.write("</td><td>")
            self.response.out.write(cgi.escape(source.url))
            self.response.out.write("</td></tr>")

        self.response.out.write("</table>")

    def show_sources(self):
        self.response.out.write("<h1>My Accounts</h1>")
        self.response.out.write("<table>")
        self.response.out.write("<tr><th>Source</th><th>Credentials</th></tr>")

        for account in UserAccount.gql("WHERE user = :user", user=users.get_current_user()):
            self.response.out.write("<tr><td>")
            self.response.out.write(cgi.escape(account.source.name))
            self.response.out.write("</td><td>")
            self.response.out.write(cgi.escape(account.credentials))
            self.response.out.write("</td></tr>")

        self.response.out.write("</table>")
        self.response.out.write("""<h2>Add Account</h2>
          <form action="/addaccount" method="post">
           <label>Type: <select name="type">""")

        for source in AchievementSource.all():
            self.response.out.write('<option value="')
            self.response.out.write(source.key())
            self.response.out.write('">')
            self.response.out.write(cgi.escape(source.name))
            self.response.out.write('</option>')

        self.response.out.write("""</select></label>
           <label>Credentials: <input type="text" name="credentials"/></label>
           <input type="submit" value="Add"/>
          </form>""")

class AddSourcePage(webapp.RequestHandler):
    def post(self):

        if not users.is_current_user_admin():
            self.error(403)
            return

        source = AchievementSource(name=self.request.get('name'),
                                   url=self.request.get('url'))
        source.put()

        self.redirect('/')

class AddAccountPage(webapp.RequestHandler):
    def post(self):

        if not users.get_current_user():
            self.error(403)
            return

        source = db.get(db.Key(self.request.get('type')))

        account = UserAccount(user=users.get_current_user(),
                              source=source,
                              credentials=self.request.get('credentials'))
        account.put()

        self.redirect('/')


class UpdatePage(webapp.RequestHandler):
    def post(self):
        account = db.get(db.Key(self.request.get('key')))

application = webapp.WSGIApplication([('/', MainPage),
                                      ('/admin/addsource', AddSourcePage),
                                      ('/worker/update', UpdatePage),
                                      ('/addaccount', AddAccountPage)],
                                     debug=True)

def main():
    run_wsgi_app(application)

if __name__ == "__main__":
    main()