diff options
Diffstat (limited to 'uprofile.py')
-rwxr-xr-x | uprofile.py | 24 |
1 files changed, 15 insertions, 9 deletions
diff --git a/uprofile.py b/uprofile.py index 8a290d5..cc04512 100755 --- a/uprofile.py +++ b/uprofile.py @@ -28,6 +28,7 @@ class Profile(Module): self.output = [] self.modules = [] str = '' + for line in filesystem.read_file('.uprofile/' + path): str += line @@ -82,11 +83,11 @@ class UniversalProfileTool: def get_profiles(self): """ Returns the list of available uprofiles """ - for profile in filesystem.list_dir('.uprofile/'): - if re.match('.*.json$', profile): - self.profiles.append(Profile(profile)) - return - + if filesystem.path_exists('.uprofile'): + for profile in filesystem.list_dir('.uprofile/'): + if re.match('.*.json$', profile): + self.profiles.append(Profile(profile)) + def parse_argv(self, args): global verbose, version profile = None @@ -117,10 +118,15 @@ class UniversalProfileTool: self.get_profiles() profiles = self.profiles elif len(args) < 1: - profile = self.get_profile('folder') - action = profile.get_action('activate') - action.build() - action.do_action(['activate'], profile.modules) + try: + profile = self.get_profile('folder') + action = profile.get_action('activate') + action.build() + action.do_action(['activate'], profile.modules) + except IOError: + self.get_profiles() + profiles = self.profiles + list = True elif len(args) == 1: try: profile = self.get_profile(args[0]) |