Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
itminedu
synnefo
Commits
83dfe64a
Commit
83dfe64a
authored
Apr 22, 2013
by
Giorgos Korfiatis
Browse files
astakos: Use `with' to open file in management command
parent
a0f09a72
Changes
1
Show whitespace changes
Inline
Side-by-side
snf-astakos-app/astakos/im/management/commands/user-set-initial-quota.py
View file @
83dfe64a
...
@@ -49,6 +49,7 @@ AddResourceArgs = namedtuple('AddQuotaArgs', ('resource',
...
@@ -49,6 +49,7 @@ AddResourceArgs = namedtuple('AddQuotaArgs', ('resource',
'capacity'
,
'capacity'
,
))
))
class
Command
(
BaseCommand
):
class
Command
(
BaseCommand
):
help
=
"""Import user quota limits from file or set quota
help
=
"""Import user quota limits from file or set quota
for a single user from the command line
for a single user from the command line
...
@@ -171,12 +172,8 @@ for a single user from the command line
...
@@ -171,12 +172,8 @@ for a single user from the command line
qh_sync_user
(
user
.
id
)
qh_sync_user
(
user
.
id
)
def
import_from_file
(
self
,
location
):
def
import_from_file
(
self
,
location
):
try
:
f
=
open
(
location
,
'r'
)
except
IOError
,
e
:
raise
CommandError
(
e
)
users
=
set
()
users
=
set
()
with
open
(
location
)
as
f
:
for
line
in
f
.
readlines
():
for
line
in
f
.
readlines
():
try
:
try
:
t
=
line
.
rstrip
(
'
\n
'
).
split
(
' '
)
t
=
line
.
rstrip
(
'
\n
'
).
split
(
' '
)
...
@@ -190,7 +187,8 @@ for a single user from the command line
...
@@ -190,7 +187,8 @@ for a single user from the command line
user
=
AstakosUser
.
objects
.
get
(
uuid
=
user
)
user
=
AstakosUser
.
objects
.
get
(
uuid
=
user
)
users
.
add
(
user
.
id
)
users
.
add
(
user
.
id
)
except
AstakosUser
.
DoesNotExist
:
except
AstakosUser
.
DoesNotExist
:
self
.
stdout
.
write
(
'Not found user having uuid: %s
\n
'
%
user
)
self
.
stdout
.
write
(
'Not found user having uuid: %s
\n
'
%
user
)
continue
continue
else
:
else
:
try
:
try
:
...
@@ -198,9 +196,6 @@ for a single user from the command line
...
@@ -198,9 +196,6 @@ for a single user from the command line
except
Exception
,
e
:
except
Exception
,
e
:
self
.
stdout
.
write
(
'Failed to policy: %s
\n
'
%
e
)
self
.
stdout
.
write
(
'Failed to policy: %s
\n
'
%
e
)
continue
continue
finally
:
f
.
close
()
qh_sync_users
(
users
)
qh_sync_users
(
users
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment