def test_existing_properties(self):
backend.create_user(username2)
backend.create_property(user=username2, key=propkey1, value=propval3) # propval1 -> json
backend.create_property(user=username2, key="date joined", value=propval3)
path = os.path.join(self.base, 'users1.json')
with capture() as (stdout, stderr):
restauth_import([path])
self.assertCountEqual(backend.list_users(), [username1, username2, username3])
pattern = r'^%s: Property "%s" already exists\.$' % (username2, propkey1)
self.assertHasLine(stdout, pattern)
self.assertHasLine(stdout, r'^%s: Property "date joined" already exists\.$' % username2)
expected_props = {
propkey1: propval3, # propva1 is in json-file - we don't overwrite!
propkey2: propval2,
u'last login': u'2013-12-01 19:27:44', # date from json file
}
props = backend.get_properties(user=username2)
# delete 'date joined' prop because it was created by the backend and
# restauth-import doesn't overwrite in this invocation:
del props['date joined']
self.assertEqual(props, expected_props)
def get(self, request, largs, name):
"""Get all properties of a user."""
if not request.user.has_perm('Users.props_list'):
return HttpResponseForbidden()
# If UserNotFound: 404 Not Found
props = backend.get_properties(user=name)
return HttpRestAuthResponse(request, props)
def assertProperties(self, username, expected):
actual = backend.get_properties(user=username)
self.assertTrue('date joined' in actual, 'No "date joined" property in %s' % actual)
del actual['date joined']
self.assertDictEqual(actual, expected)
def main(args=None):
# parse arguments
args = parser.parse_args(args=args)
if args.action == 'add':
password = args.get_password(args)
if args.password_generated:
print(args.pwd)
backend.set_password(user=args.user, password=password)
elif args.action in ['ls', 'list']:
for username in sorted(backend.list_users()):
if six.PY3: # pragma: py3
print(username)
else: # pragma: py2
print(username.encode('utf-8'))
elif args.action == 'verify':
if not args.pwd: # pragma: no cover
args.pwd = getpass.getpass('password: '******'Ok.')
else:
print('Failed.')
sys.exit(1)
elif args.action == 'set-password':
password = args.get_password(args)
if args.password_generated:
print(args.pwd)
backend.set_password(user=args.user, password=args.pwd)
elif args.action == 'view':
props = backend.get_properties(user=args.user)
if 'date joined' in props:
print('Joined: %s' % props['date joined'])
if 'last login' in props:
print('Last login: %s' % props['last login'])
if args.service:
groups = backend.list_groups(service=args.service, user=args.user)
if groups:
print('Groups: %s' % ', '.join(sorted(groups)))
else:
print('No groups.')
else:
if backend.SUPPORTS_GROUP_VISIBILITY:
groups = {}
none_groups = backend.list_groups(service=None, user=args.user)
for service in Service.objects.all():
subgroups = backend.list_groups(service=service, user=args.user)
if subgroups:
groups[service.username] = subgroups
if groups or none_groups:
print('Groups:')
if none_groups:
print('* no service: %s' % ', '.join(sorted(none_groups)))
for service, groups in sorted(groups.items(), key=lambda t: t[0]):
print('* %s: %s' % (service, ', '.join(sorted(groups))))
else:
print('No groups.')
else:
groups = backend.list_groups(service=None, user=args.user)
if groups:
print('Groups: %s' % ', '.join(sorted(groups)))
else:
print('No groups.')
elif args.action == 'rename':
try:
backend.rename_user(user=args.user, name=args.name)
except UserNotFound as e:
parser.error("%s: %s" % (args.name, e))
except UserExists as e:
parser.error("%s: %s" % (args.name, e))
elif args.action in ['delete', 'rm', 'remove']: # pragma: no branch
backend.remove_user(user=args.user)
