HaiTong-node2.py

import datetime
import os

import easytrader
import json
import sys
from Strategy.gridTrade import grid_bs, time, code2name



if sys.platform == 'linux':
    user = ''
elif sys.platform == 'darwin':
    user = ''
else:
    自己找吧

class User():
    def __init__(self, user):
        self.user = user
        print(user.balance)
        self.zi_jin_yu_e = user.balance['资金余额']
        self.ke_yong_jin_e = user.balance['可用金额']
        self.ke_qu_jin_e = user.balance['可取金额']
        self.zong_zi_chan = user.balance['总资产']
        self.stock = Stock(user.position)

    def buy(self, code, price, amount):
        self.user.buy(code, price, amount)

    def sell(self, code, price, amount):
        self.user.sell(code, price, amount)

    def get_balance(self):
        return self.ke_yong_jin_e

    def user_refresh(self):
        self.user.refresh()

    def get_today_trades(self):
        return self.user.today_trades

    def get_today_entrusts(self):
        return self.user.today_entrusts

    def show(self):
        print("资金余额：%s\n-->\t可用资金：%s\n-->\t可取金额：%s\n-->\t总资产：%s\n-->\t"%(self.zi_jin_yu_e, self.ke_yong_jin_e, self.ke_qu_jin_e, self.zong_zi_chan))
        print("当前持仓股票:\n-->\t", self.stock.get_position())

class Stock():
    def __init__(self, position):
        self.position = position

    def get_position(self):
        return self.position



def get_Account():
    return User(user)


if __name__ == '__main__':
    # codes = [i['证券代码'] for i in user.position]
    codes = ['000488', '000519', '002044', '600048', '601311', '002457', '002026']

    print(codes)
    code2name(codes)
    # existFile = os.listdir('cache/')
    # gaps = load_gaps()
    # buy_rates = load_rates('buy')
    # sell_rates = load_rates('sell')
    # for i in user.position:
    #     code = i['证券代码']
    #     if "%s-log.txt"%code in existFile:
    #         print("%s-log.txt exist, replace it please!"%code)
    #         continue
    #     codes.append(code)
    #     price = i['成本价']
    #     amount = int(i['股票余额']//500) + 100
    #     gap = round(price * 0.02, 3)
    #     save_trade_log_once(code, price, amount)
    #     gaps[code] = gap
    #     buy_rates[code] = 1.5
    #     sell_rates[code] = 2
    # save_gaps_once(gaps)
    # save_rates_once(buy_rates,'buy')
    # save_rates_once(sell_rates,'sell')

    while len(codes) > 0:
        s = grid_bs(codes, user)
        codes.remove(s)