def connect_to_database(user=0, password=0):
if not user:
user = config.env['user']
if not password:
password = config.env['password']
options = {
'host': config.env['host'],
'user': user,
'passwd': password,
'db': config.env['db'],
'cursorclass': MySQLdb.cursors.DictCursor
}
db = MySQLdb.connect(**options)
db.autocommit(True)
return db
class User(UserMixin):
def __init__(self, username):
self.id = username
lm.session_protection = "strong"
login_manager = lm.LoginManager()
@login_manager.user_loader
def load_user(user_id):
return User(user_id)
#condition:
if (ENV == 'dev'):
app.debug = True
app.config[
'SQLALCHEMY_DATABASE_URI'] = 'postgresql://*****:*****@localhost/test'
else:
app.debug = False
app.config['SQLALCHEMY_DATABASE_URI'] = ''
app.config['SECRET_KEY'] = b'secretism'
db = SQLAlchemy(app)
app.config["SESSION_PERMANENT"] = False
login_manager = login_manager.LoginManager()
login_manager.init_app(app)
login_manager.login_view = 'login'
# database models
class registration(UserMixin, db.Model):
__tablename__ = 'registration'
uuid = db.Column(db.Integer(), primary_key=True)
number = db.Column(db.String(), unique=True, nullable=False)
password = db.Column(db.String(), nullable=False)
accountType = db.Column(db.String(), unique=True, nullable=False)
def __init__(self, number, password, accountType):
self.number = number
self.password = password
app.register_blueprint(dbgcollect_page)
app.register_blueprint(sysbackup_page)
app.register_blueprint(reportgen_page)
app.register_blueprint(trendaudit_page)
app.register_blueprint(auditstrategy_page)
app.register_blueprint(macfilter_page)
app.register_blueprint(start_pcap_page)
app.register_blueprint(license_page)
app.register_blueprint(assets_locate_page)
app.register_blueprint(static_route_config_page)
app.register_blueprint(logaudit_page)
app.secret_key = 'A0Zr98j/3yX C~XHH!jmOR]LWN/,?RT'
#flask-login
import os
login_manager_ins = login_manager.LoginManager()
login_manager_ins.session_protection = 'strong'
login_manager_ins.init_app(app)
def load_config():
config_name = '/app/local/share/new_self_manage/app_config.ini'
if os.path.exists(config_name) == False:
return False
app_config = ConfigParser.ConfigParser()
try:
app_config.readfp(open(config_name))
login_dis = app_config.get("APPCONFIG","LOGIN_DISABLED")
# login_dis="True"
if login_dis == "False":
return False
return True
# Imports
from flask import Flask, render_template, session, request, redirect, url_for
from DatabaseManager import OrderingManager, ItemManager, UserManager, User, Item, GetTableHeaders, EndpointManager
from flask_login import login_manager, login_required, login_user, logout_user
# DEFINE app as Flask instance
app = Flask(__name__)
# Init all managers
OrderMan = OrderingManager()
UserMan = UserManager()
ItemMan = ItemManager()
# Init the Flask login manager
loginman = login_manager.LoginManager()
# Register the login manager with the Flask app
loginman.init_app(app)
# Set Secret key of Flask app
app.secret_key = b'_5#y2L"F4Q8z\n\xec]/'
# :route / -> Main landing page of website
@app.route('/')
def Landing():
return render_template('index.html')
# :route '/login' -> Login page of application