pythonoop

Create a Python User() class that both creates new users and modifies existing users


I'm trying to figure out the best way to create a class that can modify and create new users all in one. This is what I'm thinking:

class User(object):

    def __init__(self,user_id):
      if user_id == -1
          self.new_user = True
      else:
          self.new_user = False

          #fetch all records from db about user_id
          self._populateUser() 

    def commit(self):
        if self.new_user:
            #Do INSERTs
        else:
            #Do UPDATEs

    def delete(self):
        if self.new_user == False:
            return False

        #Delete user code here

    def _populate(self):
        #Query self.user_id from database and
        #set all instance variables, e.g.
        #self.name = row['name']

    def getFullName(self):
        return self.name

#Create a new user
>>u = User()
>>u.name = 'Jason Martinez'
>>u.password = 'linebreak'
>>u.commit()
>>print u.getFullName()
>>Jason Martinez

#Update existing user
>>u = User(43)
>>u.name = 'New Name Here'
>>u.commit()
>>print u.getFullName()
>>New Name Here

Is this a logical and clean way to do this? Is there a better way?

Thanks.


Solution

  • Off the top of my head, I would suggest the following:

    1: Use a default argument None instead of -1 for user_id in the constructor:

    def __init__(self, user_id=None):
        if user_id is None:
             ...
    

    2: Skip the getFullName method - that's just your Java talking. Instead use a normal attribute access - you can convert it into a property later if you need to.