Python update_fire_groups_data_filesの例

コード例 #1

ファイル: site.py プロジェクト: spascou/ps2.liquidwarp.net

def update_all_data_files(census_service_id: str):

    update_fire_groups_data_files(
        directory=DATA_FILES_DIRECTORY,
        service_id=census_service_id,
    )
    update_infantry_weapons_data_files(
        directory=DATA_FILES_DIRECTORY,
        service_id=census_service_id,
    )
    update_vehicle_weapons_data_files(
        directory=DATA_FILES_DIRECTORY,
        service_id=census_service_id,
    )

コード例 #2

from ps2_analysis.weapons.infantry.data_files import (
    update_data_files as update_infantry_weapons_data_files,
)
from ps2_analysis.weapons.infantry.generate import generate_all_infantry_weapons
from ps2_analysis.weapons.infantry.infantry_weapon import InfantryWeapon

logging.basicConfig(level=logging.INFO)

SERVICE_ID: Optional[str] = os.environ.get("CENSUS_SERVICE_ID")
DATAFILES_DIRECTORY: str = "../datafiles"

if not SERVICE_ID:
    raise ValueError("CENSUS_SERVICE_ID envvar not found")

update_fire_groups_data_files(
    directory=DATAFILES_DIRECTORY, service_id=SERVICE_ID,
)

update_infantry_weapons_data_files(
    directory=DATAFILES_DIRECTORY, service_id=SERVICE_ID,
)

infantry_weapons: List[InfantryWeapon] = list(
    generate_all_infantry_weapons(data_files_directory=DATAFILES_DIRECTORY)
)

print(f"Generated {len(infantry_weapons)} infantry weapons")

wp: InfantryWeapon = next(x for x in infantry_weapons if x.item_id == 43)

fm: FireMode = wp.fire_groups[0].fire_modes[1]