def save_rong_for_days_before(stock, day_before):
    for i in range(1, day_before):
        day = datetime.now()+timedelta(days=-i)
        date = day.strftime('%Y-%m-%d')
        result = get_rong_sz(stock[:6], date)
        if result:
            to_db(stock, date, result)
Example #2
0
__author__ = 'yu'

from pymongo import MongoClient
from rong import get_rong_sz

c = MongoClient('121.199.5.143').stock

stocks = c.history.distinct('stock')
for stock in stocks:
    if stock.find('002') == 0:
        result = get_rong_sz(stock[:6], '2015-09-28')
        if result:
            rongzi1 = int(result[2].replace(',', ''))
            rongzi2 = 0

            r0914 = get_rong_sz(stock[:6], '2015-09-25')
            if r0914:
                rongzi2 = int(r0914[2].replace(',', ''))

            if rongzi1 > 2*rongzi2:
                print '{} {} {}'.format(stock, rongzi2, rongzi1)
        else:
            continue
from datetime import datetime, timedelta
from rong import get_rong_sz
from util import get_all_stock, c


if __name__ == '__main__':
    for stock in get_all_stock():
        if stock.startswith('000') or stock.startswith('002'):
            day = datetime.now()+timedelta(days=-1)
            if day.weekday() == 6:
                day = datetime.now()+timedelta(days=-3)
            elif day.weekday() == 5:
                day = datetime.now()+timedelta(days=-2)

            date = day.strftime('%Y-%m-%d')
            result = get_rong_sz(stock[:6], date)
            if result:
                code, name, rongzi_buy, rongzi_balance, rongquan_sell_liang, rongquan_balance_liang, rongquan_balance, rong_all_balance = result
                print code

                c.history.update({
                    'date': date,
                    'stock': stock
                }, {
                    '$set': {
                        'rongzi_buy': rongzi_buy,
                        'rongzi_balance': rongzi_balance,
                        'rongquan_sell_liang': rongquan_sell_liang,
                        'rongquan_balance_liang': rongquan_balance_liang,
                        'rongquan_balance': rongquan_balance,
                        'rong_all_balance': rong_all_balance