122 lines
2.9 KiB
Python
122 lines
2.9 KiB
Python
from datetime import datetime
|
|
import re
|
|
from typing import List, Optional
|
|
|
|
DEFAULT_TEMPLATE = "* %U %:description %:tags\n%:link\n%:initial\n"
|
|
|
|
def date2org(t: datetime) -> str:
|
|
return t.strftime("%Y-%m-%d %a")
|
|
|
|
def datetime2orgtime(t: datetime) -> str:
|
|
return t.strftime("%H:%M")
|
|
|
|
def datetime2org(t: datetime) -> str:
|
|
return date2org(t) + " " + datetime2orgtime(t)
|
|
|
|
|
|
def empty(s) -> bool:
|
|
return s is None or len(s.strip()) == 0
|
|
|
|
|
|
def as_org(
|
|
url: str,
|
|
title: str,
|
|
selection: str,
|
|
comment: str,
|
|
tags: List[str],
|
|
org_template: str,
|
|
):
|
|
"""
|
|
Formats captured results according to org template. Supports all sensible (e.g. non-interactive) template expansions from https://orgmode.org/manual/Template-expansion.html#Template-expansion.
|
|
|
|
Additionally, supports :selection, :comment and :tags expansions.
|
|
|
|
You can look at `test_templates` for some specific examples.
|
|
"""
|
|
py_template = re.sub(r'%[:]?([?\w]+)', r'{\1}', org_template)
|
|
|
|
NOW = datetime.now()
|
|
org_date = date2org(NOW)
|
|
org_datetime = datetime2org(NOW)
|
|
|
|
tags_s = '' if len(tags) == 0 else (':' + ':'.join(tags) + ':')
|
|
|
|
# TODO tabulate selection and comments?
|
|
# TODO not sure, maybe add as org quote?
|
|
parts = []
|
|
if not empty(selection):
|
|
parts.extend([
|
|
'Selection:',
|
|
selection,
|
|
])
|
|
if not empty(comment):
|
|
parts.extend([
|
|
'Comment:',
|
|
comment
|
|
])
|
|
initial = '\n'.join(parts)
|
|
|
|
res = py_template.format(
|
|
t=f'<{org_date}>',
|
|
u=f'[{org_date}]',
|
|
T=f'<{org_datetime}>',
|
|
U=f'[{org_datetime}]',
|
|
description=title,
|
|
link=url,
|
|
|
|
initial=initial,
|
|
i=initial,
|
|
|
|
tags=tags_s,
|
|
selection=selection,
|
|
comment=comment,
|
|
annotation=f'[[{url}][{title}]]', # https://orgmode.org/manual/capture-protocol.html
|
|
**{'?': ''}, # just ignore it
|
|
)
|
|
return res
|
|
|
|
# TODO escaping?
|
|
# just checks it's not crashing for now
|
|
def test_templates():
|
|
url = 'https://whatever'
|
|
title = 'hello'
|
|
selection = """some
|
|
selected
|
|
text
|
|
"""
|
|
comment = 'fafewfewf'
|
|
tags = ['aba', 'caba']
|
|
|
|
# https://orgmode.org/guide/Capture-templates.html
|
|
org_templates = [
|
|
DEFAULT_TEMPLATE,
|
|
"* %? [[%:link][%:description]] \nCaptured On: %U",
|
|
"* %:annotation",
|
|
"* %U %:description :protocol: \n %:link \n%:initial \n\n",
|
|
"* %t captured stuff %:tags \n ",
|
|
"\n** Selection\n%:selection\n** Comment\n%:comment\n",
|
|
"""* TODO [#A] %:link
|
|
** Selection
|
|
%:selection
|
|
** Comment
|
|
%:comment
|
|
""",
|
|
"""* %T %:link %:description
|
|
#+BEGIN_QUOTE
|
|
%:selection
|
|
#+END_QUOTE
|
|
""",
|
|
]
|
|
for org_template in org_templates:
|
|
res = as_org(
|
|
url,
|
|
title,
|
|
selection,
|
|
comment,
|
|
tags,
|
|
org_template=org_template
|
|
|
|
)
|
|
print()
|
|
print(res)
|