1 files changed, 105 insertions, 0 deletions
diff --git a/weekly/weekly.py b/weekly/weekly.py
new file mode 100644
index 0000000..076192a
--- /dev/null
+++ b/weekly/weekly.py
@@ -0,0 +1,105 @@
+#!/usr/bin/env python
+import argparse
+import os
+import datetime
+from subprocess import Popen
+from . import __version__
+
+
+def user_message(tm, status):
+    banner = '=' * 80
+    marker = '{0}\n{1}\n{2}\n{0}'.format(banner, tm, status)
+    done = False
+    lines = []
+
+    while not done:
+        msg = ''
+        output = ''
+
+        try:
+            print('>>>', end='')
+            msg = input()
+        except EOFError:
+            done = True
+
+        if msg == '.' or msg == 'EOF':
+            done = True
+
+        if not done:
+            if len(lines) < 1:
+                lines.append(marker)
+            lines.append(msg)
+
+        if lines:
+            output = '\n'.join([x for x in lines])
+            output += '\n'
+
+    return output
+
+
+def show_messages(path):
+    for root, dirs, files in os.walk(path):
+        for f in sorted(files, reverse=False):
+            ingest = os.path.join(root, f)
+            with open(ingest, 'r') as fp:
+                for line in fp:
+                    print(line.rstrip())
+
+
+def main():
+    DATA = os.path.abspath(os.path.expanduser('~/.reports/data'))
+    now = datetime.datetime.now()
+    year, weekno, day = now.isocalendar()
+
+    if not os.path.exists(DATA):
+        os.makedirs(DATA)
+
+    todays_path = os.path.join(DATA, str(year), str(weekno))
+    todays_file = os.path.join(todays_path, str(day))
+
+    if not os.path.exists(todays_path):
+        os.makedirs(todays_path)
+
+    parser = argparse.ArgumentParser()
+    parser.add_argument('-i', '--important', action='store_true',
+                        required=False)
+    parser.add_argument('-r', '--retrieve', action='store_true')
+    parser.add_argument('-e', '--edit', action='store_true')
+    parser.add_argument('-V', '--version', action='store_true')
+    args = parser.parse_args()
+
+    if args.version:
+        print(__version__)
+        return 0
+
+    if args.edit:
+        editor = '/usr/bin/vim'
+        if 'EDITOR' in os.environ:
+            editor = os.environ['EDITOR']
+        command = ' '.join([editor, todays_file])
+        proc = Popen(command.split())
+        proc.communicate()
+        retval = proc.wait()
+        if retval is None:
+            retval = 1
+
+        return retval
+
+    if args.retrieve:
+        show_messages(todays_path)
+        return 0
+
+    status = 'NOTE'
+    if args.important:
+        status = 'IMPORTANT'
+
+    message = user_message(now, status)
+    with open(todays_file, 'a') as fp:
+        fp.write(message)
+        fp.write(os.linesep)
+
+    return 0
+
+
+if __name__ == '__main__':
+    sys.exit(main())