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')