def generate_data()

in src/LdapDataGenerator.py [0:0]

• 30 lines of code
• 4 McCabe index (conditional complexity)

    def generate_data(self):
        tree = Tree()
        df = pd.DataFrame(columns=['Employee_id','Manager_id','name', 'dob','dept'])

        for emp_id in range(self.number_of_employees):
            if emp_id >1:
                Manager_id=random.randrange(0, emp_id-1, 1)
                tree.create_node(emp_id,emp_id, parent=Manager_id)
                df = df.append({'Employee_id': int(emp_id), 
                                'Manager_id':int(Manager_id),
                                'name':self.random_name(),
                                'dob':self.random_dob(),
                                'dept':self.random_department()
                            }, ignore_index=True)
            elif emp_id==1:
                tree.create_node(emp_id,emp_id, parent=0)
                df = df.append({'Employee_id': int(emp_id), 
                                'Manager_id':int(0),
                                'name':self.random_name(),
                                'dob':self.random_dob(),
                                'dept':self.random_department()
                            }, ignore_index=True)
            else:
                tree.create_node(emp_id,emp_id)
                df = df.append({'Employee_id': int(emp_id), 
                                'Manager_id':None,
                                'name':self.random_name(),
                                'dob':self.random_dob(),
                                'dept':None
                            }, ignore_index=True)
        return tree, df