org_add_user - cleaning up and adding to output

cknowles-moz · Aug 18, 2023 · e872dc9 · e872dc9
1 parent 9263c1a
commit e872dc9
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/org_add_user.py b/org_add_user.py
@@ -76,7 +76,12 @@ def main():
         print(f"An exception on creation occurred: {error}")
         exit(1)
     else:
-        print(f"Invited user {args.username} to org {args.org}.")
+        if args.teams is None:
+            print(f"Invited user {args.username} to org {args.org}.")
+        else:
+            print(
+                f"Invited user {args.username} to org {args.org} and teams {', '.join(args.teams)}"
+            )
 
 
 if __name__ == "__main__":