| | |
|---|
| | | for block in datas: |
|---|
| | | codes = [] |
|---|
| | | for code_data in block[2]: |
|---|
| | | code = code_data[0] |
|---|
| | | code = code_data[0].split(".")[0] |
|---|
| | | if code not in code_block_dict: |
|---|
| | | code_block_dict[code] = set() |
|---|
| | | code_block_dict[code].add(block[0]) |
|---|
| | | codes.append(code) |
|---|
| | | block_codes_dict[block] = codes |
|---|
| | | block_codes_dict[block[0]] = codes |
|---|
| | | __save_block_codes(block_codes_dict) |
|---|
| | | for key in code_block_dict: |
|---|
| | | __save_code_block(key, code_block_dict[key]) |
|---|
| | |
|---|
| | | |
|---|
| | | # 保存代码所属板块 |
|---|
| | | def __save_code_block(code, blocks): |
|---|
| | | __get_redis().setex(f"code_blocks-{code}", tool.get_expire(), json.dumps(blocks)) |
|---|
| | | __get_redis().setex(f"code_blocks-{code}", tool.get_expire(), json.dumps(list(blocks))) |
|---|
| | | |
|---|
| | | |
|---|
| | | # 保存板块下的代码 |
|---|
| | |
|---|
| | | if block_codes: |
|---|
| | | block_codes.get(block) |
|---|
| | | return None |
|---|
| | | |
|---|
| | | if __name__ == "__main__": |
|---|
| | | print(get_code_blocks("600468")) |
|---|
