magic.py

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

from datetime import datetime
from HTMLParser import HTMLParser
import argparse
import json
import os
import sys
import urllib2


###############################################################################
# Options
###############################################################################
MAGIC_URL = 'http://magic.gae-init.appspot.com'

PARSER = argparse.ArgumentParser(description='Visit %s for more.' % MAGIC_URL)
PARSER.add_argument(
    '-p', '--project', dest='project_id', action='store',
    help='project ID of the project that you want to sync',
  )
PARSER.add_argument(
    '-r', '--remote', dest='remote_url', action='store', default=MAGIC_URL,
    help="set the remote URL if it's not http://magic.gae-init.appspot.com",
  )
ARGS = PARSER.parse_args()


###############################################################################
# Constants
###############################################################################
DIR_MAIN = 'main'

DIR_CONTROL = os.path.join(DIR_MAIN, 'control')
FILE_CONTROL_INIT = os.path.join(DIR_CONTROL, '__init__.py')

DIR_MODEL = os.path.join(DIR_MAIN, 'model')
FILE_MODEL_INIT = os.path.join(DIR_MODEL, '__init__.py')

DIR_API = os.path.join(DIR_MAIN, 'api', 'v1')
FILE_API_INIT = os.path.join(DIR_API, '__init__.py')

DIR_TEMPLATES = os.path.join(DIR_MAIN, 'templates')
FILE_HEADER = os.path.join(DIR_TEMPLATES, 'bit', 'header.html')
FILE_ADMIN = os.path.join(DIR_TEMPLATES, 'admin', 'admin.html')


###############################################################################
# Helpers
###############################################################################
def print_out(script, filename=''):
  timestamp = datetime.now().strftime('%H:%M:%S')
  if not filename:
    filename = '-' * 46
    script = script.rjust(12, '-')
  print '[%s] %12s %s' % (timestamp, script, filename)


def make_dirs(directory):
  if not os.path.exists(directory):
    os.makedirs(directory)


def append_to(project_url, destination):
  response = urllib2.urlopen('%smagic/%s' % (project_url, destination))
  if response.getcode() == 200:
    with open(destination, 'r') as dest:
      lines = ''.join(dest.readlines())
      content = response.read()
      if content in lines:
        print_out('IGNORED', destination)
        return

    with open(destination, 'a') as dest:
      dest.write(content)
      print_out('APPEND', destination)


def insert_to(project_url, destination, find_what, indent=0):
  response = urllib2.urlopen('%smagic/%s' % (project_url, destination))
  if response.getcode() == 200:
    with open(destination, 'r') as dest:
      dest_contents = dest.readlines()
      lines = ''.join(dest_contents)
      content = HTMLParser().unescape(response.read())
      if content.replace(' ', '') in lines.replace(' ', ''):
        print_out('IGNORED', destination)
        return

    generated = []
    for line in dest_contents:
      generated.append(line)
      if line.lower().find(find_what.lower()) >= 0:
        spaces = len(line) - len(line.lstrip())
        for l in content.split('\n'):
          if l:
            generated.append('%s%s\n' % (' ' * (spaces + indent), l))

    with open(destination, 'w') as dest:
      for line in generated:
        dest.write(line)
      print_out('INSERT', destination)


def create_file(project_url, destination):
  response = urllib2.urlopen('%smagic/%s' % (project_url, destination))
  if response.getcode() == 200:
    with open(destination, 'w') as dest:
      dest.write(HTMLParser().unescape(response.read()))
      print_out('CREATE', destination)


def get_project_url():
  return '%s/api/v1/project/%s/' % (ARGS.remote_url, ARGS.project_id.split('/')[0])


def get_project_db():
  response = urllib2.urlopen(get_project_url())
  if response.getcode() == 200:
    project_body = response.read()
    return json.loads(project_body)['result']
  return None


def sync_from_magic():
    project_db = ''
    model_dbs = ''

    project_url = get_project_url()
    model_url = '%smodel/' % project_url

    response = urllib2.urlopen(model_url)
    if response.getcode() == 200:
      models_body = response.read()
      model_dbs = json.loads(models_body)['result']

    append_to(project_url, FILE_MODEL_INIT)
    append_to(project_url, FILE_CONTROL_INIT)
    append_to(project_url, FILE_API_INIT)
    insert_to(project_url, FILE_HEADER, '<ul class="nav navbar-nav">', 2)
    insert_to(project_url, FILE_ADMIN, "url_for('user_list'")

    for model_db in model_dbs:
      name = model_db['variable_name']
      create_file(project_url, os.path.join(DIR_MODEL, '%s.py' % name))
      create_file(project_url, os.path.join(DIR_CONTROL, '%s.py' % name))
      create_file(project_url, os.path.join(DIR_API, '%s.py' % name))

      root = os.path.join(DIR_TEMPLATES, name)
      make_dirs(root)
      create_file(project_url, os.path.join(root, '%s_update.html' % name))
      create_file(project_url, os.path.join(root, '%s_view.html' % name))
      create_file(project_url, os.path.join(root, '%s_list.html' % name))
      create_file(project_url, os.path.join(root, 'admin_%s_update.html' % name))
      create_file(project_url, os.path.join(root, 'admin_%s_list.html' % name))


###############################################################################
# Main
###############################################################################
def magic():
  if len(sys.argv) == 1:
    PARSER.print_help()
    sys.exit(1)

  os.chdir(os.path.dirname(os.path.realpath(__file__)))

  if ARGS.project_id:
    project_db = get_project_db()
    answer = raw_input(
        'Are you sure you want to sync "%(name)s" with %(model_count)d '
        'model(s) that was modified on %(modified)s? (Y/n): '
        % {
           'name': project_db['name'],
           'model_count': project_db['model_count'],
           'modified': project_db['modified'][:16].replace('T', ' at '),
        }
      )
    if not answer or answer.lower() == 'y':
      sync_from_magic()
  else:
    print 'Project ID is not provided.'
    PARSER.print_help()

if __name__ == '__main__':
  magic()