def listenabled(options, profile='default', all=False): '''Lista las comasters''' from models import Profile, COMaster if not all: cos = Profile.get(name=profile).comaster_set else: cos = COMaster.select() for co in Profile.get(name=profile).comaster_set: print co.address, co.enabled
def enableone(options, comaster='192.168.1.98', profile='default'): '''Hablita un comaster en particular %s''' % comaster from models import Profile comasters = Profile.get(name=profile).comaster_set comasters.exclude(address=comaster).update(enabled=False) co = comasters.get(address=comaster) co.enabled = True co.save()
def testBuyTradeLikelihood(self): prediction_key = Prediction(contract_one=0.00, contract_two=0.00, liquidity=100, statement="Test", end_time=datetime.datetime.now()).put() user_key = Profile(balance=100).put() profile = user_key.get() prediction = prediction_key.get() probability = 90 trade_likeliness = calculate_trade_from_likelihood( probability, prediction, profile) self.assertEqual(29.789603833999628, trade_likeliness.quantity)
def maraserver(options): """Run server protocol""" from models import database, Profile from twisted.internet import reactor try: database.connect() profile = Profile.get(name=options.profile) #import IPython; IPython.embed() for comaster in profile.comaster_set.filter(enabled=True): print "Listening in port %d" % comaster.port reactor.listenTCP(comaster.port, MaraServerFactory(comaster)) #@UndefinedVariable reactor.run() #@UndefinedVariable except Exception: from traceback import format_exc print format_exc() raise
def testScoringCase(self): prediction_key = Prediction(contract_one=0.00, contract_two=0.00, liquidity=100, resolved=False, outcome='CONTRACT_ONE', statement='Test', end_time=datetime.datetime.now()).put() user_key = Profile(balance=100, user_ledger=[ LedgerRecords( prediction_id=prediction_key.urlsafe(), contract_one=10.00, contract_two=0.00) ]).put() trade_key = Trade(prediction_id=prediction_key, user_id=user_key, direction='BUY', contract='CONTRACT_ONE', quantity=10).put() user = user_key.get() audit = scoring() self.assertEqual(10, audit[0]['earned'])
def maraclient(options): ''' Conexión con el mara ''' from twisted.internet import reactor try: from models import database, Profile database.connect() profile = Profile.get(name=options.profile) for comaster in profile.comaster_set.filter(enabled=True): print "Conectando con %s" % comaster reactor.connectTCP(comaster.address, comaster.port, MaraClientProtocolFactory(comaster), timeout=comaster.timeout, #bindAddress=None, ) # Una vez que está conectado todo, debemos funcionar... reactor.run() #@UndefinedVariable except Exception as e: from traceback import format_exc print format_exc() print e