<filename>controller/c_stock.py
import datetime
import tushare as ts
from app_registry import appRegistry as ar
from model.m_area import MArea
from model.m_industry import MIndustry
from model.m_stock import MStock
from model.m_user_stock import MUserStock
from model.m_stock_daily import MStockDaily
from util.app_util import AppUtil

'''
获取沪深两市所有挂牌股票基本信息，并保存到数据库中
'''
class CStock(object):
    def __init__(self):
        self.name = 'CStock'
        
    @staticmethod
    def get_stocks():
        ''' 获取市场上所有挂牌股票基本信息 '''
        ts.set_token(ar.ts_token)
        pro = ts.pro_api()
        data = pro.stock_basic(exchange='', list_status='L', 
                    fields='ts_code,symbol,name,area,industry,list_date')
        rec_nums = data.shape[0]
        for j in range(rec_nums):
            rec = list(data.ix[rec_nums - 1 - j])
            flds = []
            area_id = CStock.process_area(rec[3])
            industry_id = CStock.process_industry(rec[4])
            stock_id = CStock.process_stock(rec[0], rec[1], rec[2], 
                        area_id, industry_id, rec[5])
        print('^_^ End caller={0}'.format(ar.caller))
        
    @staticmethod
    def process_area(area_name):
        ''' 
        处理地区信息：如果地区不存在添加到数据库表中，如果存在
        则求出其area_id
        '''
        area_id = MArea.get_area_id_by_name(area_name)
        if area_id>0:
            print('地区存在：{0}'.format(area_name))
        else:
            area_id = MArea.add_area(area_name)
            print('添加地区：{0}---{1}'.format(area_id, area_name))
        return area_id
        
    @staticmethod
    def process_industry(industry_name):
        ''' 
        处理行业信息：如果行业不存在则添加表数据库表中，如果
        存在则返回industry_id
        '''
        industry_id = MIndustry.get_industry_id_by_name(industry_name)
        if industry_id>0:
            print('行业存在：{0}'.format(industry_name))
        else:
            industry_id = MIndustry.add_industry(industry_name)
            print('添加行业：{0}---{1}'.format(industry_id, industry_name))
        return industry_id
        
    @staticmethod
    def process_stock(ts_code, symbol, stock_name, area_id, 
                industry_id, list_date):
        '''
        处理股票基本信息：如果股票不存在则添加到数据库中，如果存在
        则返回股票编号
        '''
        stock_id = MStock.get_stock_id_by_name(stock_name)
        if stock_id > 0:
            print('股票存在：{0}'.format(stock_name))
        else:
            stock_id = MStock.add_stock(ts_code, symbol, 
                        stock_name, area_id, industry_id, list_date)
            print('添加股票：{0}---{1}'.format(stock_id, stock_name))
        return stock_id

    @staticmethod
    def get_user_stock_hold(user_stock_id):
        '''
        获取指定用户当前指定股票的持有量
        @param user_stock_id：用户股票组合编号，可以通过user_id和stock_id求出
        @return 股票持有量
        @version v0.0.1 闫涛 2019-03-04
        '''
        rc, rows = MUserStock.get_user_stock_hold(user_stock_id)
        if rc <= 0:
            return 0
        else:
            return rows[0][0]

    @staticmethod
    def get_prev_day_close_price(ts_code, curr_date):
        '''
        获取前一天股票的收盘价，如果前一天是非交易日，则循环向前取，直到
        产易日为止
        @param curr_date：当前日期，格式为20190304
        @return 前一交易日的收盘价（以分为单位）
        @version v0.0.1 闫涛 2019-03-05
        '''
        prev_date = AppUtil.get_delta_date(curr_date, delta=-1)
        curr_date = prev_date
        rc, rows = MStockDaily.get_close(ts_code, prev_date)
        while rc <= 0:
            prev_date = AppUtil.get_delta_date(curr_date, delta=-1)
            curr_date = prev_date
            rc, rows = MStockDaily.get_close(ts_code, prev_date)
            print('处理日期：{0}; rc={1}'.format(prev_date, rc))
        return rows[0][0]

    @staticmethod
    def get_stock_vo_of_user(user_stock_id):
        '''
        通过用户持股代码查询股票基本信息
        @param user_stock_id：用户持股编号
        @return 股票基本信息
        @version v0.0.1 闫涛 2019-03-05
        '''
        rc, rows = MUserStock.get_stock_vo(user_stock_id)
        if rc <= 0:
            return []
        else:
            return rows[0]

    @staticmethod
    def get_stock_vo_by_id(stock_id):
        '''
        通过股票编号查询股票基本信息
        @param stock_id：股票编号
        @return 若存在返回：股票编码、股票代码、股票名称
        @version v0.0.1 闫涛 2019-03-06
        '''
        rc, rows = MStock.get_stock_vo_by_id(stock_id)
        if rc <= 0:
            return []
        else:
            return rows[0]

    @staticmethod
    def get_stock_id_by_ts_code(ts_code):
        '''
        根据ts_code求出股票编号stock_id
        @param ts_code：股票编码
        @return stock_id
        @version v0.0.1 闫涛 2019-03-06
        '''
        rc, rows = MStock.get_stock_id_by_ts_code(ts_code)
        if rc <= 0:
            return 0
        else:
            return rows[0][0]
        
        
        
        
        
        
        
        
        
        
        
        
        
        
        
        