def add():
form = Contract_add()
time = str(datetime.datetime.today()).split(' ')[0].split('-')
nav = [
'Добавить договор',
[['Лист неоплативших', '/unpaid'], ['Архив', '/archive'],
['Все договоры', '/all'], ['История', '/history'],
['Удалить контракт', '/delete']]
]
history_cancel = History_cancel()
history_clear = History_clear()
deals = get_non_deleted_deals()
if form.validate_on_submit():
contract = Contract(number=form.number.data,
is_arch=form.is_arch.data,
date=form.date.data,
paid=form.paid.data)
db.session.add(contract)
db.session.commit()
# return redirect(url_for('quiz_2'))
print(form.data)
return render_template('add-user.html',
form=form,
time=1,
nav=nav,
history_clear=history_clear,
deals=deals,
history_cancel=history_cancel)
def counter_contract(contract_id):
contract = Contract.query.filter(Contract.id == contract_id).first_or_404()
parent = contract if contract.parent_id is None else Contract.query.get(contract.parent_id)
role = 'owner' if contract.owner_id == current_user.id else 'cparty'
if 'counter' not in contract_transitions[contract.status][role]:
flash('This action is not permitted')
return redirect(url_for('index'))
form = EditProposalForm()
form.template_id.choices = [(t.id, t.title) for t in Template.query.order_by('title')]
if form.validate_on_submit():
proposal = Contract(template_id=form.template_id.data, params=form.params.data, memo=form.memo.data, status="draft", owner_id=current_user.id, parent_id=parent.id)
db.session.add(proposal)
db.session.flush()
for p in contract.party:
party = Party(contract_id=proposal.id, role=p.role, user_id=p.user_id)
db.session.add(party)
db.session.flush()
description = 'Created a counter draft'
alog = ActivityLog(contract_id=proposal.id, timestamp=datetime.now(), method='/contract/counter', description=description, user_id=current_user.id)
db.session.add(alog)
db.session.commit()
flash('Draft edited.')
return redirect(url_for('index'))
form.template_id.data = contract.template_id
form.params.data = contract.params
return render_template('edit_draft.html', title='Edit a Draft Proposal', form=form, contract_id=contract_id, parties=contract.party, deal=parent.memo)
def test_create_invalid_contract(self):
contract_data = {
'organizer_email': '1234',
'signed_date': INVALID_SIGN_DATE,
'description': 'some description',
'case_number': '345978',
'salesforce_id': '23465789',
'salesforce_case_id': '4680990',
}
expected_error_dict_messages = {
'link_to_salesforce_case': ['This field cannot be blank.'],
'organizer_account_name': ['This field cannot be blank.'],
'organizer_email': ['Enter a valid email address.'],
'signed_date': [
"'{}' value has an invalid date format. "
"It must be in YYYY-MM-DD format.".format(INVALID_SIGN_DATE)
]
}
contract = Contract(**contract_data)
with self.assertRaises(ValidationError) as cm:
contract.full_clean()
self.assertEqual(expected_error_dict_messages,
cm.exception.message_dict)
def setUp(self):
# create contract
contract = Contract.find_contract_by_id(1)
if contract is None:
contract = Contract(id=1,
contract_name="contract1",
contract_address="0x123",
json_name="name1")
db.session.add(contract)
db.session.commit()
# create match
match = Match.find_match_by_id(1)
if match is None:
match = Match(id=1)
db.session.add(match)
db.session.commit()
# create user
user = User.find_user_with_id(88)
if user is None:
user = User(id=88)
db.session.add(user)
db.session.commit()
user = User.find_user_with_id(99)
if user is None:
user = User(id=99)
db.session.add(user)
db.session.commit()
user = User.find_user_with_id(66)
if user is None:
user = User(id=66)
db.session.add(user)
db.session.commit()
# create outcome
outcome = Outcome.find_outcome_by_id(88)
if outcome is None:
outcome = Outcome(id=88,
match_id=1,
hid=88,
contract_id=contract.id)
db.session.add(outcome)
db.session.commit()
else:
outcome.contract_id = contract.id
def test_create_valid_contract(self):
contract_data = {
'organizer_account_name': 'Planner Eventos',
'organizer_email': '*****@*****.**',
'signed_date': '2019-09-14',
'description': 'some description',
'case_number': '345978',
'salesforce_id': '23465789',
'salesforce_case_id': '4680990',
'link_to_salesforce_case': 'https://pe33.zzxxzzz.com/5348fObs',
}
contract = Contract(**contract_data)
contract.full_clean()
contract_data.pop('signed_date')
contract.full_clean()
def contract_create():
try:
data = request.get_json()
except:
return jsonify( {'status': 'error', 'error': 'missing data'} )
if data == None:
return jsonify( {'status': 'error', 'error': 'missing data'} )
# query if contract already exists
current_app.logger.info('check contract: ' + data['localSymbol'])
contract = Contract.query.get(data['localSymbol'])
if contract != None:
current_app.logger.info('existing contract')
else:
if 'localSymbol' in data:
if data['localSymbol'] == '':
return jsonify( {'status': 'error', 'error': 'empty localSymbol'} )
else:
contract = Contract(
localSymbol=data['localSymbol']
)
else:
return jsonify( {'status': 'error', 'error': 'missing localSymbol'} )
if 'strike' in data:
if data['strike'] > 0:
contract.strike = data['strike']
if 'right' in data:
if data['right'] != '':
contract.right = data['right']
# no processing
std_fields = ['secType', 'symbol', 'currency', 'exchange', 'primaryExchange', 'lastTradeDateOrContractMonth', 'multiplier' ]
for std_field in std_fields:
if std_field in data:
setattr(contract, std_field, data[std_field])
db.session.add(contract)
db.session.commit()
current_app.logger.info('new contract')
return jsonify( {'status': 'ok'} )
def create_draft():
form = CreateProposalForm()
form.template_id.choices = [(t.id, t.title) for t in Template.query.order_by('title')]
if form.validate_on_submit():
proposal = Contract(template_id=form.template_id.data, memo=form.memo.data, params=form.params.data, status="draft", owner_id=current_user.id)
db.session.add(proposal)
db.session.flush()
for p in json.loads(form.parties.data):
party = Party(contract_id=proposal.id, role=p['label'], user_id=p['user_id'])
db.session.add(party)
db.session.flush()
description = 'Creation of a new draft'
alog = ActivityLog(contract_id=proposal.id, timestamp=datetime.now(), method='/contract/new', description=description, user_id=current_user.id)
db.session.add(alog)
db.session.commit()
flash('Draft saved.')
return redirect(url_for('index'))
contacts = User.query.all()
return render_template('create_draft.html', title='Create a new Draft Proposal', form=form, contacts=contacts)
def showContracts():
date = datetime.today() + timedelta(2)
form = NewContract()
if form.validate_on_submit():
newContract = Contract(number=form.number.data,
sign_date=form.sign_date.data,
ground=form.ground.data,
company=form.company.data,
notes=form.notes.data,
customer=form.customer.data,
supply=form.supply.data,
author=current_user)
db.session.add(newContract)
db.session.commit()
flash('Новый контракт в "{}" добавлен!'.format(form.customer.data))
return redirect(url_for('contract.showContracts'))
contracts = Contract.query.order_by(Contract.sign_date.asc())
return render_template("/contract/contracts.html",
title='Контракты',
contracts=contracts,
form=form,
date=date)
def add(): try: uid = int(request.headers['Uid']) data = request.json if data is None: return response_error(MESSAGE.INVALID_DATA, CODE.INVALID_DATA) response_json = [] for item in data: contract = Contract( contract_name=item['contract_name'], contract_address=item['contract_address'], json_name=item['json_name'] ) db.session.add(contract) db.session.flush() response_json.append(contract.to_json()) db.session.commit() return response_ok(response_json) except Exception, ex: db.session.rollback() return response_error(ex.message)
def process_bill():
if request.method == "POST":
file = request.files["inputFile"]
if not file.filename:
flash("No se ha seleccionado ningún archivo")
return redirect(url_for("customer.upload_bill"))
if file and __allowed_file(file.filename):
# save file to upload directory with a hash code
file_extension = file.filename.rsplit(".", 1)[1].lower()
filename = str(uuid.uuid4()) + "." + file_extension
bill_path = os.path.join(app.config["UPLOAD_FOLDER"], filename)
file.save(bill_path)
# information extraction from the bill
results = docreco.process_bill(bill_path, file_extension)
# return results
# Delete the bill uploaded
os.remove(bill_path)
contract_number = __get_first_value(results["Datos del contrato"]["ReferenciaContrato"]) \
.split('/')[0] \
.split('-')[0] \
.split(' ')[0]
if contract_number:
contract = Contract.get_by_contract_number(contract_number)
if not contract:
company_name = __get_first_value(
results["Datos de la factura"]["Comercializadora"])
if company_name:
trading_company = Company.get_trading_company_by_name(
company_name, unidecode.unidecode(company_name))
if trading_company:
cif = trading_company.cif
else:
flash(
"No se encuentra la comercializadora ni existe cif en la factura"
)
return redirect(url_for("customer.my_bills"))
else:
flash(
"No se encuentra el nombre de la comercializadora en la factura"
)
return redirect(url_for("customer.my_bills"))
contract_data = __get_contract_data(results)
contract = Contract(
contract_number=contract_number,
contracted_power=contract_data["contracted_power"],
toll_access=contract_data["toll_access"],
end_date=contract_data["end_date"],
CNAE=contract_data["CNAE"],
tariff_access=contract_data["tariff_access"],
cif=cif)
contract.save()
else:
flash("No se encuentra el número de referencia del contrato")
return redirect(url_for("customer.my_bills"))
invoice_data = __get_invoice_data(results, contract_number)
invoice = Invoice(
invoice_number=invoice_data["invoice_number"],
contracted_power_amount=invoice_data[
"contracted_power_amount"],
consumed_energy_amount=invoice_data["consumed_energy_amount"],
issue_date=invoice_data["issue_date"],
charge_date=invoice_data["charge_date"],
init_date=invoice_data["init_date"],
end_date=invoice_data["end_date"],
total_amount=invoice_data["total_amount"],
contract_reference=invoice_data["contract_reference"],
contract_number=invoice_data["contract_number"],
document=file.read())
try:
invoice.save()
except IntegrityError:
flash("Esta factura ya está registrada")
return redirect(url_for("customer.my_bills"))
cups = __get_first_value(results["Datos del contrato"]["CUPS"])
if cups and not Dwelling.get_by_cups(cups):
__create_dwelling_with_cups(results, cups)
else:
cups = __create_dwelling_with_random_cups(results)
customer_dwelling_contract = Customer_Dwelling_Contract(
nif=Customer.get_by_user_id(current_user.id).nif,
cups=cups,
contract_number=contract_number)
customer_dwelling_contract.save()
flash("La factura se ha guardado con éxito")
return redirect(
url_for("customer.show_bill",
invoice_number=invoice.invoice_number))
else:
flash(
"Los tipos de fichero permitidos son txt, pdf, png, jpg, jpeg, gif"
)
return redirect(url_for("customer.upload_bill"))
return "Error POST"
def submit(request):
contractname = request.POST['contractname']
a = request.POST['a']
b = request.POST['b']
c = request.POST['c']
d = request.POST['d']
e = request.POST['e']
f = request.POST['f']
g = request.POST['g']
h = request.POST['h']
i = request.POST['i']
j = request.POST['j']
k = request.POST['k']
l = request.POST['l']
m = request.POST['m']
n = request.POST['n']
o = request.POST['o']
time_format = time.strftime('%Y-%m-%d_%H%M%S', time.localtime(time.time()))
file = open('LCR_' + time_format + '.txt', 'wt')
file.write('Letter of Credit' + '\n'
'1.Advising bank:' + a + '\n'
'2.Credit No.:' + b + '\n'
'3.Beneficiary:' + c + '\n'
'4.Applicant:' + d + '\n'
'5.L/C Amount and Tolerance:' + e + '\n'
'6.Type:' + f + '\n'
'7.Partial shipment:' + g + '\n'
'8.Transshipment:' + h + '\n'
'9.Trnasport mode:' + i + '\n'
'10.Loading(shipment from):' + j + '\n'
'11.Discharging(shipment to):' + k + '\n'
'12.Latest shipment date:' + l + '\n'
'13.All banking charges:' + m + '\n'
'14.Confirmation:' + n + '\n'
'15.T/T reimbursement:' + o + '\n')
file.close()
file = open('LCR_' + time_format + '.txt', 'rb')
data = file.read()
# hasher = hashlib.md5()
# with open('myfile.jpg', 'rb') as afile:
# buf = afile.read()
# hasher.update(buf)
# print(hasher.hexdigest())
# a = 'MD5 : ' + hashlib.md5(data).hexdigest()
# b = 'SHA-1 : ' + hashlib.sha1(data).hexdigest()
hash = 'SHA-256 : ' + hashlib.sha256(data).hexdigest()
file.close()
# 데이터 저장
contract = Contract(contractname=contractname,
sha256=hash,
filename='LCR_' + time_format + '.txt')
# 로그인한 사용자 정보를 Contract에 같이 저장
user_id = request.session['user_id']
member = Member.objects.get(user_id=user_id)
contract.owner = member
contract.save()
return redirect('ing')
def create():
try:
data = request.get_json()
except:
return jsonify({'status': 'error', 'error': 'missing data'})
if data == None:
return jsonify({'status': 'error', 'error': 'missing data'})
# query if contract already exists
current_app.logger.info('check contract: ' + data['localSymbol'])
contract = Contract.query.get(data['localSymbol'])
if contract != None:
current_app.logger.info('existing contract')
else:
if data['localSymbol'] == '':
localSymbol = data['symbol']
else:
localSymbol = data['localSymbol']
if data['strike'] > 0:
strike = data['strike']
else:
strike = None
if data['right'] != '':
right = data['right']
else:
right = None
contract = Contract(
secType=data['secType'],
localSymbol=localSymbol,
symbol=data['symbol'],
currency=data['currency'],
exchange=data['exchange'],
primaryExchange=data['primaryExchange'],
lastTradeDateOrContractMonth=data[
'lastTradeDateOrContractMonth'], #datetime
multiplier=data['multiplier'],
strike=strike,
right=right)
db.session.add(contract)
db.session.commit()
current_app.logger.info('new contract')
# insert executions
current_app.logger.info('check execution: ' + data['execId'])
execution = Execution.query.get(data['execId'])
if execution != None:
current_app.logger.info('existing execution')
else:
shares = abs(data['shares'])
cumQty = abs(data['cumQty'])
if data['side'][0].upper() == 'B':
pass
elif data['side'][0].upper() == 'S':
shares = -shares
cumQty = -cumQty
execution = Execution(
execId=data['execId'],
orderId=data['orderId'],
asset=contract, # object
time=datetime.strptime(data['time'],
'%Y%m%d %H:%M:%S'), # 20180604 13:32:52
acctNumber=data['acctNumber'],
exchange=data['exchange'],
side=data['side'],
shares=shares,
cumQty=cumQty,
price=data['price'],
avgPrice=data['avgPrice'],
permId=data['permId'])
db.session.add(execution)
db.session.commit()
current_app.logger.info('new execution')
return jsonify({
'status': 'ok',
'inputData': data,
'contract': {
'localSymbol': contract.localSymbol
},
'order': {
'orderId': execution.orderId
},
'execution': {
'execId': execution.execId
}
})
def process_bill():
try:
file = request.files["file"]
except BadRequestKeyError:
return {
"message": "No se ha seleccionado ningún archivo",
"type": "error"
}, 200
if not __allowed_file(file.filename):
return {
"message":
"Los tipos de fichero permitidos son txt, pdf, png, jpg, jpeg, gif",
"type": "error"
}, 200
# save file to upload directory with a hash code
file_extension = file.filename.rsplit(".", 1)[1].lower()
filename = str(uuid.uuid4()) + "." + file_extension
bill_path = os.path.join(app.config["UPLOAD_FOLDER"], filename)
file.save(bill_path)
# information extraction from the bill
results = docreco.process_bill(bill_path, file_extension)
# Delete the bill uploaded
os.remove(bill_path)
contract_number = __get_first_value(
results["Datos del contrato"]["ReferenciaContrato"]).split(
'/')[0].split('-')[0].split(' ')[0]
if contract_number:
contract = Contract.get_by_contract_number(contract_number)
if not contract:
cif = __get_first_value(results["Datos de la factura"]["CIF"])
if cif:
trading_company = Company.get_by_cif(cif)
if not trading_company:
return {
"message": "No se encuentra la comercializadora",
"type": "error"
}, 200
else:
company_name = __get_first_value(
results["Datos de la factura"]["Comercializadora"])
if company_name:
trading_company = Company.get_trading_company_by_name(
company_name, unidecode.unidecode(company_name))
if trading_company:
cif = trading_company.cif
else:
return {
"message":
"No se encuentra la comercializadora ni el cif en la factura",
"type": "error"
}, 200
else:
return {
"message":
"No se encuentra el nombre de la comercializadora en la factura",
"type": "error"
}, 200
contract_data = __get_contract_data(results)
contract = Contract(
contract_number=contract_number,
contracted_power=contract_data["contracted_power"],
toll_access=contract_data["toll_access"],
end_date=contract_data["end_date"],
CNAE=contract_data["CNAE"],
tariff_access=contract_data["tariff_access"],
cif=cif)
contract.save()
else:
return {
"message": "No se encuentra el número de referencia del contrato",
"type": "error"
}, 200
invoice_data = __get_invoice_data(results, contract_number)
invoice = Invoice(
invoice_number=invoice_data["invoice_number"],
contracted_power_amount=invoice_data["contracted_power_amount"],
consumed_energy_amount=invoice_data["consumed_energy_amount"],
issue_date=invoice_data["issue_date"],
charge_date=invoice_data["charge_date"],
init_date=invoice_data["init_date"],
end_date=invoice_data["end_date"],
total_amount=invoice_data["total_amount"],
contract_reference=invoice_data["contract_reference"],
contract_number=invoice_data["contract_number"],
document=file.read())
try:
invoice.save()
except IntegrityError:
return {
"message": "Esta factura ya está registrada",
"type": "error"
}, 200
cups = __get_first_value(results["Datos del contrato"]["CUPS"])
if cups:
if not Dwelling.get_by_cups(cups):
__create_dwelling_with_cups(results, cups)
else:
cups = __create_dwelling_with_random_cups(results)
logged_user = User.get_by_username(get_jwt_identity())
logged_customer = Customer.get_by_user_id(logged_user.id)
nif = logged_customer.nif
if not Customer_Dwelling_Contract.get_by_nif_and_contract_number(
nif, contract_number):
customer_dwelling_contract = Customer_Dwelling_Contract(
nif=nif, cups=cups, contract_number=contract_number)
try:
customer_dwelling_contract.save()
except IntegrityError:
pass
return {
"message": "La factura se ha guardado con éxito",
"type": "success"
}, 200
def init_data_before_test(self):
# create token
token = Token.find_token_by_id(1)
if token is None:
token = Token(id=1, name="SHURIKEN", symbol="SHURI", decimal=18)
db.session.add(token)
db.session.commit()
# create contract
contract = Contract.find_contract_by_id(1)
if contract is None:
contract = Contract(id=1,
contract_name="contract1",
contract_address="0x123",
json_name="name1")
db.session.add(contract)
db.session.commit()
# create match
match = Match.find_match_by_id(1)
if match is None:
match = Match(id=1)
db.session.add(match)
db.session.commit()
# create user
user = User.find_user_with_id(88)
if user is None:
user = User(id=88)
db.session.add(user)
db.session.commit()
user = User.find_user_with_id(99)
if user is None:
user = User(id=99)
db.session.add(user)
db.session.commit()
user = User.find_user_with_id(100)
if user is None:
user = User(id=100)
db.session.add(user)
db.session.commit()
user = User.find_user_with_id(109)
if user is None:
user = User(id=109)
db.session.add(user)
db.session.commit()
user = User.find_user_with_id(66)
if user is None:
user = User(id=66)
db.session.add(user)
db.session.commit()
# create outcome
outcome = Outcome.find_outcome_by_id(88)
if outcome is None:
outcome = Outcome(id=88,
match_id=1,
hid=88,
contract_id=contract.id)
db.session.add(outcome)
db.session.commit()
else:
outcome.result = -1
outcome.contract_id = contract.id
db.session.commit()
# add redeem
try:
for i in range(1, 3):
chars = string.ascii_uppercase + string.ascii_lowercase
code = ''.join(random.choice(chars) for _ in range(i))
if Redeem.find_redeem_by_code(code) is None:
r = Redeem(code=code)
db.session.add(r)
db.session.commit()
except Exception as ex:
db.session.rollback()