def _init_test(self):
if os.path.isfile(FILE_NAME):
os.remove(FILE_NAME)
create_db(FILE_NAME)
p = Portfolio(FILE_NAME)
prt = p.create_portfolio()
return Algorithm(FILE_NAME, prt[0])
def test_create_portfolio(self):
if os.path.isfile(FILE_NAME):
os.remove(FILE_NAME)
create_db(FILE_NAME)
p = Portfolio(FILE_NAME)
c = p.create_portfolio()
self.assertEqual((1, -1), c)
os.remove(FILE_NAME)
def test_delete_id(self):
if os.path.isfile(FILE_NAME):
os.remove(FILE_NAME)
create_db(FILE_NAME)
p = Portfolio(FILE_NAME)
p.create_portfolio()
c = p.delete_id(1)
self.assertEqual(True, c)
os.remove(FILE_NAME)
def test_retrieve_portfolios(self):
if os.path.isfile(FILE_NAME):
os.remove(FILE_NAME)
create_db(FILE_NAME)
p = Portfolio(FILE_NAME)
p.create_portfolio()
c = p.retrieve_portfolios()
self.assertEqual([(1, -1)], c)
os.remove(FILE_NAME)
def test_update_id(self):
if os.path.isfile(FILE_NAME):
os.remove(FILE_NAME)
create_db(FILE_NAME)
p = Portfolio(FILE_NAME)
p.create_portfolio()
c = p.update_id(id=1, main_account=1)
self.assertEqual((1, 1), c)
os.remove(FILE_NAME)
def test_create_Fund(self):
if os.path.isfile(FILE_NAME):
os.remove(FILE_NAME)
create_db(FILE_NAME)
fund = Funds(FILE_NAME)
c = fund.create_fund(-20000, 1)
ts = c[2]
self.assertEqual((1, -20000, ts, 1), c)
os.remove(FILE_NAME)
def test_update_fund(self):
if os.path.isfile(FILE_NAME):
os.remove(FILE_NAME)
create_db(FILE_NAME)
trs = PortfolioFund(FILE_NAME)
c = trs.create_fund(-20000, 1)
ts = c[2]
self.assertEqual((1, -20000, ts, 1), c)
os.remove(FILE_NAME)
def test_create_transaction(self):
if os.path.isfile(FILE_NAME):
os.remove(FILE_NAME)
create_db(FILE_NAME)
trs = Transactions(FILE_NAME)
c = trs.create_transaction(-20000, 1)
ts = c[2]
self.assertEqual((1, -20000, ts, 1), c)
os.remove(FILE_NAME)
def test_retrieve_id(self):
if os.path.isfile(FILE_NAME):
os.remove(FILE_NAME)
create_db(FILE_NAME)
fund = Funds(FILE_NAME)
fund.create_fund(10, 1)
c = fund.retrieve_id(id=1)
ts = c[2]
self.assertEqual((1, 10, ts, 1), c)
os.remove(FILE_NAME)
def test_retrieve_id(self):
if os.path.isfile(FILE_NAME):
os.remove(FILE_NAME)
create_db(FILE_NAME)
trs = PortfolioFund(FILE_NAME)
trs.create_fund(10, 1)
c = trs.retrieve_id(id=1)
ts = c[2]
self.assertEqual((1, 10, ts, 1), c)
os.remove(FILE_NAME)
def test_retrieve_id(self):
if os.path.isfile(FILE_NAME):
os.remove(FILE_NAME)
create_db(FILE_NAME)
trs = Transactions(FILE_NAME)
trs.create_transaction(10, 1)
c = trs.retrieve_id(id=1)
ts = c[2]
self.assertEqual((1, 10, ts, 1), c)
os.remove(FILE_NAME)
def test_retrieve_accounts(self):
if os.path.isfile(FILE_NAME):
os.remove(FILE_NAME)
create_db(FILE_NAME)
fund = Funds(FILE_NAME)
fund.create_fund(10, 1)
fund.create_fund(20, 1)
c = fund.retrieve_funds()
ts1 = c[0][2]
ts2 = c[1][2]
self.assertEqual([(1, 10, ts1, 1), (2, 20, ts2, 1)], c)
os.remove(FILE_NAME)
def test_retrieve_transactions(self):
if os.path.isfile(FILE_NAME):
os.remove(FILE_NAME)
create_db(FILE_NAME)
trs = Transactions(FILE_NAME)
trs.create_transaction(10, 1)
trs.create_transaction(20, 1)
c = trs.retrieve_transactions()
ts1 = c[0][2]
ts2 = c[1][2]
self.assertEqual([(1, 10, ts1, 1), (2, 20, ts2, 1)], c)
os.remove(FILE_NAME)
def test_retrieve_portfolio(self):
if os.path.isfile(FILE_NAME):
os.remove(FILE_NAME)
create_db(FILE_NAME)
trs = PortfolioFund(FILE_NAME)
trs.create_fund(10, 1)
trs.create_fund(20, 1)
c = trs.retrieve_portfolio()
ts1 = c[0][2]
ts2 = c[1][2]
self.assertEqual([(1, 10, ts1, 1), (2, 20, ts2, 1)], c)
os.remove(FILE_NAME)
def test_delete_id(self):
if os.path.isfile(FILE_NAME):
os.remove(FILE_NAME)
create_db(FILE_NAME)
trs = PortfolioFund(FILE_NAME)
trs.create_fund(100, 1)
trs.create_fund(200, 1)
c = trs.delete_id(1)
self.assertEqual(True, c)
c = trs.delete_id(2)
self.assertEqual(True, c)
os.remove(FILE_NAME)
def test_delete_id(self):
if os.path.isfile(FILE_NAME):
os.remove(FILE_NAME)
create_db(FILE_NAME)
trs = Transactions(FILE_NAME)
trs.create_transaction(10, 1)
trs.create_transaction(20, 1)
c = trs.delete_id(1)
self.assertEqual(True, c)
c = trs.delete_id(2)
self.assertEqual(True, c)
os.remove(FILE_NAME)
def test_delete_id(self):
if os.path.isfile(FILE_NAME):
os.remove(FILE_NAME)
create_db(FILE_NAME)
fund = Funds(FILE_NAME)
fund.create_fund(10, 1)
fund.create_fund(20, 1)
c = fund.delete_id(1)
self.assertEqual(True, c)
c = fund.delete_id(2)
self.assertEqual(True, c)
os.remove(FILE_NAME)
def test_retrieve_funds_for_account(self):
if os.path.isfile(FILE_NAME):
os.remove(FILE_NAME)
create_db(FILE_NAME)
trs = Funds(FILE_NAME)
trs.create_fund(10, 1)
trs.create_fund(20, 2)
trs.create_fund(30, 1)
trs.create_fund(40, 3)
c = trs.retrieve_funds_for_account(1)
ts1 = c[0][2]
ts2 = c[1][2]
self.assertEqual([(1, 10, ts1, 1), (3, 30, ts2, 1)], c)
os.remove(FILE_NAME)
def retrieve_funds_for_portfolio(self):
if os.path.isfile(FILE_NAME):
os.remove(FILE_NAME)
create_db(FILE_NAME)
trs = PortfolioFund(FILE_NAME)
trs.create_fund(10, 1)
trs.create_fund(20, 2)
trs.create_fund(30, 1)
trs.create_fund(40, 3)
c = trs.retrieve_funds_for_portfolio(1)
ts1 = c[0][2]
ts2 = c[1][2]
self.assertEqual([(1, 10, ts1, 1), (3, 30, ts2, 1)], c)
os.remove(FILE_NAME)
def test_update_id(self):
if os.path.isfile(FILE_NAME):
os.remove(FILE_NAME)
create_db(FILE_NAME)
trs = PortfolioFund(FILE_NAME)
trs.create_fund(10, 1)
c = trs.update_id(id=1, amount=20, portfolio_fk=2)
ts = c[2]
self.assertEqual((1, 20, ts, 2), c)
c = trs.update_id(id=1, amount=10)
ts = c[2]
self.assertEqual((1, 10, ts, 2), c)
c = trs.update_id(id=1, portfolio_fk=1)
ts = c[2]
self.assertEqual((1, 10, ts, 1), c)
c = trs.update_id(id=1, amount=0)
ts = c[2]
self.assertEqual((1, 0, ts, 1), c)
os.remove(FILE_NAME)
def test_update_id(self):
if os.path.isfile(FILE_NAME):
os.remove(FILE_NAME)
create_db(FILE_NAME)
trs = Transactions(FILE_NAME)
trs.create_transaction(10, 1)
c = trs.update_id(id=1, amount=20, account_fk=2)
ts = c[2]
self.assertEqual((1, 20, ts, 2), c)
c = trs.update_id(id=1, amount=10)
ts = c[2]
self.assertEqual((1, 10, ts, 2), c)
c = trs.update_id(id=1, account_fk=1)
ts = c[2]
self.assertEqual((1, 10, ts, 1), c)
c = trs.update_id(id=1, amount=0)
ts = c[2]
self.assertEqual((1, 0, ts, 1), c)
os.remove(FILE_NAME)
def test_update_id(self):
if os.path.isfile(FILE_NAME):
os.remove(FILE_NAME)
create_db(FILE_NAME)
fund = Funds(FILE_NAME)
fund.create_fund(10, 1)
c = fund.update_id(id=1, amount=20, account_fk=2)
ts = c[2]
self.assertEqual((1, 20, ts, 2), c)
c = fund.update_id(id=1, amount=10)
ts = c[2]
self.assertEqual((1, 10, ts, 2), c)
c = fund.update_id(id=1, account_fk=1)
ts = c[2]
self.assertEqual((1, 10, ts, 1), c)
c = fund.update_id(id=1, amount=0)
ts = c[2]
self.assertEqual((1, 0, ts, 1), c)
os.remove(FILE_NAME)
from init_db import db, create_db
create_db()
from server import app, login_manager
from flask import request, render_template, session, redirect, url_for, flash
from flask_login import LoginManager, UserMixin, login_required, login_user, current_user, logout_user
from user import *
from question import *
from audit import *
#from datetime import datetime,timedelta
n = 1
series_num = 4
part_num = 1
ques_num = 0
num = [3, 2, 1]
Final = "False"
curr_audit = None
def next_ques():
global series_num
global part_num
global ques_num
if series_num == 10:
return False
while 1 != 0:
if get_question(series_num, part_num, ques_num + 1) != None:
ques_num += 1
break
elif get_question(series_num, part_num + 1, 0) != None:
part_num += 1
def _init_test(self):
if os.path.isfile(FILE_NAME):
os.remove(FILE_NAME)
create_db(FILE_NAME)
return Account(FILE_NAME)
import traceback
import logging
logging.basicConfig(filename = "log.log",
level = logging.DEBUG,
format = '%(asctime)s - %(name)s - %(levelname)s - %(message)s')
import init_db
from flask import Flask
from flaskext.mysql import MySQL
def log_uncaught_exceptions(ex_cls, ex, tb):
logging.critical(''.join(traceback.format_tb(tb)))
logging.critical('{0}: {1}'.format(ex_cls, ex))
sys.excepthook = log_uncaught_exceptions
init_db.create_db("counter")
init_db.create_tbl("counter", "count ( num INT NOT NULL )")
init_db.create_entry("counter", "count", "count ( num )", "( 0 )")
app = Flask(__name__)
app.config['MYSQL_DATABASE_USER'] = '******'
app.config['MYSQL_DATABASE_HOST'] = 'db'
app.config['MYSQL_DATABASE_DB'] = 'counter'
SELECT_NUM = "SELECT num FROM count;"
UPDATE_NUM = "UPDATE count SET num = {0};"
mysql = MySQL()
mysql.init_app(app)
def execute_sql(mysql, statement, fetch_type='one'):
con = mysql.connect()