"""
股票数据工具
"""

# 股票代码相关的参数
import redis_manager

_redisManager = redis_manager.RedisManager(0)


# 自由流通股本工具类
class ZYLTGBUtil:
    @classmethod
    def save(cls, code, val, unit):
        redis = _redisManager.getRedis()
        redis.setex("zyltgb-{}".format(code), 60 * 60 * 24 * 10,
                    round(float(val) * 100000000) if int(unit) == 0 else round(
                        float(val) * 10000))

    @classmethod
    def get(cls, code):
        redis = _redisManager.getRedis()
        val = redis.get(code)
        if val is not None:
            return int(val)
        return None