From b61c088e58ab534d2b876202a0148420965c8637 Mon Sep 17 00:00:00 2001 From: bt0dotninja Date: Wed, 4 Mar 2020 07:16:40 -0600 Subject: [PATCH] add parseGroup --- output.py | 4 ++-- parseGroup.py | 35 +++++++++++++++++++++++++++++++++++ stats.py | 3 ++- 3 files changed, 39 insertions(+), 3 deletions(-) create mode 100644 parseGroup.py diff --git a/output.py b/output.py index 51ee2cf..f23eb45 100644 --- a/output.py +++ b/output.py @@ -1,4 +1,5 @@ - +from __future__ import absolute_import +from __future__ import print_function from datetime import date, timedelta import fedmsg.meta import fedmsg @@ -275,4 +276,3 @@ def generate_graph(output_json, title, category=None, gtype=None): show_gource(output_json) else: print("[!] That output mode is not supported! Check README for help.") - diff --git a/parseGroup.py b/parseGroup.py new file mode 100644 index 0000000..356b8cb --- /dev/null +++ b/parseGroup.py @@ -0,0 +1,35 @@ +import ConfigParser +import os.path +from fedora.client import AccountSystem, AuthError + +class GroupParser: + + def __init__(self): + self.username = None + self.password = None + self.config = ConfigParser.RawConfigParser() + try: + self.config.read('fas_credentials.cfg') + self.username = self.config.get('fas', 'username').strip('\'') + self.password = self.config.get('fas', 'password').strip('\'') + self.check_config() + except: + print("[*] Invalid / Missing Configuration file.") + def check_config(self): + if self.username.strip('\'') == 'FAS_USERNAME_HERE': + print("[*] Please enter FAS credentials in fas_credentials.cfg") + return False + else: + return True + + def group_users(self, group_name): + group_json = dict() + account = AccountSystem(username=self.username, + password=self.password) + try: + group_json = account.group_members(group_name) + except AuthError: + print("[*] Invalid Username / Password") + return 1 + userlist = [user_desc.values()[0] for user_desc in group_json] + return userlist diff --git a/stats.py b/stats.py index 1f3caa2..fd9f19f 100644 --- a/stats.py +++ b/stats.py @@ -1,3 +1,5 @@ +from __future__ import absolute_import +from __future__ import print_function import fedmsg import fedmsg.meta import calendar @@ -136,4 +138,3 @@ def return_interactions(subcategories): for key in interaction_dict: interaction_dict[key] = Counter(interaction_dict[key]) return interaction_dict -