utils.py

from pymongo import Connection
import otterapi
from random import randint
Connection=Connection('mongo2.stuycs.org')

db=Connection.admin
res=db.authenticate('ml7','ml7')
players=db.players
market=db.market
bank=db.bank
date=db.date
count = 0
#name:string
def add_user(name):
    db=Connection['EmotionStock']
    if not name in get_players():
        player={"name":str(name),"money":10000,"soul":100,'stocks':[]}
        players.insert(player)
    else:
        return False

#returns a list
def get_players():
    db=Connection['EmotionStock']
    names=[]
    for line in players.find():
        names.append(line['name'])
    return names

#returns an int
def get_money(user):
    if authenticate(user):
        db=Connection['EmotionStock']
        name=players.find_one({"name":user})
        return name["money"]
    else:
        return False

'''
parameters:
        name:string
returns:
        boolean
'''
def authenticate(name):
    db=Connection["EmotionStock"]
    if not name in get_players():
        return False
    return True

def auth_stock(name):
    db=Connection["EmotionStock"]
    stock=market.find_one({"stock":name})
    if stock == None:
        return False
    return True

def get_market_x():
    db=Connection["EmotionStock"]
    x={}
    for name in get_stocks_names():
        x[name]=(get_x(name))
    return x

def get_market_y():
    db=Connection["EmotionStock"]
    y={}
    for name in get_stocks_names():
        y[name]=(get_y(name))
    return y

def get_x(stock):
    db=Connection["EmotionStock"]
    name=market.find_one({"stock":stock})
    x=[]
    for data in name["data"]:
        x.append(data["time"])
    return x

def get_y(stock):
    db=Connection["EmotionStock"]
    name=market.find_one({"stock":stock})
    y=[]
    for data in name["data"]:
        y.append(data["price"])
    return y

'''
parameters:
      name:string
      stock:string
      count:int
returns:
      boolean
'''
def buy_stock(name,stock,count):
    db=Connection["EmotionStock"]
    if (authenticate(name) and auth_stock(stock)):
        user=players.find_one({"name":name})
        cost=get_stock_price(stock)*count
        if (user["money"]<cost or count < 0):
            return False
        else:
            user["money"]=user["money"]-cost
        if stock in get_stocks_user(name):
            mystocks=user["stocks"]
            for i in mystocks:
                if i["stock"]==stock:
                    i["shares"]=i["shares"]+count
        else:
            stocks={"stock":stock,"shares":count}
            user["stocks"].append(stocks)
        players.update({"name":name},user)
        return True
    return False

'''
parameters:
      name:string
      stock:string
      count:int
returns:
      boolean
'''
def sell_stock(name,stock,count):
    db=Connection["EmotionStock"]
    if (authenticate(name) and auth_stock(stock)):
        stocks=get_stocks(name)[0]
        for item in stocks:
            if item["stock"]==stock:
                if (item["shares"]<count):
                    return False
                item["shares"]=item["shares"]-count
                if item["shares"]==0:
                    stocks.remove(item)
        user=players.find_one({"name":name})
        user["stocks"]=stocks
        gain=get_stock_price(stock)*count
        user["money"]=user["money"]+gain
        players.update({"name":name},user)
        return True
    return False

#return: list of stocks for a specific user
def get_stocks(user):
    db=Connection["EmotionStock"]
    if(authenticate(user)):
        name=players.find_one({"name":user})
        stock=[]
        gain=0
        for item in name["stocks"]:
            tmp_name=item["stock"]
            tmp_share=item["shares"]
            tmp_price=get_stock_price(tmp_name)*tmp_share
            gain=gain+tmp_price
            x={"shares":tmp_share,"stock":tmp_name,"price":tmp_price}
            stock.append(x)
        return [stock,gain]

#return: names of stocks for a specific user
def get_stocks_user(user):
    db=Connection["EmotionStock"]
    if(authenticate(user)):
        name=players.find_one({"name":user})
        stock=[]
        for item in name["stocks"]:
            stock.append(item["stock"])
        return stock

#get all the stocks' data 
def get_market():
    db=Connection["EmotionStock"]
    stocks=[]
    for item in market.find():
        stocks.append(item)
    return stocks

#get all stocks' name in the market
def get_stocks_names():
    db=Connection["EmotionStock"]
    stocks=get_market()
    names=[]
    for item in stocks:
        names.append(str(item["stock"]))
    return names

#get a specific stock's dictionary 
def get_stock(name):
    db=Connection["EmotionStock"]
    return market.find_one({"stock":name})

#get a specific stock's price
def get_stock_price(name):
    db=Connection["EmotionStock"]
    stock=market.find_one({"stock":name})
    return stock["data"][len(stock["data"])-1]["price"]

def get_icon(name):
    db=Connection["EmotionStock"]
    stock=market.find_one({"stock":name})
    price = stock["data"][len(stock["data"])-1]["price"]
    last_price = stock["data"][len(stock["data"])-2]["price"]
    if price>last_price:
        return ["Stock Index Up.png",abs(price-last_price)]
    if price==last_price:
        return ["zoom_out.png",abs(price-last_price)]
    if price<last_price:
        return ["Stock Index Down.png",abs(price-last_price)]

#soul:int
def sell_soul(name,soul):
    db=Connection["EmotionStock"]
    if get_soul(name)<=soul:
        return False
    else:
        user=players.find_one({"name":name})
        user["money"]=user["money"]+soul*100
        user["soul"]=user["soul"]-soul
        if in_bank(name):
            buser=bank.find_one({"name":name})
            buser["soul"]=buser["soul"]+soul
            bank.update({"name":name},buser)
        else:
            nbalance={"name":name,"soul":soul}
            bank.insert(nbalance)
        players.update({"name":name},user)
        return True

