def postgres_handler(sketch: Sketch):
if sketch.database == "postgres":
sketch.add_requirements("psycopg2")
handle_sql_db(sketch)
sketch.settings["production"][
"SQLALCHEMY_DATABASE_URI"] = "postgres://<user>:<password>@<server_ip>/MY_DATABASE"
return True
def handle_caching(sketch: Sketch):
sketch.add_requirements("flask-caching")
sketch.settings["development"]["CACHE_TYPE"] = "simple"
sketch.settings["testing"]["CACHE_TYPE"] = "simple"
sketch.settings["production"]["CACHE_TYPE"] = "simple"
sketch.add_extensions("caching")
sketch.write_template(
"ext_caching_tpl",
templates.ext,
pjoin(sketch.app_folder, "ext", "caching.py"),
)
sketch.write_template(
"examples_caching_tpl",
templates.examples,
pjoin(
sketch.app_folder,
"examples",
"caching_examples.py",
),
)
sketch.write_template(
"examples_init_caching_tpl",
templates.examples,
pjoin(
sketch.app_folder,
"examples",
"__init__.py",
),
)
def mysql_handler(sketch: Sketch):
if sketch.database == "mysql":
sketch.add_requirements("mysqlclient")
handle_sql_db(sketch)
sketch.settings["production"][
"SQLALCHEMY_DATABASE_URI"] = "mysql+mysqldb://<user>:<password>@<server_ip>/MY_DATABASE"
return True
def handle_limiter(sketch: Sketch):
sketch.add_requirements("flask-limiter")
sketch.settings["default"]["RATELIMIT_DEFAULT"] = "200 per day;50 per hour"
sketch.settings["default"]["RATELIMIT_ENABLED"] = True
sketch.settings["development"]["RATELIMIT_ENABLED"] = False
sketch.add_extensions("limiter")
sketch.write_template(
"ext_limiter_tpl",
templates.ext,
pjoin(sketch.app_folder, "ext", "limiter.py"),
)
sketch.write_template(
"examples_limiter_tpl",
templates.examples,
pjoin(
sketch.app_folder,
"examples",
"limiter_examples.py",
),
)
sketch.write_template(
"examples_init_limiter_tpl",
templates.examples,
pjoin(
sketch.app_folder,
"examples",
"__init__.py",
),
)
def handle_migrate(sketch: Sketch):
sketch.add_requirements("flask-migrate")
sketch.add_extensions("migrate")
sketch.write_template(
"ext_migrate_tpl",
templates.ext,
pjoin(sketch.app_folder, "ext", "migrate.py"),
)
def security_web_handler(sketch: Sketch):
if sketch.auth_framework == "security_web":
sketch.add_requirements("flask-security-too", "argon2-cffi")
sketch.settings["default"]["SECURITY_REGISTERABLE"] = True
sketch.settings["default"]["SECURITY_POST_LOGIN_VIEW"] = "/"
sketch.settings["default"]["SECURITY_PASSWORD_HASH"] = "argon2"
sketch.add_extensions("auth")
sketch.secrets["default"]["SECURITY_PASSWORD_SALT"] = random_string()
sketch.write_template(
"security_web_only_tpl",
templates.commands,
pjoin(
sketch.app_folder,
"commands",
"__init__.py",
),
)
sketch.write_template(
"ext_security_web_only_tpl",
templates.ext,
pjoin(sketch.app_folder, "ext", "auth.py"),
)
sketch.write_template(
"models_security_web_only_tpl",
templates.models,
pjoin(sketch.app_folder, "models", "user.py"),
)
sketch.write_template(
"examples_security_auth_tpl",
templates.examples,
pjoin(
sketch.app_folder,
"examples",
"auth_examples.py",
),
)
sketch.write_template(
"examples_init_security_tpl",
templates.examples,
pjoin(
sketch.app_folder,
"examples",
"__init__.py",
),
)
return True
def handle_sql_db(sketch: Sketch):
sketch.add_requirements("flask-sqlalchemy")
sketch.settings["default"]["SQLALCHEMY_TRACK_MODIFICATIONS"] = False
sketch.settings["development"]["SQLALCHEMY_TRACK_MODIFICATIONS"] = True
sketch.settings["default"][
"SQLALCHEMY_DATABASE_URI"] = "sqlite:///db.sqlite3"
sketch.add_extensions("database")
sketch.write_template(
"ext_sqlalchemy_tpl",
templates.ext,
pjoin(sketch.app_folder, "ext", "database.py"),
)
def handle_debugtoolbar(sketch: Sketch):
sketch.add_requirements("flask-debugtoolbar", dev=True)
sketch.add_extensions("debugtoolbar")
sketch.settings["development"]["DEBUG_TB_INTERCEPT_REDIRECTS"] = False
if sketch.config_framework != "dynaconf":
sketch.write_template(
"ext_debugtoolbar_tpl",
templates.ext,
pjoin(
sketch.app_folder,
"ext",
"debugtoolbar.py",
),
)
def dynaconf_handler(sketch: Sketch):
if sketch.config_framework == "dynaconf":
sketch.add_requirements("dynaconf")
settings_toml = sort_settings(sketch.settings)
secrets_toml = sort_settings(sketch.secrets)
try:
sketch.extensions.remove("debugtoolbar")
settings_toml["development"]["EXTENSIONS"] = [
"flask_debugtoolbar:DebugToolbarExtension",
"dynaconf_merge_unique",
]
except ValueError:
settings_toml["development"]["EXTENSIONS"] = [
"dynaconf_merge_unique"
]
settings_toml["default"]["EXTENSIONS"] = [
"{}.ext.{}:init_app".format(sketch.app_folder_name, ext)
for ext in sketch.extensions
]
with open(pjoin(sketch.project_folder, "settings.toml"), "w") as f:
toml.dump(settings_toml, f, encoder=FlaskSketchTomlEncoder())
with open(pjoin(sketch.project_folder, ".secrets.toml"), "w") as f:
toml.dump(secrets_toml, f, encoder=FlaskSketchTomlEncoder())
sketch.write_template(
"config_dynaconf_tpl",
templates.config,
pjoin(sketch.app_folder, "config", "__init__.py",),
)
sketch.write_template(
"app_dynaconf_conf_tpl",
templates.app,
pjoin(sketch.app_folder, "app.py"),
)
return True
def handle_admin(sketch: Sketch):
sketch.add_requirements("flask-admin")
sketch.add_extensions("admin")
sketch.settings["default"]["ADMIN_TEMPLATE_MODE"] = "bootstrap3"
sketch.settings["development"][
"ADMIN_NAME"] = f"{sketch.project_name} (Dev)"
sketch.settings["testing"][
"ADMIN_NAME"] = f"{sketch.project_name} (Testing)"
sketch.settings["production"]["ADMIN_NAME"] = sketch.project_name
# TODO refact this part to not use a lot of if statements
if sketch.auth_framework == "security_web":
sketch.write_template(
"ext_admin_security_tpl",
templates.ext.admin,
pjoin(
sketch.app_folder,
"ext",
"admin",
"__init__.py",
),
)
def login_jwt_extended_handler(sketch: Sketch):
if sketch.auth_framework == "login_jwt_extended":
sketch.add_requirements("flask-login", "flask-jwt-extended")
return True
def security_jwt_extended_handler(sketch: Sketch):
if sketch.auth_framework == "security_jwt_extended":
sketch.add_requirements("flask-security-too", "flask-jwt-extended")
return True
def basicauth_api_handler(sketch: Sketch):
if sketch.auth_framework == "basicauth_api":
sketch.add_requirements("flask-basicauth")
return True
def basicauth_web_handler(sketch: Sketch):
if sketch.auth_framework == "basicauth_web":
sketch.add_requirements("flask-basicAuth")
sketch.secrets["default"]["BASIC_AUTH_PASSWORD"] = "******"
sketch.secrets["default"]["BASIC_AUTH_PASSWORD"] = random_string()
return True
def mongodb_handler(sketch: Sketch):
if sketch.database == "mongodb":
sketch.add_requirements("flask-mongoengine")
return True
