from flask import Flask, render_template, request, redirect, url_for, make_response from flask_pymongo import PyMongo from flask_wtf import FlaskForm from wtforms import StringField, validators from flask-login import LoginManager() mongo = PyMongo() login_manager = LoginManager() app = Flask(__name__) app.config['MONGO_URI'] = 'mongodb+srv://eoghan:[email protected]/test?retryWrites=true&w=majority' app.config['SECRET_KEY'] = "somekey" login_manager.init_app(app) mongo.init_app(app) client = mongo.cx db = client.beta_db comments = db.comments comment = 'Submit YES to get started.' class LoginForm(FlaskForm): name = StringField('username', validators = [validators.DataRequired()]) @app.route('/login', methods = ['GET', 'POST']) def login(): form = LoginForm() if form.validate_on_submit(): # login_user(user) flask.flash('Loggin in successfully.') return redirect(flask.url_for('index'))
from LoginManager import *
'''
..
----------------------------------------------------------------------
Ran 2 tests in 0.001s
OK
'''
# global managers to point to
passwordManagerOne = PasswordManager()
loginManagerOne = LoginManager(passwordManagerOne)
passwordManagerTwo = PasswordManager()
loginManagerTwo = LoginManager(passwordManagerTwo)
passwordManagerThree = PasswordManager()
loginManagerThree = LoginManager(passwordManagerThree)
class LoginManagerTest(unittest.TestCase):
global loginManagerOne, loginManagerTwo, loginManagerThree, passwordManagerOne, passwordManagerTwo, passwordManagerThree
def test_gettingPasswords(self):
# test successful connection
self.assertEqual(passwordManagerOne.getDatabasePassword(), "password")
self.assertEqual(passwordManagerOne.getFtpPassword(), "password")
resp = s.recv(1024)
if resp != b"Accept":
s.close()
sys.exit(0)
def sendToServer(plaintext):
"""Function to send message with signature to server"""
signature = keys.signUsingPrivateKey(plaintext)
encryptedText = keys.encrypt(plaintext, serverSessionKey)
s.send(encryptedText)
time.sleep(1)
s.send(signature)
# Creates log-in manager
login = lm.LoginManager()
currentUser = None
# Loop infinitely until user is logged into the system
while True:
cmd = input("What would you like to do? [R] for register, [L] for login, [Q] for quit. \n")
# Quitting program
if cmd.upper() == "Q":
print("Exiting, goodbye!")
s.send(b"Close")
s.close()
sys.exit(0)
# Sending registration request with digital signature
elif cmd.upper() == "R":
import os
import logging
from logging.handlers import RotatingFileHandler
from flask import Flask
from flask-sqlalchemy import SQLAlchemy
from flask-login import LoginManager
from flask-migrate import Migrate
from flask-bootstrap import Bootstrap
from config import Config
app = Flask(__name__)
app.config.from_object(Config)
db = SQLAlchemy(app)
login = LoginManager(app)
migrate = Migrate(app, db)
login.login_view = 'login'
if not app.debug:
if not os.path.exists('logs'):
os.mkdir('logs')
