ClassIterTree.py

"""
CLASS PROGRAM TO ITERATE ON THE TREE
"""

import ast
import csv
import json
import levels


class IterTree():
    """ Class to iterate tree. """

    # CSV header
    myDataCsv = [['Repository',
                  'File Name',
                  'Class',
                  'Start Line',
                  'End Line',
                  'Displacement',
                  'Level']]

    # JSON dictionary
    myDataJson = {}

    def __init__(self, tree, attrib, file, repo):
        """ Class constructor. """
        self.tree = tree
        self.attrib = attrib
        self.name = file
        self.repo = repo
        self.locate_Tree()

    def locate_Tree(self):
        """ Method iterating on the tree. """
        for self.node in ast.walk(self.tree):
            # Find attributes
            if type(self.node) == eval(self.attrib):
                self.level = ''
                self.clase = ''
                levels.levels(self)
                self.assign_List()
                self.assign_Dict()
                self.read_FileJson()

    def assign_List(self):
        """ Create object list. """
        if (self.clase != '') and (self.level != ''):
            self.list = [self.repo, self.name, self.clase, self.node.lineno,
                         self.node.end_lineno, self.node.col_offset,
                         self.level]
            # print(self.list)
            self.add_Csv()

    def add_Csv(self):
        """ Add object list to CSV. """
        self.myDataCsv.append(self.list)
        # print(self.myDataList)
        self.read_FileCsv()

    def read_FileCsv(self, file_csv=""):
        """ Create and add data in the .csv file. """
        if not file_csv:
            file_csv = open('data.csv', 'w')
            with file_csv:
                writer = csv.writer(file_csv)
                writer.writerows(self.myDataCsv)
        else:
            with open(r'data.csv', 'a') as f:
                writer = csv.writer(f)
                writer.writerow(self.myDataCsv)

    def assign_Dict(self):
        """ Create object dictionary. """
        if (self.clase != '') and (self.level != ''):
            if self.repo not in self.myDataJson:
                self.myDataJson[self.repo] = {}

            if self.name not in self.myDataJson[self.repo]:
                self.myDataJson[self.repo][self.name] = []

            self.myDataJson[self.repo][self.name].append({
                'Class': str(self.clase),
                'Start Line': str(self.node.lineno),
                'End Line': str(self.node.end_lineno),
                'Displacement': str(self.node.col_offset),
                'Level': str(self.level)})

    def read_FileJson(self):
        """ Create and add data in the .json file. """
        with open('data.json', 'w') as file:
            json.dump(self.myDataJson, file, indent=4)