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
