在Python中解析CS：GO脚本文件

Question

我正在使用CS：GO的一些脚本文件，我必须从该文件中获取一些有用的信息，并将这些数据导入到我的python应用程序中。

这是txt数据格式的示例：

https://steamcdn-a.akamaihd.net/apps/730/scripts/items/items_game.83a9ad4690388868ab33c627af730c43d4b0f0d9.txt

这些值是随机格式（Color \ Pos \ String），但是我只需要一个包含所有值的字符串。我需要将这些信息放入字典中，例如：

print(global_dict['items_game']['game_info']['first_valid_class'])
<<2

我现在正在使用解析器，但是遇到很多问题。有没有针对该文件格式的现成解决方案？

Answer 1

这是一个基于pyparsing的解析器，它将解析此格式：

from pyparsing import Suppress, QuotedString, Forward, Group, Dict, ZeroOrMore

LBRACE,RBRACE = map(Suppress, "{}")
qs = QuotedString('"')

# forward-declare value, since this expression will be recursive
# (will contain expressions which use contain value's)
value = Forward()

key_value = Group(qs + value)
struct = LBRACE + Dict(ZeroOrMore(key_value)) + RBRACE

# define content of value using <<= operator
value <<= qs | struct

# define top-level parser
parser = Dict(key_value)

将配置加载到字符串中，然后调用parser.parseString()：

sample = open('cs_go_sample.txt').read()
config = parser.parseString(sample)

print config.keys()
for k in config.items_game.keys():
    print '-', k

config.items_game.pprint()

印刷品：

['items_game']
- sticker_kits
- client_loot_lists
- prefabs
- quest_definitions
- alternate_icons2
- music_definitions
- rarities
- colors
- campaign_definitions
- player_loadout_slots
- quest_schedule
- item_levels
- revolving_loot_lists
- game_info
- pro_players
- recipes
- items_game_live
- paint_kits_rarity
- paint_kits
- qualities
- items
- attributes
- item_sets
- quest_reward_loot_lists
- kill_eater_score_types

[['game_info',
  ['first_valid_class', '2'],
  ['last_valid_class', '3'],
  ['first_valid_item_slot', '0'],
  ['last_valid_item_slot', '54'],
  ['num_item_presets', '4']],
 ['rarities',
  ['default',
   ['value', '0'],
... etc. ...

编辑

如果希望在解析时将整数值转换为int，则可以定义解析动作来执行此操作。但是，您只想将此（我认为）附加到带引号的字符串，而不是键的值。

# use this code to convert integer values to ints at parse time
key_qs = qs.copy()
value_qs = qs.copy()
def convert_integers(tokens):
    if tokens[0].isdigit():
        tokens[0] = int(tokens[0])
value_qs.setParseAction(convert_integers)

value = Forward()
key_value = Group(key_qs + value)
struct = LBRACE + Dict(ZeroOrMore(key_value)) + RBRACE
value <<= value_qs | struct
parser = Dict(key_value)

现在，输出值如下所示：

[['game_info',
  ['first_valid_class', 2],
  ['last_valid_class', 3],
  ['first_valid_item_slot', 0],
  ['last_valid_item_slot', 54],
  ['num_item_presets', 4]],
 ['rarities',
  ['default',
   ['value', 0],
   ['loc_key', 'Rarity_Default'],
   ['loc_key_weapon', 'Rarity_Default_Weapon'],

请注意，整数值不再显示为字符串，而是显示为实际的Python整数。