Move wizard.scripts module to plugins, added hooks accordingly.

[wizard.git] / wizard / command / quota.py

diff --git a/wizard/command/quota.py b/wizard/command/quota.py
index 2a640c91f26db96458a434c15dce81b1418f37d5..11fb04dd57ac207fab78eee0632c39afbf85b98d 100644 (file)
--- a/wizard/command/quota.py
+++ b/wizard/command/quota.py
@@ -2,15 +2,12 @@ import logging
 import os.path
 import sys
 
-from wizard import command, scripts
+from wizard import command, user
 
 def main(argv, baton):
-    options, dirs = parse_args(argv, baton)
-    if not dirs:
-        dir = os.getcwd()
-    else:
-        dir = dirs[0]
-    r = scripts.get_quota_usage_and_limit(dir)
+    options, args = parse_args(argv, baton)
+    dir = os.path.abspath(args[0]) if args else os.getcwd()
+    r = user.quota(dir)
     if r[0] is None or r[1] is None:
         sys.exit(1)
     print "%d %d" % r