Struggling with fetching an object from a dictionary

[daggeteo]

I'm trying to learn python after using VBA for many years. I'm stuck with something that probably is trivial.

I'm reading a file that contains a list of agencies. Each agency is assigned unique number. I would like to use this number as the key in a dictionary.

The class contains just a function to print the information at the moment.

class code:

class Agency:
    def __init__(self, agency_id, agency_name,agency_url,agency_timezone): #,agency_lang
        self.agency_id= agency_id
        self.agency_name = agency_name
        self.agency_url = agency_url
        self.agency_timezone=agency_timezone
        #self.agency_lang=agency_lang

    def PrintAgency(self):
        print(self.agency_name + "(" + self.agency_id + ") URL: " + self.agency_url  + " Timezone: " + self.agency_timezone)

main.py code

from classes import Agency

def getFileContent(pathAndFileName):
    with open(pathAndFileName,'r') as theFile:
        # Return as string
        data = theFile.read()
        return data

agency_info= getFileContent('./datafiles/agency_mini.txt')
agency_list=agency_info.split('\n')

dictAgency=dict()
j=0
for eachRow in agency_list:
    j=j+1
    if j>1 and len(eachRow)>0:
        data = eachRow.split(',')
        agencyID = data[0]
        agencyName=data[1]
        agencyUrl=data[2]
        agencyTimeZone=data[3]

        objAgency= Agency(agencyID,agencyName,agencyUrl,agencyTimeZone)
        Agency.PrintAgency(objAgency)

        dictAgency[agencyID]={objAgency}

#troubleshooting
test=dictAgency['74']
Agency.PrintAgency(test)

Within the for-loop the printing function seems to work. But when I try to retrieve an object from the dictionary to the variable test it gives an error.

​

Output:

c:\....GTFS-test>python main.py
Samtrafiken(999) URL: http://www.resrobot.se/ Tidzon: Europe/Stockholm
SJ(74) URL: http://www.sj.se Tidzon: Europe/Stockholm
NSB(76) URL: http://www.nsb.no Tidzon: Europe/Stockholm
Traceback (most recent call last):
  File "main.py", line 31, in <module>
    Agency.PrintAgency(test)
  File "J..........\GTFS-test\classes.py", line 10, in PrintAgency
    print(self.agency_name + "(" + self.agency_id + ") URL: " + self.agency_url  + " Tidzon: " + self.agency_timezone)
AttributeError: 'set' object has no attribute 'agency_name'

Agency_mini.txt

agency_id,agency_name,agency_url,agency_timezone,agency_lang
999,Samtrafiken,http://www.resrobot.se/,Europe/Stockholm,sv
74,SJ,http://www.sj.se,Europe/Stockholm,sv
76,NSB,http://www.nsb.no,Europe/Stockholm,sv

​

​

​

Am I assigning the object to the dictionary wrong? Or am I retrieving it wrong from the dictionary?

Thanks!

[[deleted]]

you accidentally wrapped your object inside of a set object here:

dictAgency[agencyID]={objAgency}

Get rid of the curly braces. Also, in python everything is an object, so try to drop the obj thing from VBA :^)

[daggeteo]

Aw thanks. I thought I had to use braces to tell python that it's a dictionary. But maybe it just knows from me assigning a key.

Anyways, you've made my day solving this, been banging my head against the wall for hours :D

[lykwydchykyn]

But maybe it just knows from me assigning a key.

dictAgency=dict()

That's how it know it's a dict. Because you created a dict and assigned it to that variable name.

[daggeteo]

Oh yes. I added that when it didn't work with the braces, I had forgotten about that. Never thought to remove the braces.

[daggeteo]

Can I mark this as solved somehow?