def list_messages():
"""Return list of recieved mail messages"""
session = Session()
messages = session.query(Message).all()
msg_lst = []
for message in messages:
msg_lst.append(create_message(message))
return msg_lst
def setup(db, response_converter, airtable_loader):
db = Session()
RunAirtableSync(airtable_loader, db, response_converter)
yield # this is where the testing happens
db.rollback()
cleanup(db)
def show_message(id):
"""
Return message selected by id
"""
session = Session()
message = session.query(Message).filter_by(id=id).one()
msg_dct = create_message(message)
msg_dct["size"] = message.size
msg_dct["recipients"] = create_recipients(message.recipients)
return msg_dct
def connected(self):
root_switch = rootDPID
pol = Filter(SwitchEq(root_switch)
& IP4DstEq("100.0.0.102")) >> SetPort(2)
pol = pol | Filter(SwitchEq(root_switch)
& IP4DstEq("100.0.0.103")) >> SetPort(1)
pol = pol | Filter(SwitchEq(root_switch)
& IP4DstEq("100.0.0.104")) >> SetPort(1)
pol = pol | Filter(SwitchEq(root_switch)
& IP4DstEq("100.0.0.105")) >> SetPort(1)
slave_switch_1 = slaveDPID
pol = pol | Filter(SwitchEq(slave_switch_1)
& IP4DstEq("100.0.0.105")) >> SetPort(4)
pol = pol | Filter(SwitchEq(slave_switch_1)
& IP4DstEq("100.0.0.103")) >> SetPort(2)
pol = pol | Filter(SwitchEq(slave_switch_1)
& IP4DstEq("100.0.0.104")) >> SetPort(3)
pol = pol | Filter(SwitchEq(slave_switch_1)
& IP4DstEq("100.0.0.102")) >> SetPort(1)
session = Session()
policies = session.query(Policies).filter_by(loaded=1)
for p in policies:
pol = pol | Filter(
SwitchEq(slave_switch_1)
& IP4SrcEq("100.0.0.103")) >> SetIP4Dst("100.0.0.104")
pol = pol | Filter(
SwitchEq(slave_switch_1)
& EthSrcEq("62:99:69:d7:ef:49")) >> SetIP4Dst("100.0.0.104")
pol = pol | Filter(
SwitchEq(slave_switch_1)
& IP4SrcEq("100.0.0.103")) >> SetEthDst("d2:33:20:b8:08:40")
pol = pol | Filter(
SwitchEq(slave_switch_1) & EthSrcEq("62:99:69:d7:ef:49")
) >> SetEthDst("d2:33:20:b8:08:40")
pol = pol | Filter(
SwitchEq(slave_switch_1)
& IP4SrcEq("100.0.0.103")) >> SetPort(3)
pol = pol | Filter(
SwitchEq(slave_switch_1)
& EthSrcEq("62:99:69:d7:ef:49")) >> SetPort(3)
app.update(pol)
def connected(self):
root_switch = 196040413341508
pol = Filter(SwitchEq(root_switch) & IP4DstEq("10.0.0.102")) >> SetPort(2)
pol = pol | Filter(SwitchEq(root_switch) & IP4DstEq("10.0.0.103")) >> SetPort(3)
pol = pol | Filter(SwitchEq(root_switch) & IP4DstEq("10.0.0.104")) >> SetPort(4)
pol = pol | Filter(SwitchEq(root_switch) & IP4DstEq("10.0.0.105")) >> SetPort(1)
pol = pol | Filter(SwitchEq(root_switch) & IP4DstEq("10.0.0.106")) >> SetPort(1)
pol = pol | Filter(SwitchEq(root_switch) & IP4DstEq("10.0.0.107")) >> SetPort(1)
pol = pol | Filter(SwitchEq(root_switch) & IP4DstEq("10.0.0.108")) >> SetPort(1)
pol = pol | Filter(SwitchEq(root_switch) & IP4DstEq("10.0.0.109")) >> SetPort(1)
pol = pol | Filter(SwitchEq(root_switch) & IP4DstEq("10.0.0.110")) >> SetPort(1)
slave_switch_1 = 77043891114308
pol = pol | Filter(SwitchEq(slave_switch_1) & IP4DstEq("10.0.0.105")) >> SetPort(5)
pol = pol | Filter(SwitchEq(slave_switch_1) & IP4DstEq("10.0.0.106")) >> SetPort(6)
pol = pol | Filter(SwitchEq(slave_switch_1) & IP4DstEq("10.0.0.107")) >> SetPort(7)
pol = pol | Filter(SwitchEq(slave_switch_1) & IP4DstEq("10.0.0.108")) >> SetPort(3)
pol = pol | Filter(SwitchEq(slave_switch_1) & IP4DstEq("10.0.0.109")) >> SetPort(4)
pol = pol | Filter(SwitchEq(slave_switch_1) & IP4DstEq("10.0.0.110")) >> SetPort(2)
pol = pol | Filter(SwitchEq(slave_switch_1) & IP4DstEq("10.0.0.102")) >> SetPort(1)
pol = pol | Filter(SwitchEq(slave_switch_1) & IP4DstEq("10.0.0.103")) >> SetPort(1)
pol = pol | Filter(SwitchEq(slave_switch_1) & IP4DstEq("10.0.0.104")) >> SetPort(1)
session = Session()
policies = session.query(Policies).filter_by(loaded=1)
for p in policies:
pol = pol | Filter(SwitchEq(p.switch1) & IP4SrcEq(p.src_ip)) >> SetIP4Dst(QVM_IP)
app.update(pol)
def send_notification(recipient: str,
message: str,
channel: NotificationChannel,
scheduled_send_date: datetime = None,
subject=None):
notification = Notification(recipient=recipient,
subject=subject,
message=message,
channel=channel,
scheduled_send_date=scheduled_send_date)
db = Session()
db.add(notification)
if notification.channel is NotificationChannel.EMAIL:
send_email_notification(notification)
elif notification.channel is NotificationChannel.SMS:
send_sms_notification(notification)
elif notification.channel is NotificationChannel.SLACK:
send_slack_notification(notification)
else:
raise ValueError()
db.commit()
class UserMigration(Base):
__tablename__ = 'usermigration'
userMigrationUID = Column('userMigrationUID',
String(100),
nullable=False,
primary_key=True)
userIP = Column('userIP', String(45), nullable=True)
originalServerIP = Column('originalServerIP', String(45), default=None)
migratedServerIP = Column('migratedServerIP', String(45), default=None)
migrationStartTime = Column('migrationStartTime', DateTime, default=None)
migrationStopTime = Column('migrationStopTime', DateTime, default=None)
class Users(Base):
__tablename__ = 'users'
userUID = Column('userID', String(100), nullable=False, primary_key=True)
username = Column('username', String(100), default=None)
ipAddressuserIP = Column('ipAddress', String(20), default=None)
connectionStartTime = Column('connectionStartTime', DateTime, default=None)
connectionStopTime = Column('connectionStopTime', DateTime, default=None)
class Whitelist(Base):
__tablename__ = 'whitelist'
ipv4 = Column('ipv4', String(15), nullable=False, primary_key=True)
session = Session()
Base.metadata.create_all(engine)
session.commit()
def setup(db):
db = Session()
yield # this is where the testing happens
db.rollback()
def db():
return Session()
from settings import Session __author__ = 'jb' from orm_example import * b = TabB() sess = Session()
from local_values.reader import env_value
from flask import Flask, flash, render_template, redirect, url_for, request, session
from models import User
from forms import LoginForm
from settings import Session
from api.User import api_bp
from auth import auth_check
app = Flask(__name__)
app.debug = env_value('DEBUG', default='False')
app.secret_key = env_value('SECRET_KEY')
app.register_blueprint(api_bp)
db_session = Session()
@app.route('/')
@auth_check
def profile_info(user=None):
user = db_session.query(User).filter_by(
email=session['login'], password=session['password']).first()
return render_template('profile.html', user=user, page_name='Profile')
@app.route('/login', methods=['GET', 'POST'])
def login_post():
form = LoginForm()
if request.method == 'GET':
return render_template('login.html', form=form, page_name='Login')
login = form.login.data
def setup(db):
db = Session()
yield # this is where the testing happens
db.query(Account).delete()
db.commit()
