In [40]:

    

import numpy as np
import pandas as pd
class PwModel:
    def __init__(self):
        self.alts = []
        self.pw_votes = {}
        self.nAlts = 0
        pass
    
    def addAlt(self, altName):
        if altName in self.alts:
            raise NameError("Alt already existed")
        self.alts.append(altName)
        self.nAlts+=1
        
    def addUser(self, userName):
        if userName in self.pw_votes:
            raise NameError("User name alrady existed")
        self.pw_votes[userName] = np.identity(self.nAlts)


    

In [41]:

    

model = PwModel()
model.addAlt("Kitchen")
model.addAlt("Math")
print(model.alts)


    

    




['Kitchen', 'Math']

In [42]:

    

model.addUser("Bill")
model.addUser("Jeff")


    

In [43]:

    

model.addUser("Jeff")


    

    




---------------------------------------------------------------------------
NameError                                 Traceback (most recent call last)
<ipython-input-43-de252f210060> in <module>()
----> 1 model.addUser("Jeff")

<ipython-input-40-5d10fcea6941> in addUser(self, userName)
     16     def addUser(self, userName):
     17         if userName in self.pw_votes:
---> 18             raise NameError("User name alrady existed")
     19         self.pw_votes[userName] = np.identity(self.nAlts)

NameError: User name alrady existed

In [ ]: