In [1]:

    

import os
from pymongo import MongoClient

def get_code_from_colname(colname):
    (market, code) = colname.split('_')
    return code + '.' + market

wind_db_name = os.environ.get('WIND_DB', 'emquant')
chan_db_name = os.environ.get('CHAN_DB', 'emchan')
db_ip = os.environ.get('MONGO_DB_IP_ADDR', '172.17.0.1')
db_port = int(os.environ.get('MONGO_DB_PORT', 27017))

# wind_db_name = os.environ.get('WIND_DB', 'wind')
# chan_db_name = os.environ.get('CHAN_DB', 'chan')
# db_ip = os.environ.get('MONGO_DB_IP_ADDR', '192.168.1.103')
# db_port = int(os.environ.get('MONGO_DB_PORT', 27017))

client = MongoClient(db_ip, db_port)
db_wind = client[wind_db_name]
db_chan = client[chan_db_name]

chankline_col = db_chan['chankline']

code_list = db_wind.collection_names(include_system_collections=False)
code_list = [get_code_from_colname(code) for code in code_list if 'SH' in code or 'SZ' in code]

for code in code_list:
    filter = {'windCode': code, 'ktype': '1_1'}
    cursor = chankline_col.find(filter).sort('index', -1).limit(1)
    for doc in cursor:
        if doc['inclusive'] != 0:
            print(doc)
        
print('Finished')


    

    




Finished