231 lines
8.5 KiB
Python
231 lines
8.5 KiB
Python
|
from typing import Optional, Dict, List
|
||
|
from os import walk, path
|
||
|
import urllib
|
||
|
import csv
|
||
|
|
||
|
from read import BaseReader
|
||
|
from core.config import CoreConfig
|
||
|
from titles.sao.database import SaoData
|
||
|
from titles.sao.const import SaoConstants
|
||
|
|
||
|
|
||
|
class SaoReader(BaseReader):
|
||
|
def __init__(
|
||
|
self,
|
||
|
config: CoreConfig,
|
||
|
version: int,
|
||
|
bin_arg: Optional[str],
|
||
|
opt_arg: Optional[str],
|
||
|
extra: Optional[str],
|
||
|
) -> None:
|
||
|
super().__init__(config, version, bin_arg, opt_arg, extra)
|
||
|
self.data = SaoData(config)
|
||
|
|
||
|
try:
|
||
|
self.logger.info(
|
||
|
f"Start importer for {SaoConstants.game_ver_to_string(version)}"
|
||
|
)
|
||
|
except IndexError:
|
||
|
self.logger.error(f"Invalid project SAO version {version}")
|
||
|
exit(1)
|
||
|
|
||
|
def read(self) -> None:
|
||
|
pull_bin_ram = True
|
||
|
|
||
|
if not path.exists(f"{self.bin_dir}"):
|
||
|
self.logger.warn(f"Couldn't find csv file in {self.bin_dir}, skipping")
|
||
|
pull_bin_ram = False
|
||
|
|
||
|
if pull_bin_ram:
|
||
|
self.read_csv(f"{self.bin_dir}")
|
||
|
|
||
|
def read_csv(self, bin_dir: str) -> None:
|
||
|
self.logger.info(f"Read csv from {bin_dir}")
|
||
|
|
||
|
self.logger.info("Now reading QuestScene.csv")
|
||
|
try:
|
||
|
fullPath = bin_dir + "/QuestScene.csv"
|
||
|
with open(fullPath, encoding="UTF-8") as fp:
|
||
|
reader = csv.DictReader(fp)
|
||
|
for row in reader:
|
||
|
questSceneId = row["QuestSceneId"]
|
||
|
sortNo = row["SortNo"]
|
||
|
name = row["Name"]
|
||
|
enabled = True
|
||
|
|
||
|
self.logger.info(f"Added quest {questSceneId} | Name: {name}")
|
||
|
|
||
|
try:
|
||
|
self.data.static.put_quest(
|
||
|
questSceneId,
|
||
|
0,
|
||
|
sortNo,
|
||
|
name,
|
||
|
enabled
|
||
|
)
|
||
|
except Exception as err:
|
||
|
print(err)
|
||
|
except:
|
||
|
self.logger.warn(f"Couldn't read csv file in {self.bin_dir}, skipping")
|
||
|
|
||
|
self.logger.info("Now reading HeroLog.csv")
|
||
|
try:
|
||
|
fullPath = bin_dir + "/HeroLog.csv"
|
||
|
with open(fullPath, encoding="UTF-8") as fp:
|
||
|
reader = csv.DictReader(fp)
|
||
|
for row in reader:
|
||
|
heroLogId = row["HeroLogId"]
|
||
|
name = row["Name"]
|
||
|
nickname = row["Nickname"]
|
||
|
rarity = row["Rarity"]
|
||
|
skillTableSubId = row["SkillTableSubId"]
|
||
|
awakeningExp = row["AwakeningExp"]
|
||
|
flavorText = row["FlavorText"]
|
||
|
enabled = True
|
||
|
|
||
|
self.logger.info(f"Added hero {heroLogId} | Name: {name}")
|
||
|
|
||
|
try:
|
||
|
self.data.static.put_hero(
|
||
|
0,
|
||
|
heroLogId,
|
||
|
name,
|
||
|
nickname,
|
||
|
rarity,
|
||
|
skillTableSubId,
|
||
|
awakeningExp,
|
||
|
flavorText,
|
||
|
enabled
|
||
|
)
|
||
|
except Exception as err:
|
||
|
print(err)
|
||
|
except:
|
||
|
self.logger.warn(f"Couldn't read csv file in {self.bin_dir}, skipping")
|
||
|
|
||
|
self.logger.info("Now reading Equipment.csv")
|
||
|
try:
|
||
|
fullPath = bin_dir + "/Equipment.csv"
|
||
|
with open(fullPath, encoding="UTF-8") as fp:
|
||
|
reader = csv.DictReader(fp)
|
||
|
for row in reader:
|
||
|
equipmentId = row["EquipmentId"]
|
||
|
equipmentType = row["EquipmentType"]
|
||
|
weaponTypeId = row["WeaponTypeId"]
|
||
|
name = row["Name"]
|
||
|
rarity = row["Rarity"]
|
||
|
flavorText = row["FlavorText"]
|
||
|
enabled = True
|
||
|
|
||
|
self.logger.info(f"Added equipment {equipmentId} | Name: {name}")
|
||
|
|
||
|
try:
|
||
|
self.data.static.put_equipment(
|
||
|
0,
|
||
|
equipmentId,
|
||
|
name,
|
||
|
equipmentType,
|
||
|
weaponTypeId,
|
||
|
rarity,
|
||
|
flavorText,
|
||
|
enabled
|
||
|
)
|
||
|
except Exception as err:
|
||
|
print(err)
|
||
|
except:
|
||
|
self.logger.warn(f"Couldn't read csv file in {self.bin_dir}, skipping")
|
||
|
|
||
|
self.logger.info("Now reading Item.csv")
|
||
|
try:
|
||
|
fullPath = bin_dir + "/Item.csv"
|
||
|
with open(fullPath, encoding="UTF-8") as fp:
|
||
|
reader = csv.DictReader(fp)
|
||
|
for row in reader:
|
||
|
itemId = row["ItemId"]
|
||
|
itemTypeId = row["ItemTypeId"]
|
||
|
name = row["Name"]
|
||
|
rarity = row["Rarity"]
|
||
|
flavorText = row["FlavorText"]
|
||
|
enabled = True
|
||
|
|
||
|
self.logger.info(f"Added item {itemId} | Name: {name}")
|
||
|
|
||
|
try:
|
||
|
self.data.static.put_item(
|
||
|
0,
|
||
|
itemId,
|
||
|
name,
|
||
|
itemTypeId,
|
||
|
rarity,
|
||
|
flavorText,
|
||
|
enabled
|
||
|
)
|
||
|
except Exception as err:
|
||
|
print(err)
|
||
|
except:
|
||
|
self.logger.warn(f"Couldn't read csv file in {self.bin_dir}, skipping")
|
||
|
|
||
|
self.logger.info("Now reading SupportLog.csv")
|
||
|
try:
|
||
|
fullPath = bin_dir + "/SupportLog.csv"
|
||
|
with open(fullPath, encoding="UTF-8") as fp:
|
||
|
reader = csv.DictReader(fp)
|
||
|
for row in reader:
|
||
|
supportLogId = row["SupportLogId"]
|
||
|
charaId = row["CharaId"]
|
||
|
name = row["Name"]
|
||
|
rarity = row["Rarity"]
|
||
|
salePrice = row["SalePrice"]
|
||
|
skillName = row["SkillName"]
|
||
|
enabled = True
|
||
|
|
||
|
self.logger.info(f"Added support log {supportLogId} | Name: {name}")
|
||
|
|
||
|
try:
|
||
|
self.data.static.put_support_log(
|
||
|
0,
|
||
|
supportLogId,
|
||
|
charaId,
|
||
|
name,
|
||
|
rarity,
|
||
|
salePrice,
|
||
|
skillName,
|
||
|
enabled
|
||
|
)
|
||
|
except Exception as err:
|
||
|
print(err)
|
||
|
except:
|
||
|
self.logger.warn(f"Couldn't read csv file in {self.bin_dir}, skipping")
|
||
|
|
||
|
self.logger.info("Now reading Title.csv")
|
||
|
try:
|
||
|
fullPath = bin_dir + "/Title.csv"
|
||
|
with open(fullPath, encoding="UTF-8") as fp:
|
||
|
reader = csv.DictReader(fp)
|
||
|
for row in reader:
|
||
|
titleId = row["TitleId"]
|
||
|
displayName = row["DisplayName"]
|
||
|
requirement = row["Requirement"]
|
||
|
rank = row["Rank"]
|
||
|
imageFilePath = row["ImageFilePath"]
|
||
|
enabled = True
|
||
|
|
||
|
self.logger.info(f"Added title {titleId} | Name: {displayName}")
|
||
|
|
||
|
if len(titleId) > 5:
|
||
|
try:
|
||
|
self.data.static.put_title(
|
||
|
0,
|
||
|
titleId,
|
||
|
displayName,
|
||
|
requirement,
|
||
|
rank,
|
||
|
imageFilePath,
|
||
|
enabled
|
||
|
)
|
||
|
except Exception as err:
|
||
|
print(err)
|
||
|
elif len(titleId) < 6: # current server code cannot have multiple lengths for the id
|
||
|
continue
|
||
|
except:
|
||
|
self.logger.warn(f"Couldn't read csv file in {self.bin_dir}, skipping")
|