cv-flask.py

#! /usr/bin/python3.6
import os, sys

subs = ['models', 'resources', 'utilities']
app_file = '''from flask import Flask
from flask_restful import Api, Resource

app = Flask(__name__)
api = Api(app)

@app.route('/', methods=['GET'])
def index():
  return 'Hello, World!' '''

def usage() -> None:
  print('''  ______     __   __ _           _
 / ___\ \   / /  / _| | __ _ ___| | __
| |    \ \ / /  | |_| |/ _` / __| |/ /
| |___  \ V /   |  _| | (_| \__ \   <  Darkitsune
 \____|  \_/    |_| |_|\__,_|___/_|\_\ v1.0.0

Usage: cv-flask [ module | package ] [ name ]
Options:
  module:       creates new single module of Flask-RESTful
  package:      creates new package module of Flask-RESTful
''')
  sys.exit(2)

def make_package(name: str) -> str:
  run_file = '''from {} import app

if __name__ == '__main__':
  app.run(debug=True, port=5000)'''.format(name)
  base = './flask-project/{}'.format( name)

  try:
    for sub in subs:
      folder = '{}/{}'.format(base, sub)
      os.makedirs(folder)
      file = open('{}/__init__.py'.format(folder), 'w')
      file.close()

    file = open('./flask-project/run.py', 'w')
    file.write(run_file)
    file.close()

    file = open('./flask-project/{}/__init__.py'.format(name), 'w')
    file.write(app_file)
    file.close()
    return 'Flask Package created!'
  except:
    return 'File/folder already exists!'

def make_module(name: str) -> str:
  try:
    for sub in subs:
      folder = './{}/{}'.format(name, sub)
      os.makedirs(folder)
      file = open('{}/__init__.py'.format(folder), 'w')
      file.close()

    file = open('./{}/app.py'.format(name), 'w')
    file.write(app_file)
    file.close()
    return 'Flask Package created!'
  except:
    return 'File/folder already exists!'

if __name__ == '__main__':
  if len(sys.argv) != 3:
    usage()

  if sys.argv[1] == 'module':
    make_module(sys.argv[2])
  elif sys.argv[1] == 'package':
    make_package(sys.argv[2])
  else:
    usage()