def buy_soul(name,soul):
    db=Connection["EmotionStock"]
    user=players.find_one({"name":name})
    if get_bank_soul(name)>=soul and user["money"]>=soul*100:
        user["money"]=user["money"]-(soul*100)
        user["soul"]=user["soul"]+soul
        buser=bank.find_one({"name":name})
        buser["soul"]=buser["soul"]-soul
        bank.update({"name":name},buser)
        if buser["soul"]==0:
            bank.remove(buser)
        players.update({"name":str(name)},user)
        return True
    return False

#returns an int 
def get_soul(name):
    db=Connection["EmotionStock"]
    user=players.find_one({"name":name})
    return user["soul"]

def get_bank_soul(name):
    db=Connection["EmotionStock"]
    if in_bank(name):
        user=bank.find_one({"name":name})
        return user["soul"]
    else:
        return -1

def in_bank(name):
    db=Connection["EmotionStock"]
    names=[]
    for line in bank.find():
        names.append(line["name"])
    if not name in names:
        return False
    return True

'''
called daily by JS timer
'''

def update_market():
    db=Connection["EmotionStock"]
    names=get_stocks_names()
    for name in names:
        test(name)

def update_price(name):
    db=Connection["EmotionStock"]
    stock=market.find_one({"stock":name})
    if len(stock["data"])>6:
        #pops the first one
        market.update({"stock":name},{"$pop":{"data":-1}});
    count=otterapi.setup(name)[1]
    time=otterapi.setup(name)[0]
    #print time
    price = (float(count)-float(stock["last count"]))/float(stock["last count"])
    price = float('%.1f' % (round(price,1)))+1
    price = int(stock["data"][len(stock["data"])-1]["price"]*price)

    price = randint(-5,5)+price
    if price<1:
        price=price*(randint(1,3))
    if price == 0:
        price = 1
    #print [price, count, stock["last count"]]
    #print [count, stock["last count"]]
    data=({"time":time,"price":price})
    stock["last count"]=count
    #print [price, count]
    market.update({"stock":name},{"$push":{"data":data}})
    market.update({"stock":name},{"$set":{"last count":count}})

def fix(name):
    db=Connection["EmotionStock"]
    stock=market.find_one({"stock":name})
    data=stock["data"]
    data[len(data)-2]["time"]="01/20"
    market.update({"stock":name},{"$set":{"data":data}})
    market.update({"stock":name},{"$pop":{"data":-1}})
    
def test(name):
    db=Connection["EmotionStock"]
    stock=market.find_one({"stock":name})
    if len(stock["data"])>6:
        market.update({"stock":name},{"$pop":{"data":-1}});
    times=otterapi.create_times()
    count=otterapi.pre_set(name,times[5],times[6])[1]
    time=otterapi.pre_set(name,times[5],times[6])[0]
    price=stock["data"][len(stock["data"])-1]["price"]*(1+((count-stock["last count"])/(stock["last count"]*100)))
    price = (float(count)-float(stock["last count"]))/float(stock["last count"])
    price = float('%.1f' % (round(price,1)))+1
    price = int(stock["data"][len(stock["data"])-1]["price"]*price)
    price = randint(-5,5)+price
    if price<1:
        price=1
    print [price, count, stock["last count"]]
    #print [count, stock["last count"]]
    data=({"time":time,"price":price})
    stock["last count"]=count
    market.update({"stock":name},{"$push":{"data":data}})
    market.update({"stock":name},{"$set":{"last count":count}})


def market_setup():
    db=Connection["EmotionStock"]
    names=["happy","love","sad","tired","bored","mad","sick"]
    for name in names:
       # times=otterapi.create_times()
       # count=otterapi.pre_set(name,times[0],times[1])[1]
       # time=otterapi.pre_set(name,times[0],times[1])[0]
        stock={"stock":name,"last count":10000,"data":[{"time":"01/14","price":40}]}
        market.insert(stock)

def date_setup():
    db=Connection["EmotionStock"]
    last_date={"last date":otterapi.get_date()}
    date.insert(last_date)

def update_date(stuff):
    db=Connection["EmotionStock"]
    date.update({"last date":get_date()},{"$set":{"last date":stuff}})

def get_date():
    db=Connection["EmotionStock"]
    tmp=date.find_one()
    return tmp["last date"]

'''
testing purpose only
'''
def delete_players():
    db=Connection["EmotionStock"]
    players.drop()

def delete_market():
    db=Connection["EmotionStock"]
    market.drop()
def delete_stuff(name):
    db=Connection["EmotionStock"]
    stock=market.find_one({"stock":name})
    #print stock
    market.update({"stock":name},{"$pop":{"data":len(stock["data"])-1}});

if __name__=="__main__":
    #market_setup()
    name="mengdi"
    stock="happy"
    names=["happy","love","sad","tired","bored","mad","sick"]
  
 
    print get_stock("mad")
    #delete_market()
    #market_setup()
    #update_market()
   
    #print otterapi.real_time(get_date())
    #update_price(names[0])
    #print otterapi.real_time(get_date())
    #print otterapi.real_time(get_date())
    #print get_stocks_names();
    #print get_stock(names[0])
    # print get_market()
    #update_market()
    #add_user(name)
    #buy_stock(name,stock,count)
    #print auth_stock("happy")
    #sell_stock(name,stock,10)
    #print get_stocks(name)
    #sell_soul(name,10)
    #buy_soul(name,10)

    #print bank.find_one({"name":"mengdi"})
    #print players.find_one({"name":"mengdi"})
    #add_user(name)
    #buy_stock(name,stock,count)
    #print get_stock(name)