initial commit
This commit is contained in:
commit
18f2b8bef5
2 changed files with 156 additions and 0 deletions
1
.gitignore
vendored
Normal file
1
.gitignore
vendored
Normal file
|
@ -0,0 +1 @@
|
||||||
|
/content/
|
155
import.py
Normal file
155
import.py
Normal file
|
@ -0,0 +1,155 @@
|
||||||
|
#!/usr/bin/env python
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
#
|
||||||
|
# import.py
|
||||||
|
#
|
||||||
|
# Copyright 2021 <avana@disperazione>
|
||||||
|
#
|
||||||
|
# This program is free software; you can redistribute it and/or modify
|
||||||
|
# it under the terms of the GNU General Public License as published by
|
||||||
|
# the Free Software Foundation; either version 2 of the License, or
|
||||||
|
# (at your option) any later version.
|
||||||
|
#
|
||||||
|
# This program is distributed in the hope that it will be useful,
|
||||||
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
# GNU General Public License for more details.
|
||||||
|
#
|
||||||
|
# You should have received a copy of the GNU General Public License
|
||||||
|
# along with this program; if not, write to the Free Software
|
||||||
|
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
|
||||||
|
# MA 02110-1301, USA.
|
||||||
|
#
|
||||||
|
#
|
||||||
|
|
||||||
|
import datetime
|
||||||
|
from itertools import chain
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
from lxml import etree
|
||||||
|
import toml
|
||||||
|
|
||||||
|
class Article:
|
||||||
|
BASE = Path('content')
|
||||||
|
def __init__(self, post_id: str, title: str, body: str, img=None, date=None):
|
||||||
|
self.post_id = post_id
|
||||||
|
self.title = title
|
||||||
|
self.body = body
|
||||||
|
self.img = img
|
||||||
|
self.date = date
|
||||||
|
|
||||||
|
def get_dir(self):
|
||||||
|
return self.BASE / 'posts'
|
||||||
|
|
||||||
|
def get_fname(self):
|
||||||
|
return self.get_dir() / (self.post_id + '.md')
|
||||||
|
|
||||||
|
def get_frontmatter(self):
|
||||||
|
fm = {
|
||||||
|
'title': self.title,
|
||||||
|
}
|
||||||
|
if self.img:
|
||||||
|
fm['cover'] = self.img
|
||||||
|
if self.date:
|
||||||
|
fm['date'] = self.date
|
||||||
|
return fm
|
||||||
|
|
||||||
|
def save(self):
|
||||||
|
with open(self.get_fname(), 'w') as buf:
|
||||||
|
buf.write('+++\n')
|
||||||
|
toml.dump(self.get_frontmatter(), buf)
|
||||||
|
buf.write('+++\n')
|
||||||
|
buf.write(self.body)
|
||||||
|
|
||||||
|
def __str__(self):
|
||||||
|
return "<%s '%s'>" % (self.__class__.__name__, self.post_id)
|
||||||
|
|
||||||
|
class Event(Article):
|
||||||
|
def __init__(self, time=None, location=None, **kwargs):
|
||||||
|
super().__init__(**kwargs)
|
||||||
|
self.time = time
|
||||||
|
self.location = location
|
||||||
|
|
||||||
|
def get_dir(self):
|
||||||
|
return self.BASE / 'events'
|
||||||
|
|
||||||
|
def get_frontmatter(self):
|
||||||
|
fm = super().get_frontmatter()
|
||||||
|
fm['eventDate'] = self.time
|
||||||
|
fm['location'] = self.location
|
||||||
|
return fm
|
||||||
|
|
||||||
|
def xpath_in(xpath: str, elem):
|
||||||
|
xp = etree.XPath(xpath)
|
||||||
|
return xp(elem)
|
||||||
|
|
||||||
|
def get_text(node):
|
||||||
|
if type(node) is str:
|
||||||
|
return node
|
||||||
|
s = node.text
|
||||||
|
if s is None:
|
||||||
|
s = ''
|
||||||
|
for child in node:
|
||||||
|
s += get_text(s) #etree.tostring(child, encoding='unicode')
|
||||||
|
return s
|
||||||
|
|
||||||
|
def stringify_children(node):
|
||||||
|
parts = ([node.text] +
|
||||||
|
list(chain(*([c.text, etree.tostring(c).decode('utf8'), c.tail] for c in node.getchildren()))) +
|
||||||
|
[node.tail])
|
||||||
|
# filter removes possible Nones in texts and tails
|
||||||
|
return ''.join(filter(None, parts))
|
||||||
|
|
||||||
|
def main(args):
|
||||||
|
articles = []
|
||||||
|
parser = etree.HTMLParser()
|
||||||
|
dom = etree.parse(open('home.html'), parser)
|
||||||
|
for article in dom.xpath('//article'):
|
||||||
|
elems = {}
|
||||||
|
# XXX: questa cosa funziona male
|
||||||
|
elems['body'] = '\n'.join(l.strip() for l in stringify_children(article).split('\n'))
|
||||||
|
try:
|
||||||
|
elems['title'] = str(xpath_in('.//h2/text()', article)[0])
|
||||||
|
except IndexError:
|
||||||
|
continue
|
||||||
|
|
||||||
|
elems['post_id'] = article.attrib['id']
|
||||||
|
|
||||||
|
try:
|
||||||
|
time_el = xpath_in('.//time[@datetime]', article)[0]
|
||||||
|
elems['time'] = datetime.datetime.fromisoformat(time_el.attrib['datetime'])
|
||||||
|
elems['date'] = elems['time']
|
||||||
|
except IndexError:
|
||||||
|
pass
|
||||||
|
|
||||||
|
links = xpath_in('.//a', article)
|
||||||
|
for l in links:
|
||||||
|
if l.attrib.get('href', '').startswith('https://forteprenestino.net'):
|
||||||
|
elems['location'] = 'CSOA Forte Prenestino'
|
||||||
|
|
||||||
|
# print('t', [get_text(e) for e in xpath_in('./h2', article)])
|
||||||
|
try:
|
||||||
|
elems['img'] = xpath_in('.//img', article)[0].attrib['src']
|
||||||
|
except IndexError:
|
||||||
|
img = None
|
||||||
|
|
||||||
|
if 'time' in elems and 'location' in elems:
|
||||||
|
cls = Event
|
||||||
|
else:
|
||||||
|
cls = Article
|
||||||
|
for k in ['time', 'location']:
|
||||||
|
if k in elems:
|
||||||
|
del elems[k]
|
||||||
|
|
||||||
|
a = cls(**elems)
|
||||||
|
articles.append(a)
|
||||||
|
|
||||||
|
for a in articles:
|
||||||
|
print(a)
|
||||||
|
a.save()
|
||||||
|
return 0
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
import sys
|
||||||
|
sys.exit(main(sys.argv))
|
||||||
|
|
Loading…
Reference in a new issue