diff --git a/wcs/has_role.py b/wcs/has_role.py index 05850e7..677c0d7 100644 --- a/wcs/has_role.py +++ b/wcs/has_role.py @@ -2,7 +2,6 @@ import os import sys from ..qommon.ctl import Command, make_option -from wcs.roles import Role class Cmd(Command): @@ -18,7 +17,7 @@ class Cmd(Command): publisher = publisher.WcsPublisher.create_publisher(register_tld_names=False) publisher.app_dir = os.path.join(publisher.app_dir, sub_options.domain) publisher.set_config() - for role in Role.select(): + for role in publisher.role_class.select(): if role.name == args[0]: sys.exit(0) sys.exit(1)