Skip to content

Commit

Permalink
Refactoring adding a list of campaign dicts to the campaign summary d…
Browse files Browse the repository at this point in the history
…ict for JSON output
  • Loading branch information
JCantu248 committed Jul 27, 2021
1 parent 92ae2b0 commit df5f23d
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
2 changes: 1 addition & 1 deletion src/_version.py
Original file line number Diff line number Diff line change
@@ -1,2 +1,2 @@
"""This file defines the version of this project."""
__version__ = "0.0.6-rc.7"
__version__ = "0.0.6-rc.8"
6 changes: 3 additions & 3 deletions src/tools/gophish_export.py
Original file line number Diff line number Diff line change
Expand Up @@ -267,6 +267,7 @@ def write_campaign_click_summary(api, assessment_id):
"""Output a click summary report to JSON, console, and a text file."""
click_summary = dict()
click_campaign_summary = dict()
campaigns = list()
campaign_ids = get_campaign_ids(api, assessment_id)
num_campaigns = len(campaign_ids)
summary_json = assessment_id + "_click_summary.json"
Expand All @@ -275,7 +276,6 @@ def write_campaign_click_summary(api, assessment_id):
double_print(summary_outfile, "-" * 50)
double_print(summary_outfile, "Number of campaigns: %i" % num_campaigns)
click_summary["Number of campaigns"] = num_campaigns
click_summary["campaigns"] = None

for campaign_id in campaign_ids:
clicks = get_click_data(api, campaign_id)
Expand All @@ -292,7 +292,7 @@ def write_campaign_click_summary(api, assessment_id):
click_campaign_summary["total_clicks"] = api.campaigns.summary(
campaign_id=campaign_id
).stats.clicked
click_summary["campaigns"].append(click_campaign_summary)
campaigns.append(click_campaign_summary)

double_print(summary_outfile, "-" * 50)
double_print(summary_outfile, "Campaign '%i' " % campaign_id)
Expand All @@ -312,7 +312,7 @@ def write_campaign_click_summary(api, assessment_id):
double_print(
summary_outfile, "Total clicks: %i" % click_campaign_summary["total_clicks"]
)

click_summary["campaigns"] = campaigns
summary_outfile.close()
print("Writing out summary JSON to %s_" % summary_json)
with open(summary_json, "w") as fp:
Expand Down

0 comments on commit df5f23d

Please sign in to comment.