From 1f2a7801fdac113e635a1dbcac87b913917d213f Mon Sep 17 00:00:00 2001
From: AlanCoding <arominge@redhat.com>
Date: Thu, 17 Aug 2017 10:14:36 -0400
Subject: [PATCH] fix flake8 and data gen functionality

---
 .../rbac_dummy_data_generator.py              | 23 +++++++++----------
 1 file changed, 11 insertions(+), 12 deletions(-)

diff --git a/tools/data_generators/rbac_dummy_data_generator.py b/tools/data_generators/rbac_dummy_data_generator.py
index 3855bd728e..f1aab7d049 100755
--- a/tools/data_generators/rbac_dummy_data_generator.py
+++ b/tools/data_generators/rbac_dummy_data_generator.py
@@ -249,7 +249,6 @@ def make_the_data():
                     org.member_role.members.add(jt_admin)
                     org.member_role.members.add(inv_admin)
 
-            organization_gen = yield_choice(organizations)
             print('')
 
             print('# Creating %d users' % n_users)
@@ -542,7 +541,6 @@ def make_the_data():
                 wfjt_nodes = []
                 for i in range(n):
                     ids['nodes'] += 1
-                    node_id = ids['nodes']
                     sys.stdout.write('\r   Assigning %d to %s: %d     ' % (n, wfjt.name, i+ 1))
                     sys.stdout.flush()
                     kwargs = dict(
@@ -710,17 +708,18 @@ def make_the_data():
                 if n:
                     print('')
 
-    if options['pretend']:
-        with transaction.atomic():
-            try:
-                make_the_data()
-                raise Rollback()
-            except Rollback:
-                print('Rolled back changes')
-                pass
 
-    else:
-        make_the_data()
+if options['pretend']:
+    with transaction.atomic():
+        try:
+            make_the_data()
+            raise Rollback()
+        except Rollback:
+            print('Rolled back changes')
+            pass
+
+else:
+    make_the_data()
 
 
 print('')