Example #1
0
File: jot.py Project: harshays/jot
class Jot(object):

    funcs = {
        'count': lambda df, fp: int(len(df) + 1),
        'timestamp': lambda df, fp: dt.now().strftime(fp['timestamp_format']),
        'user': lambda df, fp: getpass.getuser()
    }

    def if_file_exists(func):
        'only call function if filename valid'
        def _func(*args, **kw):
            self, fname = args[0], args[1]
            if not self.utils.file_exists(fname):
                self.utils.prettyprint('No such file: '+fname, 'red')
                return
            return func(*args, **kw)
        return _func

    def __init__(self, config_path):
        self.config_obj  = JotConfig(config_path)
        self.jot_dir = self.config_obj.dir
        self.cfg_dir = self.config_obj.fpath
        self.cli     = JotCLI()
        self.utils   = JotUtils(self.jot_dir, '.csv')

    def parse_and_call(self):
        action, args = self.cli.parse()
        func = getattr(self, action, None)
        if func: return (func(*args) if isinstance(args, list) else func())
        self.prettyprint('No such action: '+action)

    def add(self, fname, msg):
        'jot'
        msg = msg.replace('\n', ' ').strip()
        columns = self.config_obj.get_file_cols(fname)
        fexists = self.utils.file_exists(fname)
        if not fexists: self.utils.prettyprint("Adding new file: "+fname, 'green')
        fpath = self.utils.get_fpath(fname)
        fprops = self.config_obj.get_file_config(fname)
        df = pd.read_csv(fpath, header=None) if fexists else pd.DataFrame(columns=columns)
        row = [self.funcs[col](df, fprops) for col in columns[:-1]] + [msg]
        df.loc[len(df)] = row
        self.utils.write_df(fname, df)

    @if_file_exists
    def view(self, fname):
        'view file content'
        df = pd.read_csv(self.utils.get_fpath(fname), header=None)
        df.columns = self.config_obj.get_file_cols(fname)
        print (df.to_string(index=False))

    @if_file_exists
    def reset(self, fname):
        'reset count, clean messages'
        fpath = self.utils.get_fpath(fname)
        columns = self.config_obj.get_file_cols(fname)
        df = pd.read_csv(fpath, header=None)
        df.columns = columns
        df['message'] = df['message'].str.strip().str.replace('\n', ' ')
        if 'count' in columns: df['count'] = pd.Series(range(1, len(df)+1))
        self.utils.write_df(fname, df)

    @if_file_exists
    def manual(self, fname):
        'open jot file in vim'
        fpath = self.utils.get_fpath(fname)
        os.system('vim {}'.format(fpath))

    def search(self, kw):
        'search for keywords in all files'
        for keyword in [k.strip() for k in kw.split(',')]:
            self.utils.prettyprint("Keyword: "+keyword, 'bold')
            grep = subprocess.Popen(['grep', '-Ri', keyword, self.jot_dir], stdout=subprocess.PIPE)
            for result in grep.stdout.readlines():
                fpath, msg = result.split(':', 1)
                msg = msg.replace('\n', '')
                fname = fpath.rsplit('/', 1)[-1].split('.')[0]
                print ('    {}: {}'.format(fname, msg))
            print ('\n')

    def list(self):
        'list all jot files'
        for num, jfile in enumerate(self.utils.all_files(), 1):
            self.utils.prettyprint('{}. {}'.format(num, jfile.rsplit('.', 1)[0]), 'bold')

    def config(self):
        'edit config in vim'
        os.system('vim {}'.format(self.cfg_dir))
Example #2
0
File: jot.py Project: harshays/jot
 def __init__(self, config_path):
     self.config_obj  = JotConfig(config_path)
     self.jot_dir = self.config_obj.dir
     self.cfg_dir = self.config_obj.fpath
     self.cli     = JotCLI()
     self.utils   = JotUtils(self.jot_dir, '.csv')