X-Git-Url: https://scripts.mit.edu/gitweb/wizard.git/blobdiff_plain/b5b5882199c3b4a43ee00900a595ed479af86131..6a287fd5e5ada7a97b0f3df96bb4c7ccba4b562d:/wizard/command/quota.py

diff --git a/wizard/command/quota.py b/wizard/command/quota.py
index 2a640c9..11fb04d 100644
--- 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