python 日k写csv文件

#!/usr/bin/env python
# coding: utf-8
import datetime
import pymongo
import csv
#from datetime import datetime
#from datetime import timedelta

global stocks
global k_ri
global list_code
global conn_list, conn_k

quotes = []





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 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 get_k_ri(code, quotes):
    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")
        #print code, date, open, high, low, close, volume, adj_close


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

    print quotes
    csvfile = file('csv_test.csv', 'wb')
    writer = csv.writer(csvfile)
    writer.writerow(['date', 'open', 'high', 'low', 'close'])
    '''
        data = [
            ('name', '25', '1234567'),
            ('name1', '18', '789456')
        ]
    '''
    writer.writerows(quotes)

    csvfile.close()

    print code, date,"----------------------------------------"

def get_stock_list(code):
    l = stocks.find({"code":code}).limit(1)
    print type(l)
    for i in l:
        name = i.get("name")

    return name


if __name__ == '__main__':
    code = "600004"
    init_mongodb_list()
    name = get_stock_list(code)
    conn_list.close()
    #print name
    init_mongodb_k()
    get_k_ri(code,quotes)

    conn_k.close()


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值