loic/pnote
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add json export

Browse source
This commit is contained in:
Loïc Guégan 2024-04-19 11:28:45 +02:00
parent aec3a51ab4
commit 9b55d1a249
1 changed files with 36 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

51
pnote/tools/export.py
View file

@ -1,17 +1,36 @@
from pnote.tools.tool import Tool from pnote.tools.tool import Tool
import argparse, os, sys import argparse, os, sys, json
from datetime import datetime from datetime import datetime
class ToolExport(Tool): class ToolExport(Tool):
def __init__(self): def __init__(self):
self.template_file=None self.template_file=None
self.date_format="%s" self.date_format=None
def add_parser(self,subparsers): def add_parser(self,subparsers):
p = subparsers.add_parser("export", description="Export notes from subpaths in stdin") p = subparsers.add_parser("export", description="Export notes from subpaths in stdin")
p.add_argument("--template", help="Export notes following a template file") p.add_argument("--template", help="Export notes following a template file")
p.add_argument("--date-format", help="Specify a format use by date in the output") p.add_argument("--date-format", help="Specify a format use by date in the output")
p.add_argument("--json", help="Export notes in json format", action='store_true')
def get_subpath_data(self, project, subpath):
data=None
with open(project.getpath(subpath),"r") as noteFile:
data={
"content":noteFile.read(),
"created":project.getfileinfo(subpath,"created"),
"added":project.getfileinfo(subpath,"added"),
"id":project.getfileinfo(subpath,"id"),
"hostname":project.getfileinfo(subpath,"hostname"),
"platform":project.getfileinfo(subpath,"platform"),
"tags":project.listtags(subpath),
"subpath":subpath
}
if self.date_format is not None:
data["created"]=datetime.fromtimestamp(data["created"]).strftime(self.date_format)
data["added"]=datetime.fromtimestamp(data["added"]).strftime(self.date_format)
return data
def run(self, project, args): def run(self, project, args):
if args.date_format: if args.date_format:
@ -24,17 +43,19 @@ class ToolExport(Tool):
self.template_file=args.template self.template_file=args.template
for line in sys.stdin: for line in sys.stdin:
subpath=line.rstrip() subpath=line.rstrip()
with open(project.getpath(subpath),"r") as noteFile: with open(self.template_file,"r") as tplFile:
with open(self.template_file,"r") as tplFile: variables=self.get_subpath_data(project,subpath)
variables={ for line in tplFile:
"content":noteFile.read(), print(line.format(**variables),end="")
"created":datetime.fromtimestamp(project.getfileinfo(subpath,"created")).strftime(self.date_format), elif args.json:
"added":datetime.fromtimestamp(project.getfileinfo(subpath,"added")).strftime(self.date_format), print("[")
"id":project.getfileinfo(subpath,"id"), first=True
"hostname":project.getfileinfo(subpath,"hostname"), for line in sys.stdin:
"platform":project.getfileinfo(subpath,"platform"), subpath=line.rstrip()
"tags":project.listtags(subpath), if not first:
"subpath":subpath} print(",",end="")
for line in tplFile: print(json.dumps(self.get_subpath_data(project,subpath),indent=4))
print(line.format(**variables),end="") first=False
print("]")