fy0/fpage

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

import os
import re
import os
import sys
import shutil
from sys import argv
from os.path import join
py_major_ver = sys.version_info[0]

src_dir = os.path.dirname(os.path.abspath(__file__))


def help():
    print('FPage v1.2 - tornado classic project generator(cli)')
    print('repo: github.com/fy0/fpage\n')
    print('Usage:')
    print('  fpage <command>')
    print('')
    print('Commands:')
    print('  new [project-name]')
    print('  help')
    print('')


def gen(project_dir, project_name, tmpl_engine, db_orm):
    shutil.copytree(join(src_dir, 'src'), project_dir)

    if tmpl_engine == 'mako':
        shutil.rmtree(join(project_dir, 'templates_jinja2'))
        shutil.rmtree(join(project_dir, 'templates_tornado'))
    elif tmpl_engine == 'jinja2':
        shutil.rmtree(join(project_dir, 'templates'))
        shutil.rmtree(join(project_dir, 'templates_tornado'))
        shutil.move(join(project_dir, 'templates_jinja2'), join(project_dir, 'templates'))
    elif tmpl_engine == 'tornado':
        shutil.rmtree(join(project_dir, 'templates'))
        shutil.rmtree(join(project_dir, 'templates_jinja2'))
        shutil.move(join(project_dir, 'templates_tornado'), join(project_dir, 'templates'))

    if db_orm == 'sqlalchemy':
        shutil.rmtree(join(project_dir, 'model_peewee'))
    elif db_orm == 'peewee':
        shutil.rmtree(join(project_dir, 'model'))
        shutil.move(join(project_dir, 'model_peewee'), join(project_dir, 'model'))
    
    config_file = join(project_dir, 'config.py')
    fp = open(config_file)
    txt = fp.read()
    fp.close()
    txt = txt.replace("TITLE = 'FPage'", "TITLE = '%s'" % project_name)
    txt = txt.replace("TEMPLATE = 'mako'", "TEMPLATE = '%s'" % tmpl_engine)
    try:
        fp = open(config_file, 'w+', encoding='utf-8')
    except:
        fp = open(config_file, 'w+')
    fp.write(txt)
    fp.close()
    return True


def startapp(appname=None):
    global input  # fix4py3
    if py_major_ver == 2:
        input = raw_input
    
    if appname:
        project_name = input('Project Name (%s):' % appname)
        if not project_name:
            project_name = appname
    else:
        project_name = input('Project Name:')
    
    tmpl_engine = input('Template Engine [M/J/T]:').lower()
    db_orm = input('Database ORM [P/S]:').lower()

    if len(project_name) == 0 or ' ' in project_name:
        print('Invalid Project Name.')
        return

    if tmpl_engine in ['', 'm']:
        tmpl_engine = 'mako'
    elif tmpl_engine in ['j', 'jinja', 'jinja2']:
        tmpl_engine = 'jinja2'
    elif tmpl_engine in ['t', 'tornado']:
        tmpl_engine = 'tornado'
    else:
        print('Invalid Value.')
        return

    if db_orm in ['', 'p', 'peewee']:
        db_orm = 'peewee'
    elif db_orm in ['s', 'sqlalchemy']:
        db_orm = 'sqlalchemy'
    else:
        print('Invalid Value.')
        return

    print('')
    print('   Project Name: %s' % project_name)
    print('Template Engine: %s' % tmpl_engine)
    print('   Database ORM: %s' % db_orm)
    print('')
    sure = input('Sure (Y/n)?')

    if sure == 'n':
        return

    project_dir = project_name
    if os.path.exists(project_dir):
        print('Already Exists!')
        return

    gen(project_dir, project_name, tmpl_engine, db_orm)
    print('Complete.\n')
    print('To get started:\n')
    print('    cd %s' % project_name)
    print('    python app.py\n')
    print('Served at http://localhost:9000')


def main():
    if len(argv) > 1:
        if argv[1] == 'help':
            help()
        elif argv[1] == 'new':
            if len(argv) > 2:
                startapp(argv[2])
            else:
                startapp()
        else:
            help()
    else:
        help()


if __name__ == "__main__":
    main()