file_db

#!/usr/bin/env python
# coding: utf-8

import pymongo


global stocks, conn_list
global k_ri, conn_k
#global list_code


#list_code = []


def init_mongodb_list():
    global stocks
    global conn_list
    # mongodb_link = 'mongodb://127.0.0.1:27017'
    # mongoClient = MongoClient(mongodb_link)
    conn_list = pymongo.MongoClient("localhost", 27017)
    conn_list.stock_list.authenticate("d", "zz")
    db = conn_list.stock_list
    stocks = db.stocks

def close_mongodb_list():
    conn_list.close()

def init_mongodb_k():
    global k_ri
    global conn_k
    conn_k = pymongo.MongoClient("localhost", 27017)
    conn_k.db_stock.authenticate("d", "zz")
    db = conn_k.db_stock
    k_ri = db.k_ri

def close_mongodb_k():
    conn_k.close()




def get_k_ri(code, quotes):
    print "code", code
    l = k_ri.find({"code": code, "close": {"$gt": 0}}).sort([('date', 1)])
    for i in l:
        code = i.get("code")
        date = i.get("date")
        open = i.get("open")
        high = i.get("high")
        low = i.get("low")
        close = i.get("close")
        volume = i.get("volume")
        adj_close = i.get("adj_close")

        # if (date == "2015-07-21"):
        #print code, date, open, high, low, close, volume, adj_close


        quotes.append((date, open, high, low, close))

        # print quotes


        # print code, date,"----------------------------------------"
'''
def get_code_price_cur(list_code, quotes):
    for code1 in list_code:
        code = code1[0]
        price = code1[4]
        #print "code", code, price,"----------------------------------------"
        l = k_ri.find({"code": code, "close": {"$gt": 0}}).sort([('date', -1)]).limit(1)
        for i in l:
            close = i.get("close")
            if (close < price):
                quotes.append((code1[0], code1[1], code1[2], code1[3], code1[4], code1[5], code1[6], code1[7], code1[8], close, close / price))
'''

def get_stock_list():
    list_code = []
    l = stocks.find()
    for i in l:
        code = i.get("code")
        name = i.get("name")
        list_code.append({"code": code, "name": name})
        # print type(name), name, type(code),code
        # list_code.append({"code": "000001", "name": "上证指数".decode('utf8')})
        # list_code.append({"code": "399001", "name": "深证指数".decode('utf8')})
        # list_code.append({"code": "399006", "name": "创业板指".decode('utf8')})

    return list_code