database.py

from random import shuffle

# TODO change wood to lumber
#indices
GRAIN_INDEX = 0
FLOUR_INDEX = 1
BEER_INDEX  = 2
BREAD_INDEX = 3
MEAT_INDEX  = 4
FRUIT_INDEX = 5
LUMBER_INDEX  = 6
CHAIR_INDEX = 7
TABLE_INDEX = 8 

#profile indices
MANU_INDEX = 0
STORE_INDEX = 1
HOME_INDEX = 2
CHURCH_INDEX = 3

#required stock- empty list if none
#each tuple contains indexes of required inputs and the ratio of input to output. eg 1 flour needs only 2 grain.
#harvest inputs are currently placeholders
grain   = []
# grain   = [(GRAIN_INDEX, .04)]
flour   = [(GRAIN_INDEX, 2)]
beer    = [(GRAIN_INDEX, 0)]
bread   = [(FLOUR_INDEX, 4)]
meat    = []
fruit   = [(FRUIT_INDEX, .04)]
lumber    = [(LUMBER_INDEX, .04)]
chair   = [(LUMBER_INDEX, 35)]
table   = [(LUMBER_INDEX, 100)]

components = [grain, flour, beer, bread, meat, fruit, lumber, chair, table]

#growth ratios
#plant 1 grain, harvest 20 grain

growratios = []

#production types
planted = ["grain", "beer", "meat", "fruit", "lumber"]
crafted = ["flour", "bread", "chair", "table"]

#planting seasons- only grain for now (which is winter wheat)
seasons = ["September", "all", "all", "all", "all", "all", "all", "all", "all"]

#name lists
maleNameList = open("boynames", "r").read().splitlines()
femaleNameList = open("girlnames", "r").read().splitlines()
lastNameList = open("lastnames", "r").read().splitlines()
busiNameList = open("businessnames", "r").read().splitlines()

#these are identified by these same indices throughout the program.
equipmentList = ["millstone"]
skillsList = [0,0,0,0,0,0,0,0,0]
materialsList = ["grain", "flour", "beer", "bread", "meat", "fruit", "lumber", "chair", "table"]
unitMissions = ["manu", "store", "house", "church"]

#game objects
#HOW MANY OF THESE ARE ACTUALLY USED?
localityList = []
businessList = []
religionList = []
unitList = []
churchList = []
peopleList = []
bossList = []

#utility
#people will buy up to the number below INCLUSIVE.
utilityLimitList = [0,2,6,4,2,2,0,2,1]
utilityScaleList = [0,2,6,100,20,15,0,2,4]
# happinessMax = sum( utilityLimitList[i] * utilityScaleList[i] for i in range(len(utilityLimitList)))

def addBusiness(business):
    businessList.append(business)

def addLocality(locality):
    localityList.append(locality)

def addPeople(person):
    peopleList.append(person)

def addBoss(boss):
    bossList.append(boss)

def removeBoss(boss):
    bossList.remove(boss)

def addUnit(unit):
    unitList.append(unit)

def addReligion(religion):
    religionList.append(religion)

def addChurch(church):
    addUnit(church)
    churchList.append(church)

def getReligions():
    return religionList

def getBosses():
    return bossList

def getChurches():
    return churchList

def getBusinesses():
    return businessList

def getEquipment():
    return equipmentList

def getFirstNameList(gender):
    if gender == 0:
        nameList = maleNameList
    else:
        nameList = femaleNameList
    return nameList

def getLastNameList():
    return lastNameList

def getLocality():
    return localityList

def getMaxHappiness():
    return happinessMax

def getMaterials():
    return materialsList

def getComponents(materialIndex):
    return components[materialIndex]

def getAllComponents():
    return components

def getSkills():
    return skillsList

def getPeople():
    return peopleList

def shufflePeople():
    shuffle(peopleList)

def getUnit():
    return unitList

def getUnitMissions():
    return unitMissions

def getUtilityLimit():
    return utilityLimitList

def getUtilityScale():
    return utilityScaleList

def is_planted(materialIndex):
    is_planted = False
    material = materialsList[materialIndex]
    
    if material in planted:
        is_planted = True

    return is_planted

def is_crafted(materialIndex):
    is_crafted = False
    material = materialsList[materialIndex]

    if material in crafted:
        is_crafted = True

    return is_crafted

def isInSeason(materialIndex, season):
    if (seasons[materialIndex] == "all") or (seasons[materialIndex] == season):
        return True
    else:
        return False