class UserDB:
def __init__(self):
self.list = Hash(USER_HASH_SIZE)
self.followRank = []
self.tweetRank = []
self.totalUser = 0
def addUser(self, id, n):
if self.list.add(id, n): # If successfully added
self.totalUser += 1
return 1
else:
return None
def deleteUser(self, id):
if self.list.delete(id): # If successfully deleted
self.totalUser -= 1
else:
return None
def getUser(self, id):
u = self.list.search(id)
if u:
return u
else:
return None
def plusFollow(self, id):
u = self.list.search(id)
if u:
u.followCount += 1
def plusTweet(self, id):
u = self.list.search(id)
if u:
u.tweetCount += 1
# In construction : need to update leaderboard
def updateFollowRank(self):
self.followRank.clear()
for x in range(len(self.list.hTable)): # Through Hash List
n = self.list.hTable[x].start
while n: # Through Linked List
if n.v:
self.followRank.append(n.v)
n = n.next
self.followRank = sorted(self.followRank, key=self.getFollow, reverse=True) # Sort by FollowCount
return self.followRank
def updateTweetRank(self):
self.tweetRank.clear()
for x in range(len(self.list.hTable)): # Through Hash List
n = self.list.hTable[x].start
while n: # Through Linked List
if n.v:
self.tweetRank.append(n.v)
n = n.next
self.tweetRank = sorted(self.tweetRank, key=self.getTweet, reverse=True) # Sort by TweetCount
return self.tweetRank
def getFollow(self, user):
return user.followCount
def getTweet(self, user):
return user.tweetCount
