test.py

import csv
from pathlib import Path
import glob
import re
import xml.etree.ElementTree as ET
import xlsxwriter

# set up xlsx file

workbook = xlsxwriter.Workbook('mapvalreport.xlsx')
worksheet = workbook.add_worksheet()
tableheading = workbook.add_format({'bold': True, 'rotation': 45})
tablegrade = workbook.add_format({'align': 'center'})
poifail = workbook.add_format({'font_color': 'red'})
mapfail = workbook.add_format({'bold': True, 'font_color': 'red'})
mappass = workbook.add_format({'bold': True, 'font_color': 'green'})

# define class keypoi to store requirements

class keypoi(object):
    def __init__(self, filename, friendlyname, required, detected):
        self.filename = filename
        self.friendlyname = friendlyname
        self.required = required
        self.detected = detected

    def found(self):
        self.detected += 1

    def detection(self):
        print(f"{self.friendlyname}: {self.detected} found.")

# read in required.csv file - populate keypoi objects
reqlist = []
  
with open('map-validation\\requirements.csv', 'r') as f:
    reader = csv.reader(f)
    for row in reader:
        reqlist.append(keypoi(row[0], row[1], row[2], 0))

# add traders to requirements

reqlist.append(keypoi("tradernw", "NW Traders", 2, 0))
reqlist.append(keypoi("traderne", "NE Traders", 2, 0))
reqlist.append(keypoi("tradersw", "SW Traders", 2, 0))
reqlist.append(keypoi("traderse", "SE Traders", 2, 0))

# add column headers in first line

row = 0
column = 0
worksheet.set_row(row, None, tableheading)
worksheet.write(row, column, 'World')
column += 1

for req in reqlist:
    worksheet.write(row, column, req.friendlyname)
    worksheet.set_column(column, column, 6)
    column += 1

worksheet.write(row, column, 'Score')
worksheet.set_column(column, column, 6)
column += 1
worksheet.write(row,column, 'Grade')
worksheet.set_column(column, column, 6, tablegrade)

# read in prefabs.xml files - recurse through world directories in a root path
glob_path = Path(r"J:\Games\7D2D\TestWorlds")
file_list = [str(pp) for pp in glob_path.glob("**/prefabs.xml")]

# define blank list to keep track of which worlds work

passes = []

# set longest world name to 0

maxworld = 0
# start checking xml files

for file in file_list:

    # start new excel line

    row += 1
    column = 0

    # get world name from path name
    world = re.search("([A-Za-z0-9\-\s]*)(?=.prefabs\.xml$)", file).group()
    print(world)
    print("")

    worksheet.write(row, column, world)

    # increase longest world name if this world is longer than the previous longest

    if len(world) > maxworld:
        maxworld = len(world)
        
    # reset detected counters
    for req in reqlist:
        req.detected = 0

    # read in xml
    prefabs = ET.parse(file)
    root = prefabs.getroot()

    # count key pois
    for decoration in root.iter('decoration'):
        name = decoration.get('name')
        if re.match("trader.*", name):
            position = decoration.get('position')
            xpos = int(re.search("^(-?\d{1,4})", position).group())
            zpos = int(re.search("(-?\d{1,4}$)", position).group())
            if zpos >= 0:
                if xpos < 0:
                    keypoi.found(reqlist[-4]) # found in NW quadrant
                else:
                    keypoi.found(reqlist[-3]) # found in NE quadrant
            else:
                if xpos < 0:
                    keypoi.found(reqlist[-2]) # found in SW quadrant
                else:
                    keypoi.found(reqlist[-1]) # found in SE quadrant
        else:
            for req in reqlist[:-4]:
                if re.match(req.filename, name):
                    keypoi.found(req)
                    break

    # reset score
    score = 0

    # calculate score
    for req in reqlist:
        
        column += 1
        
        if int(req.detected) >= int(req.required):
            print(f"{req.friendlyname}: {req.detected} of {req.required} found - Pass")
            score += 1

            worksheet.write(row,column,req.detected)
        else:
            print(f"{req.friendlyname}: {req.detected} of {req.required} found - Fail")
            
            worksheet.write(row,column,req.detected,poifail)

    print("")
    column += 1

    # print score and calculate grade

    grades = ["A", "B", "C", "D", "E", "F"]
    if len(reqlist) - score >= 5:
        grade = grades[5]
    else:
        grade = grades[len(reqlist) - score]
        if grade == "A":
            passes.append(world)
    
    print(f"{world} score out of {len(reqlist)}: {score} - Grade {grade}")
    worksheet.write(row,column,score)
    column += 1
    worksheet.write(row,column,grade)


# print summary
if len(passes) == 0:
    print("No passes in batch.")
else:
    print("List of passes:")
    for apass in passes:
        print(apass)

# set world name column width to (longest world name + 2 characters) for spacing
maxworld += 2
worksheet.set_column(0,0,maxworld)
workbook.close()