Turning Tabular Data into a Nested JSON File || Python

      import click
      import io
      import json
      import pandas as pd

      def main():
        try:
          input_file = 'input_location/input_file.csv'
          output_file = 'output_location/data.json'
          df = pd.read_csv(f'{input_file}')
          d = {'name': 'root', 'children': create_entries(df)}
          data = json.dumps(d)
          load = json.loads(data)
          with open(f'{output_file}', 'w') as f:
            json.dump(load, f)
          click.echo(f'Success!')
        except Exception as e:
          click.echo(f'Process failed because {e}')

      def create_entries(df):
        entries = []
        #base case - only 2 columns left
        if df.shape[1] == 2:
          #iterate on rows
          for i in range(df.shape[0]):
            entries.append({'name': df.iloc[i,0], df.columns[-1]: df.iloc[i,1]})
        #recursive case
        else:
          values = set(df.iloc[:,0])
          for v in values:
            entries.append(
              {'name': v, 'children': create_entries(df.loc[df.iloc[:,0] == v].iloc[:,1:])})
        return entries

      if __name__ == '__main__':
        main()