manage.py

#!/usr/bin/env python
# coding=utf-8

from datetime import datetime
from dircache import listdir
from os import mkdir, system
import os
from os.path import exists
import csv
import bleach

from flask import current_app as app
from flask.ext.script import Manager
from icalendar import Calendar

from website.application import create_app, db
from website.crm.models import Speaker, Talk, Track2


DEBUG = True

OSDC_DATA = u"""Maik,Aussendorf,maik.aussendorf@dass-it.de
Sam,Bessalah,samkiller@gmail.com
Sébastien,Blanc,scm.blanc@gmail.com
François-Xavier,Bois,fxbois@gmail.com
Amaury,Bouchard,amaury@amaury.net
Julien,Bourdin,julien.bourdin@webtales.fr
Remi,Collet,rcollet@redhat.com
Christian,Couder,chriscool@tuxfamily.org
Bertrand,Dechoux,dechouxb@gmail.com
Kamel Ibn Aziz,Derouiche,kamel.derouiche@gmail.com
Christophe,Desclaux,christophe@zone-project.org
Laurent,Doguin,ldoguin@nuxeo.com
Sébastien,Douche,sdouche@gmail.com
Christophe,Fergeau,cfergeau@redhat.com
Haïkel,Guémar,karlthered@gmail.com
Viktor,Horvath,viktor.h@laposte.net
Damien,Krotkine,dams@zarb.org
Fabrice,Le Fessant,fabrice.le_fessant@ocamlpro.com
Mathieu,Lecarme,mlecarme@bearstech.com
Jonathan,MERCIER,bioinfornatics@gmail.com
Bruno,Michel,bruno.michel@af83.com
Gael,Pasgrimaud,gael@gawel.org
Julien,Pauli,julienpauli@gmail.com
Rodolphe,Quiédeville,rodolphe@quiedeville.org
Philippe,Robin,philippe.robin@technoveo.com
Frank,Rousseau,frank.rousseau@cozycloud.cc
Romuald,Rozan,romualdx.z.rozan@intel.com
Michael,Scherer,misc@zarb.org
Pierre,Schweitzer,pierre@reactos.org
Basile,Starynkevitch,basile.starynkevitch@cea.fr
Christophe,Villeneuve,hellsct1@gmail.com"""


manager = Manager(create_app)


@manager.shell
def make_shell_context():
  """
  Updates shell. (XXX: not sure what this does).
  """
  return dict(app=app, db=db)


@manager.command
def dump_routes():
  """
  Dump all the routes declared by the application.
  """
  for rule in app.url_map.iter_rules():
    print rule, rule.methods, rule.endpoint


@manager.command
def load_data():
  feedback = csv.reader(open("feedback.csv"))
  for row in feedback:
    row = map(lambda x: unicode(x, "utf8"), row)
    date = datetime.strptime(row[3], "%Y-%m-%d %H:%M:%S.%f")
    reg = Registration(email=row[0],
                       first_name=u"",
                       last_name=row[1],
                       organization=row[2],
                       date=date)
    db.session.add(reg)
  db.session.commit()


@manager.command
def create_db():
  if not exists("data"):
    mkdir("data")

  with app.app_context():
    db.create_all()


@manager.command
def drop_db():
  """
  Drop the DB.
  """
  # TODO: ask for confirmation.
  with app.app_context():
    db.drop_all()


@manager.command
def add_user(email, password):
  from website.security import User2
  user = User2(email=email, password=password, active=True)
  db.session.add(user)
  db.session.commit()
  print "Password updated"


@manager.command
def set_password(email, password):
  from website.security import User2
  user = User2.query.filter(User2.email==email).one()
  user.password = password
  db.session.commit()
  print "Password updated"


@manager.command
def index_content():
  whoosh = app.extensions['whoosh']
  pages = app.extensions['pages']

  for page in pages:
    doc = {}
    doc['path'] = page['path']
    doc['title'] = unicode(page['title'])
    doc['content'] = page.body
    doc['summary'] = bleach.clean(page.html, tags=[], strip=True)
    if len(doc['summary']) > 200:
      doc['summary'] = doc['summary'][0:200] + '...'
    whoosh.add_document(doc)


@manager.command
def build():
  """ Builds this site.
  """
  print("Building website...")
  app.debug = False
  asset_manager = app.extensions['asset_manager']
  asset_manager.config['ASSETS_DEBUG'] = False

  freezer = app.extensions['freezer']
  freezer.freeze()

  system("cp ./static/*.ico ./build/")
  system("cp ./static/*.txt ./build/")
  system("cp ./static/*.xml ./build/")
  print("Done.")


@manager.command
def load_osdc():
  cal = Calendar.from_ical(open('data/osdc.ics','rb').read())
  speakers = {}
  osdc1 = Track2.query.get(28)
  osdc2 = Track2.query.get(48)
  osdc3 = Track2.query.get(49)

  for component in cal.walk():
    if component.name != 'VEVENT':
      continue
    title = component['summary']
    room_name = component['location']
    abstract = component['description']
    speaker_name = component['organizer']
    starts_at = component['dtstart'].dt
    ends_at = component['dtend'].dt
    duration = int((ends_at - starts_at).total_seconds() / 60)

    speaker = speakers.get(speaker_name, None)
    if not speaker:
      speaker = Speaker(first_name="", last_name=speaker_name)
      speakers[speaker_name] = speaker

    talk = Talk(title=title, abstract=abstract,
                starts_at=starts_at, duration=duration)
    talk.speakers.append(speaker)

    if 'Gopher' in room_name:
      talk.track = osdc3
    else:
      if starts_at.day == 4:
        talk.track = osdc1
      else:
        talk.track = osdc2

    db.session.commit()

@manager.command
def update_osdc():
  for line in OSDC_DATA.split("\n"):
    first_name, last_name, email = line.split(",")
    name = first_name + " " + last_name
    speaker = Speaker.query.filter(Speaker.last_name == name).first()
    if speaker:
      print "Updating speaker", name
      speaker.first_name = first_name
      speaker.last_name = last_name
      speaker.email = email
  db.session.commit()

@manager.command
def serve(server='0.0.0.0', port=5002, debug=DEBUG):
  """ Serves this site.
  """
  if not debug:
    import logging

    file_handler = FileHandler("error.log")
    file_handler.setLevel(logging.WARNING)
    app.logger.addHandler(file_handler)

  asset_manager = app.extensions['asset_manager']
  asset_manager.config['ASSETS_DEBUG'] = debug

  app.debug = debug
  app.run(host=server, port=port, debug=debug)


if __name__ == '__main__':
  manager.run()