| | |
| | | class XGBHotBlockDataManager: |
| | | total_datas = [] |
| | | __last_datas={} |
| | | latest_datas = None |
| | | |
| | | @classmethod |
| | | def save(cls,day, datas): |
| | | cls.latest_datas = datas |
| | | mysqldb = mysql_data.Mysqldb() |
| | | # 统计代码所属板块 |
| | | code_block_dict = {} |
| | |
| | | code = code_info[0].split(".")[0] |
| | | _id = f"{day}_{data[0]}_{code}" |
| | | |
| | | result = mysqldb.select_one("select count(*) from xgb_hot_block where _id='{}'".format(_id)) |
| | | if not result[0]: |
| | | result = mysqldb.select_one("select * from xgb_hot_block where _id='{}'".format(_id)) |
| | | limit_up_time = code_info[4] |
| | | if len(limit_up_time) <= 6: |
| | | limit_up_time = '' |
| | | if not result: |
| | | mysqldb.execute( |
| | | f"insert into xgb_hot_block(_id,_day,_block_name,_code,_limit_up_time,_price,_rate,_update_time) values('{_id}','{day}','{data[0]}','{code}','{code_info[4]}','{code_info[2]}','{code_info[3]}',now())") |
| | | f"insert into xgb_hot_block(_id,_day,_block_name,_code,_limit_up_time,_price,_rate,_update_time,_first_limit_up_time) values('{_id}','{day}','{data[0]}','{code}','{code_info[4]}','{code_info[2]}','{code_info[3]}',now(),'{limit_up_time}')") |
| | | else: |
| | | # 如果上次的数据和这次一样就不更新,否则需要更新数据 |
| | | if cls.__last_datas.get(_id) != code_info: |
| | | # 更新 |
| | | mysqldb.execute(f"update xgb_hot_block set _limit_up_time='{code_info[4]}',_price='{code_info[2]}',_rate='{code_info[3]}' where _id='{_id}'") |
| | | mysqldb.execute( |
| | | f"update xgb_hot_block set _limit_up_time='{code_info[4]}',_price='{code_info[2]}',_rate='{code_info[3]}',_update_time=now() where _id='{_id}'") |
| | | if (not result[8] or len(result[8]) <= 6) and len(limit_up_time) >= 6: |
| | | mysqldb.execute( |
| | | f"update xgb_hot_block set _first_limit_up_time='{limit_up_time}',_update_time=now() where _id='{_id}'") |
| | | |
| | | cls.__last_datas[_id] = code_info |
| | | # 获取原来的代码所属板块,删除之前错误的板块 |
| | | old_datas = XGBHotBlockDataManager.list_by_code(code, day) |
| | |
| | | return None |
| | | |
| | | limit_up_codes_set = set() |
| | | for block in latest_datas: |
| | | if XGBHotBlockDataManager.latest_datas: |
| | | for block in XGBHotBlockDataManager.latest_datas: |
| | | if block[0] == target_block: |
| | | for code_data in block[2]: |
| | | if len(code_data[4]) > 6: |
| | |
| | | limit_up_count = len(limit_up_codes_set) |
| | | |
| | | total_datas = XGBHotBlockDataManager.total_datas |
| | | codes = set() |
| | | break_codes = set() |
| | | for data in total_datas: |
| | | block = data[2] |
| | | if block != target_block: |
| | | continue |
| | | code = data[3] |
| | | limit_up_time = data[4] |
| | | if len(limit_up_time) > 6: |
| | | codes.add(code) |
| | | codes.discard(code) |
| | | codes.difference(limit_up_codes_set) |
| | | first_limit_up_time = data[8] |
| | | if len(limit_up_time) <= 6 and first_limit_up_time and len(first_limit_up_time) > 6: |
| | | break_codes.add(code) |
| | | # 排除自己 |
| | | break_codes.discard(code) |
| | | # 排除已经涨停的代码 |
| | | break_codes = break_codes.difference(limit_up_codes_set) |
| | | # 炸板个数 |
| | | break_size = len(codes) |
| | | return (limit_up_count,break_size) |
| | | |
| | | |
| | | latest_datas = [] |
| | | break_size = len(break_codes) |
| | | return target_block, limit_up_count, break_size |
| | | |
| | | |
| | | # 保存数据 |
| | | def save_datas(day, datas): |
| | | latest_datas = datas |
| | | XGBHotBlockDataManager.save(day, datas) |
| | | code_block_dict = {} |
| | | block_codes_dict = {} |
| | |
| | | |
| | | |
| | | if __name__ == "__main__": |
| | | XGBHotBlockDataManager.total_datas=XGBHotBlockDataManager.list_all("2023-03-23") |
| | | get_info('002230') |
| | | # XGBHotBlockDataManager.total_datas=XGBHotBlockDataManager.list_all("2023-03-23") |
| | | # get_info('002230') |
| | | codes = set([1, 2, 3, 4]) |
| | | print(codes.difference(set([1, 2]))) |