import pickle
import pandas as pd
import code

class ReadPickle:
  def __init__(self,input):

      self.input = input

  def fix_df(self,df):

      df = df.assign(class_target=int(1))

      df.to_pickle(os.path.join(os.path.dirname(self.input), 'info_df_new.pickle'))

  def run(self):
      
      df = pd.read_pickle(self.input)
      #file = pickle.load(open(self.input,"rb"))
      code.interact(local=locals())
      with pd.option_context('display.max_rows', None, 'display.max_columns', None):  # more options can be specified also
         print(df)


if __name__ == "__main__":

  #Command line parsing module
  import argparse

  parser = argparse.ArgumentParser(description='ParserForReadingPickleFile')

  parser.add_argument("--file",dest="input",required=True)

  op  = parser.parse_args()


  picklereader = ReadPickle(op.input)

  picklereader.run()