voussoir/etiquette
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Let atomify take argument web_root.

Browse source
This commit is contained in:
voussoir 2022-11-09 19:42:17 -08:00
parent e89d6fe10e
commit 638dd67cef
1 changed files with 7 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
etiquette/objects.py
View file

@ -433,7 +433,8 @@ class Album(ObjectBase, GroupableMixin):
for photo in photos:
photo.add_tag(tag)
def atomify(self) -> bs4.BeautifulSoup:
def atomify(self, web_root='') -> bs4.BeautifulSoup:
web_root = web_root.rstrip('/')
soup = bs4.BeautifulSoup('', 'xml')
entry = soup.new_tag('entry')
soup.append(entry)
@ -449,7 +450,7 @@ class Album(ObjectBase, GroupableMixin):
link = soup.new_tag('link')
link['rel'] = 'alternate'
link['type'] = 'text/html'
link['href'] = f'/album/{self.id}'
link['href'] = f'{web_root}/album/{self.id}'
entry.append(link)
published = soup.new_tag('published')
@ -991,7 +992,8 @@ class Photo(ObjectBase):
return tag
def atomify(self) -> bs4.BeautifulSoup:
def atomify(self, web_root='') -> bs4.BeautifulSoup:
web_root = web_root.rstrip('/')
soup = bs4.BeautifulSoup('', 'xml')
entry = soup.new_tag('entry')
soup.append(entry)
@ -1007,7 +1009,7 @@ class Photo(ObjectBase):
link = soup.new_tag('link')
link['rel'] = 'alternate'
link['type'] = 'text/html'
link['href'] = f'/photo/{self.id}'
link['href'] = f'{web_root}/photo/{self.id}'
entry.append(link)
published = soup.new_tag('published')
@ -1015,7 +1017,7 @@ class Photo(ObjectBase):
entry.append(published)
content = soup.new_tag('content')
content.string = bs4.CData(f'<img src="/thumbnail/{self.id}.jpg"/>')
content.string = bs4.CData(f'<img src="{web_root}/thumbnail/{self.id}.jpg"/>')
entry.append(content)
typ = soup.new_tag('etiquette:type')