def readRaceNames(fileName, races): with open(fileName, 'r') as namesFile: lineNumber = 1 blocks = namesFile.read().split("\n\n") for block in blocks: lines = block.split("\n") assertNumLines(lines, 3, fileName, lineNumber) assertNotEmpty(lines[0], fileName, lineNumber) assertNotSeparator(lines[0], fileName, lineNumber) race = Race() race.setNameFull(lines[0].strip()) lineNumber+=1 assertNotEmpty(lines[1], fileName, lineNumber) assertNotSeparator(lines[1], fileName, lineNumber) race.setNameShort(lines[1].strip()) lineNumber+=1 assertNotEmpty(lines[2], fileName, lineNumber) assertNotSeparator(lines[2], fileName, lineNumber) race.setNameSpecies(lines[2].strip()) races[race.getNameFull()] = race lineNumber+=2
def readRaceFile(fileName): race = Race() #Read race file with open(fileName, 'r') as raceFile: lineNumber = 1 sections = raceFile.read().split("--------------------------\n") i = 0 for section in sections: section = section.strip() if section != "": if i==0: #Names section nameFull, nameShort, nameSpecies, lineNumber = readRaceNames(section, fileName, lineNumber) elif i==1: #Race Sheet section startingUnits, startingTechs, specialAbilities, history, lineNumber = readRaceSheet(section, fileName, lineNumber) elif i==2: #Trade Contracts section tradeContracts, lineNumber = readTradeContracts(section, fileName, lineNumber) elif i==3: #Home System section homeSystem, lineNumber = readHomeSystem(section, fileName, lineNumber) elif i==4: #Home Planet section homePlanet, lineNumber = readHomePlanet(section, fileName, lineNumber) homeSystem.addPlanetDetails(homePlanet.getName(), homePlanet.getFlavorText(), homePlanet.getResources(), homePlanet.getInfluence()) elif i==5: #Objective section objective, lineNumber = readObjective(section, fileName, lineNumber, True) elif i==6: #Treaties section treaties, lineNumber = readTreaties(section, fileName, lineNumber, True) i += 1 #Set race attributes race.setNameFull(nameFull) race.setNameShort(nameShort) race.setNameSpecies(nameSpecies) race.setExpansion("Shards of the Throne") for unit in startingUnits.keys(): race.addStartingUnit(unit, startingUnits[unit]) for tech in startingTechs: race.addStartingTech(tech) for ability in specialAbilities: race.addSpecialAbility(ability) for paragraph in history: race.addHistoryParagraph(paragraph) for contract in tradeContracts: race.addTradeContract(contract) race.addHomeSystem(homeSystem) return race, objective, treaties