GlassBlower/glassblower.py at master · pythonfortinero/GlassBlower · 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
121
122
123
124
125
126
127
128
129
130
131
132
133
import os
import argparse
parser = argparse.ArgumentParser()
parser.add_argument("blow")
parser.add_argument("option")
parser.add_argument("--newFile")
args = parser.parse_args()
currentPath = os.getcwd()


def make_template():
    htmlDir = False
    htmlDir = os.path.join(currentPath, 'app', 'templates', args.newFile.lower() + '.html')
    with open(htmlDir,'w') as af:
        af.write('{% extends "base.html" %} \n')
        af.write('\n')
        af.write('{% block content %} \n')
        af.write('<h1 class="text-center">%s</h1> \n' % args.newFile.upper())
        af.write('\n')
        af.write('{% endblock %} \n')
    return htmlDir


def make_view(api=False):
    fileDir = os.path.join(currentPath, 'app', 'views', args.newFile.lower() + '.py')
    with open(fileDir,'w') as af:
        af.write('from flask.views import MethodView \n')
        af.write('from flask import render_template \n')
        af.write('\n')
        af.write('class %s(MethodView): \n' % args.newFile.capitalize())
        af.write('\n')
        af.write('    def get(self): \n')
        if api:
            af.write('        return "%s" \n' % args.newFile.lower())
        else:
            af.write('        return render_template("%s.html") \n' % args.newFile.lower())
        method_list = ['post', 'put', 'delete']
        for method in method_list:
            af.write('\n')
            af.write('    def %s(self): \n' % method)
            af.write('        pass \n')
    initDir = os.path.join(currentPath, 'app', 'views', '__init__.py')
    with open(initDir,'a') as af:
        af.write('\nfrom %s import * \n' % args.newFile.lower())
    htmlDir = False
    if not api:
        htmlDir = make_template()
    routesDir = os.path.join(currentPath, 'config', 'routes.blz')
    with open(routesDir,'a') as af:
        af.write("\n/%s,%s,%s" % (args.newFile.lower(),
                                  args.newFile.capitalize(),
                                  args.newFile.lower()))
    print fileDir
    print initDir
    if htmlDir:
        print htmlDir
    print routesDir


def make_model():
    fileDir = os.path.join(currentPath, 'app', 'models', args.newFile.lower() + '.py')
    with open(fileDir,'w') as af:
        af.write('from manage import app, db \n')
        af.write('from sqlalchemy import ForeignKey \n')
        af.write('from sqlalchemy.orm import relationship \n')
        af.write('\n')
        af.write('class %s(db.Model): \n' % args.newFile.capitalize())
        af.write('\n')
        af.write('    __tablename__ = "%s" \n' % args.newFile.lower() )
        af.write('    id = db.Column(db.Integer, primary_key=True)')
    initDir = os.path.join(currentPath, 'app', 'models', '__init__.py')
    with open(initDir,'a') as af:
        af.write('\nfrom %s import * \n' % args.newFile.lower())
    print fileDir
    print initDir


def make_login():
    fileDir = os.path.join(currentPath, 'app', '__init__.py')
    with open(fileDir,'a') as af:
        af.write('\n')
        af.write('from flask.ext.login import LoginManager\n')
        af.write('login_manager = LoginManager()\n')
        af.write('login_manager.init_app(app)\n')
        af.write('\n')
        af.write('login_manager.login_view = "login"\n')
        af.write('\n')
        af.write('@login_manager.user_loader\n')
        af.write('def load_user(id):\n')
        af.write('    return User.query.get(int(id))\n')

    routesDir = os.path.join(currentPath, 'config', 'routes.blz')
    with open(routesDir,'a') as af:
        af.write("\n/register,UserRegister,register")
        af.write("\n/login,UserLogin,login")
        af.write("\n/logout,UserLogout,logout")
    import shutil
    shutil.copy2(os.path.join(currentPath, 'utilities', 'user', 'models','user.py'),
                 os.path.join(currentPath, 'app','models','user.py'))
    shutil.copy2(os.path.join(currentPath, 'utilities', 'user', 'templates','login.html'),
                 os.path.join(currentPath, 'app', 'templates','login.html'))
    shutil.copy2(os.path.join(currentPath, 'utilities', 'user', 'templates','register.html'),
                 os.path.join(currentPath, 'app', 'templates','register.html'))
    shutil.copy2(os.path.join(currentPath, 'utilities', 'user', 'views','user.py'),
                 os.path.join(currentPath, 'app','views','user.py'))

    modelDir = os.path.join(currentPath, 'app', 'models', '__init__.py')
    with open(modelDir,'a') as af:
        af.write('\nfrom user import * \n')

    viewDir = os.path.join(currentPath, 'app', 'views', '__init__.py')
    with open(viewDir,'a') as af:
        af.write('\nfrom user import * \n')
    print fileDir
    print routesDir
    print modelDir
    print viewDir


print("Updated Directories")

if args.blow == "blow" and args.option == 'view':
    make_view()
elif args.blow == "blow" and args.option == 'model':
    make_model()
elif args.blow == 'blow' and args.option == 'api':
    make_view(api=True)
    make_model()
elif args.blow == 'blow' and args.option == 'scaffold':
    make_view()
    make_model()
elif args.blow == "blow" and args.option == 'login':
    make_login()