def init(ui, path, *args, **opts):
if not args:
dest = path
else:
dest = os.path.normpath(os.path.join(os.getcwd(), args[0]))
if dest is None:
raise AppError("You aren't in a couchapp.")
app.document(ui, dest, True)
def pushapps(ui, source, dest, *args, **opts):
export = opts.get('export', False)
noatomic = opts.get('no_atomic', False)
dbs = ui.get_dbs(dest)
apps = []
source = os.path.normpath(os.path.join(os.getcwd(), source))
for d in os.listdir(source):
appdir = os.path.join(source, d)
print appdir
if os.path.isdir(appdir) and os.path.isfile(os.path.join(appdir, '.couchapprc')):
localdoc = app.document(ui, appdir)
hooks.hook(ui, appdir, "pre-push", dbs=dbs, pushapps=True)
if export or not noatomic:
apps.append(localdoc)
else:
localdoc.push(dbs, True)
hooks.hook(ui, appdir, "post-push", dbs=dbs, pushapps=True)
if apps:
if export:
docs = []
docs.append([localdoc.doc() for localdoc in apps])
jsonobj = {'docs': docs}
if opts.get('output') is not None:
ui.write_json(opts.get('output'), json.dumps(jsonobj))
else:
print json.dumps(jsonobj)
return 0
else:
for db in dbs:
docs = []
docs = [doc.doc(db) for doc in apps]
db.save_docs(docs)
return 0
def pushdocs(ui, source, dest, *args, **opts):
export = opts.get('export', False)
noatomic = opts.get('no_atomic', False)
dbs = ui.get_dbs(dest)
docs = []
for d in os.listdir(source):
docdir = os.path.join(source, d)
if docdir.startswith('.'):
continue
elif os.path.isfile(docdir):
if d.endswith(".json"):
doc = ui.read_json(docdir)
docid, ext = os.path.splitext(d)
doc.setdefault('_id', docid)
doc.setdefault('couchapp', {})
if export or not noatomic:
docs.append(doc)
else:
for db in dbs:
db.save_doc(doc)
else:
doc = app.document(ui, docdir)
if export or not noatomic:
docs.append(doc)
else:
doc.push(dbs, True)
if docs:
if export:
docs1 = []
for doc in docs:
if hasattr(doc, 'doc'):
docs1.append(doc.doc())
else:
docs1.append(doc)
jsonobj = {'docs': docs}
if opts.get('output') is not None:
ui.write_json(opts.get('output'), json.dumps(jsonobj))
else:
print json.dumps(jsonobj)
else:
for db in dbs:
docs1 = []
for doc in docs:
if hasattr(doc, 'doc'):
docs1.append(doc.doc(db))
else:
newdoc = doc.copy()
try:
olddoc = db.get_doc(doc['_id'])
newdoc.update({'_rev': olddoc['_rev']})
except ResourceNotFound:
pass
docs1.append(newdoc)
db.save_docs(docs1)
return 0
def pushapps(ui, source, dest, *args, **opts):
export = opts.get('export', False)
noatomic = opts.get('no_atomic', False)
browse = opts.get('browse', False)
dbs = ui.get_dbs(dest)
apps = []
source = os.path.normpath(os.path.join(os.getcwd(), source))
for d in os.listdir(source):
appdir = os.path.join(source, d)
if os.path.isdir(appdir) and os.path.isfile(os.path.join(appdir,
'.couchapprc')):
localdoc = app.document(ui, appdir)
hooks.hook(ui, appdir, "pre-push", dbs=dbs, pushapps=True)
if export or not noatomic:
apps.append(localdoc)
else:
localdoc.push(dbs, True, browse)
hooks.hook(ui, appdir, "post-push", dbs=dbs, pushapps=True)
if apps:
if export:
docs = []
docs.append([localdoc.doc() for localdoc in apps])
jsonobj = {'docs': docs}
if opts.get('output') is not None:
ui.write_json(opts.get('output'), json.dumps(jsonobj))
else:
print json.dumps(jsonobj)
return 0
else:
for db in dbs:
docs = []
docs = [doc.doc(db) for doc in apps]
try:
db.save_docs(docs)
except BulkSaveError, e:
docs1 = []
for doc in e.errors:
try:
doc['_rev'] = db.last_rev(doc['_id'])
docs1.append(doc)
except ResourceNotFound:
pass
if docs1:
db.save_docs(docs1)
def push(ui, path, *args, **opts):
export = opts.get('export', False)
atomic = opts.get('no_atomic', False)
browse = opts.get('browse', False)
dest = None
doc_path = None
if len(args) < 2:
if export:
if path is None and args:
doc_path = args[0]
else:
doc_path = path
else:
doc_path = path
if args:
dest = args[0]
else:
doc_path = os.path.normpath(os.path.join(os.getcwd(), args[0]))
dest = args[1]
if doc_path is None:
raise AppError("You aren't in a couchapp.")
_maybe_reload(ui, path, doc_path)
localdoc = app.document(ui, doc_path, create=False, docid=opts.get('docid'))
if export:
if opts.get('output'):
ui.write_json(opts.get('output'), str(localdoc))
else:
print str(localdoc)
return 0
dbs = ui.get_dbs(dest)
hooks.hook(ui, doc_path, "pre-push", dbs=dbs)
localdoc.push(dbs, atomic, browse)
hooks.hook(ui, doc_path, "post-push", dbs=dbs)
docspath = os.path.join(doc_path, '_docs')
if os.path.exists(docspath):
pushdocs(ui, docspath, dest, *args, **opts)
return 0
