Merge pull request #116 from PagerDuty/T2D2-306-Bug-broken-in-append-in-public-support-scripts

Fix: script no longer appends on new run

Author: JosefGoodyear

get_info_on_all_users/get_users_by_role.py

@@ -10,9 +10,9 @@
 allowed_roles=['admin','read_only_user','read_only_limited_user','user','limited_user','observer','restricted_access','owner']
 team_managers=[]
 
-def write_rows(column1, column2, column3):
+def write_rows(column1, column2, column3, mode='a+'):
 # one function for writing to csv
-  with open(filename, 'a+') as csvfile:
+  with open(filename, mode) as csvfile:
     writer = csv.writer(csvfile)
     writer.writerow([column1, column2, column3])
 
@@ -73,7 +73,7 @@ def get_teams(session):
     filename = args.filename + '.csv'
   else:
     filename = args.filename
-  write_rows('Name','Role', 'Email')  
+  write_rows('Name','Role', 'Email', mode='w')  
   for role in roles:
     if role == "team_managers":
       get_teams(session)