#!/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()
05-07
2万+
09-26
09-26
09-26
04-01