post_file_to_db.py

"""
Script to upload documents t the measurements DB.

Usage: python {} doc_id file_one [file_two] ... [file_n]
"""
import pynedm
from clint.textui.progress import Bar as ProgressBar
import json
import sys
from glob import iglob
import os


if len(sys.argv) < 3:
   print(sys.modules[__name__].__doc__)
   sys.exit(1)

o = pynedm.ProcessObject(uri="http://raid.nedm1",
  username="admin",
  password="pw")

bar = None
def callback(read, total):
    global bar
    if bar is None:
        bar = ProgressBar(expected_size=total, filled_char='=')
    bar.show(read)

_doc = sys.argv[1]
_db = "nedm%2Fmeasurements"

for _fn in iglob(sys.argv[2]):
    uploading = o.upload_file(_fn, _doc, db=_db, callback=callback)
    if "ok" in uploading:
        print("Done: {}".format(_fn))
        os.unlink(_fn)
    else:
        print("Error: {}\n, exiting".format(uploading))
        sys.exit(1)