ugh

1 files changed, 20 insertions, 1 deletions

diff --git a/admin.py b/admin.py
index 4dbe1ca..5319a58 100644
--- a/admin.py
+++ b/admin.py
@@ -1,5 +1,6 @@
 
 import sys, os, urllib, StringIO, traceback, cgi, binascii, getopt
+sys.path.append('/usr/local/pypi/lib')
 
 import store, config
 
@@ -12,12 +13,30 @@ def set_password(store, name, pw):
     store.store_user(user['name'], pw.strip(), user['email'])
     print 'done'
 
+
+def remove_package(store, name):
+    ''' Remove a package from the database
+    '''
+    cursor = store.cursor
+    cursor.execute('delete from packages where name=%s', name)
+    cursor.execute('delete from releases where name=%s', name)
+    cursor.execute('delete from journals where name=%s', name)
+    cursor.execute('delete from roles where package_name=%s', name)
+    print 'done'
+
 if __name__ == '__main__':
     config = config.Config('/usr/local/pypi/config.ini', 'webui')
     store = store.Store(config)
     store.open()
+    command = sys.argv[1]
+    args = (store, ) + tuple(sys.argv[2:])
     try:
-        set_password(store, sys.argv[1], sys.argv[2])
+        if command == 'password':
+            set_password(*args)
+        elif command == 'rmpackage':
+            remove_package(*args)
+        else:
+            print "unknown command '%s'!"%command
         store.commit()
     finally:
         store.close()