def calculate(): from src.calc.calcroyalties import ProcessRoyalties pr = ProcessRoyalties() print("We are in the calculate thing..... for ", request.args) db = config.get_database() for monthly in db.select('Monthly', ExtractDate=int(request.args.get('ExtractDate')), ProdMonth=int(request.args.get('ProdMonth')), Entity=request.args.get('Entity'), EntityId=int(request.args.get('EntityID'))): try: print("about to calculate...") pr.process_one(monthly.ExtractDate, monthly.Entity, monthly.EntityID, monthly.ProdMonth, monthly.Product) except Exception as e: print("We have an error") traceback.print_exc(file=sys.stdout) tb = traceback.format_exc() return "<h2>Error displaying worksheet for " + monthly.Entity + " %s</h2><br>" % monthly.EntityID + \ str(e) + '<plaintext>' + \ tb + '</plaintext>' # print(e) # return 'Something went wrong during calculation for %s, %s, %i, %s:<br />%s' % \ # (monthly.Entity, monthly.EntityID, monthly.ProdMonth, monthly.Product, str(e)) return "Processing complete."
def calculate(): from src.calc.calcroyalties import ProcessRoyalties pr = ProcessRoyalties() well_id = int(request.args.get('WellId')) try: pr.process_one(well_id, get_proddate_int(), 'Oil') return 'Calculation successful for %s, %i, %s' % (well_id, get_proddate_int(), 'Oil') except Exception as e: print(e) return 'Something went wrong during calculation for %s, %i, %s:<br />%s' % \ (well_id, get_proddate_int(), 'Oil', str(e))
def test_process_monthly(self): db = config.get_database() dbu = DatabaseUtilities() dbu.delete_all_tables() dbu.create_some_test_well_royalty_masters() dbu.create_some_test_lease_royalty_masters() dbu.create_some_test_leases() dbu.create_some_test_well_lease_link() dbu.create_some_test_monthly() dbu.create_some_test_econdata() dbu.create_some_test_rtp_info() dbu.create_calc() pr = ProcessRoyalties() pr.process_one(4, 201501, "OIL") # Check to see if the oper record exists self.assertEqual(2, db.count("calc")) pr.process_all() self.assertEqual(3, db.count("calc"))