Handle user unsubscription

1 files changed, 7 insertions, 2 deletions

diff --git a/user.py b/user.py
index 1cf0ec8..bb4e262 100644
--- a/user.py
+++ b/user.py
@@ -1,16 +1,21 @@
 import os
 import os.path
+import shutil
 
 class UserHandler :
     def __init__(self,root):
         self.root = root
 
     def register(self,name):
-        if not os.path.exists(name) :
-            os.mkdir(self.root+'/'+name)
+        userDir = self.root+'/'+name
+        if not os.path.exists(userDir) :
+            os.mkdir(userDir)
 
     def registered(self,name):
         return os.path.exists(self.root+'/'+name)
     
+    def unregister(self,name):
+        shutil.rmtree(self.root+'/'+name)
+        
     def getUserList(self):
         return os.listdir(self.root)
\ No newline at end of file