account fixes

2 files changed, 9 insertions, 3 deletions

diff --git a/module/plugins/accounts/RapidshareCom.py b/module/plugins/accounts/RapidshareCom.py
index 233cd9801..052470ada 100644
--- a/module/plugins/accounts/RapidshareCom.py
+++ b/module/plugins/accounts/RapidshareCom.py
@@ -47,7 +47,10 @@ class RapidshareCom(Account):
                 continue
             k, v = t.split("=")
             info[k] = v
-        out = {"validuntil":None, "login":str(info["accountid"]), "trafficleft":int(info["tskb"]), "type":self.__name__}
+            
+        out = Account.getAccountInfo(self, user)
+        tmp = {"validuntil":None, "login":str(info["accountid"]), "trafficleft":int(info["tskb"]), "type":self.__name__}
+        out.update(tmp)
         
         return out
     
diff --git a/module/plugins/accounts/UploadedTo.py b/module/plugins/accounts/UploadedTo.py
index 242e6cd5e..d2aa22b22 100644
--- a/module/plugins/accounts/UploadedTo.py
+++ b/module/plugins/accounts/UploadedTo.py
@@ -42,8 +42,11 @@ class UploadedTo(Account):
         raw_valid = re.search(r"Valid until: </span> <span class=.*?>(.*?)</span>", html).group(1)
         traffic = int(self.parseTraffic(raw_traffic))
         validuntil = int(mktime(strptime(raw_valid.strip(), "%d-%m-%Y %H:%M")))
-        return {"login":user, "validuntil":validuntil, "trafficleft":traffic, "type":self.__name__}
-    
+        out = Account.getAccountInfo(self, user)
+        tmp =  {"login":user, "validuntil":validuntil, "trafficleft":traffic, "type":self.__name__}
+        out.update(tmp)
+        return out
+        
     def login(self, user, data):
         req = self.core.requestFactory.getRequest(self.__name__, user)
         req.load("http://uploaded.to/login", None, { "email" : user, "password" : data["password"]}, cookies=True)