Commit 9af4ffac authored by Vincent A's avatar Vincent A

weboob.capabilites.recipe: move toKrecipesXml in a separate module

This will avoid importing lxml and requests in weboob.capabilities.*

Also, add Comment in weboob.capabilites.recipe.__all__.
parent a9f9c082
......@@ -25,6 +25,7 @@ from weboob.capabilities.recipe import CapRecipe
from weboob.capabilities.base import empty
from weboob.tools.application.repl import ReplApplication, defaultcount
from weboob.tools.application.formatters.iformatter import IFormatter, PrettyFormatter
from weboob.tools.capabilities.recipe import recipe_to_krecipes_xml
__all__ = ['Cookboob']
......@@ -129,7 +130,7 @@ class Cookboob(ReplApplication):
recipe = self.get_object(id, 'get_recipe')
if recipe:
xmlstring = recipe.toKrecipesXml(backend_name or None)
xmlstring = recipe_to_krecipes_xml(recipe, backend_name or None)
if dest == '-':
print(xmlstring)
else:
......
......@@ -28,6 +28,7 @@ from PyQt5.QtWidgets import QFrame, QFileDialog
from weboob.applications.qcookboob.ui.recipe_ui import Ui_Recipe
from weboob.capabilities.base import empty
from weboob.tools.capabilities.recipe import recipe_to_krecipes_xml
class Recipe(QFrame):
......@@ -106,7 +107,7 @@ class Recipe(QFrame):
dest = result[0]
if not dest.endswith('.kreml'):
dest += '.kreml'
data = self.recipe.toKrecipesXml(author=self.backend.name)
data = recipe_to_krecipes_xml(self.recipe, author=self.backend.name)
try:
with codecs.open(dest, 'w', 'utf-8') as f:
f.write(data)
......
......@@ -18,16 +18,10 @@
# along with weboob. If not, see <http://www.gnu.org/licenses/>.
from .base import Capability, BaseObject, StringField, IntField, Field, empty
from .base import Capability, BaseObject, StringField, IntField, Field
import lxml.etree as ET
import requests
import base64
import re
__all__ = ['Recipe', 'CapRecipe']
__all__ = ['CapRecipe', 'Recipe', 'Comment']
class Comment(BaseObject):
......@@ -66,99 +60,6 @@ class Recipe(BaseObject):
super(Recipe, self).__init__(id, url)
self.title = title
def toKrecipesXml(self, author=None):
"""
Export recipe to KRecipes XML string
"""
sauthor = u''
if not empty(self.author):
sauthor += '%s@' % self.author
if author is None:
sauthor += 'Cookboob'
else:
sauthor += author
header = u'<?xml version="1.0" encoding="UTF-8" ?>\n'
initial_xml = '''\
<krecipes version='2.0-beta2' lang='fr' xmlns:xsi='http://www.w3.org/2001/XMLSchema-instance' xsi:noNamespaceSchemaLocation='krecipes.xsd'>
<krecipes-recipe id='1'>
</krecipes-recipe>
</krecipes>'''
doc = ET.fromstring(initial_xml)
recipe = doc.find('krecipes-recipe')
desc = ET.SubElement(recipe, 'krecipes-description')
title = ET.SubElement(desc, 'title')
title.text = self.title
authors = ET.SubElement(desc, 'author')
authors.text = sauthor
eyield = ET.SubElement(desc, 'yield')
if not empty(self.nb_person):
amount = ET.SubElement(eyield, 'amount')
if len(self.nb_person) == 1:
amount.text = '%s' % self.nb_person[0]
else:
mini = ET.SubElement(amount, 'min')
mini.text = u'%s' % self.nb_person[0]
maxi = ET.SubElement(amount, 'max')
maxi.text = u'%s' % self.nb_person[1]
etype = ET.SubElement(eyield, 'type')
etype.text = 'persons'
if not empty(self.preparation_time):
preptime = ET.SubElement(desc, 'preparation-time')
preptime.text = '%02d:%02d' % (self.preparation_time / 60, self.preparation_time % 60)
if not empty(self.picture_url) and self.picture_url != '':
data = requests.get(self.picture_url).content
datab64 = base64.encodestring(data)[:-1]
pictures = ET.SubElement(desc, 'pictures')
pic = ET.SubElement(pictures, 'pic', {'format': 'JPEG', 'id': '1'})
pic.text = ET.CDATA(datab64)
if not empty(self.ingredients):
ings = ET.SubElement(recipe, 'krecipes-ingredients')
pat = re.compile('^[0-9,.]*')
for i in self.ingredients:
sname = u'%s' % i
samount = ''
sunit = ''
first_nums = pat.match(i).group()
if first_nums != '':
samount = first_nums
sname = i.lstrip('0123456789 ')
ing = ET.SubElement(ings, 'ingredient')
am = ET.SubElement(ing, 'amount')
am.text = samount
unit = ET.SubElement(ing, 'unit')
unit.text = sunit
name = ET.SubElement(ing, 'name')
name.text = sname
if not empty(self.instructions):
instructions = ET.SubElement(recipe, 'krecipes-instructions')
instructions.text = self.instructions
if not empty(self.comments):
ratings = ET.SubElement(recipe, 'krecipes-ratings')
for c in self.comments:
rating = ET.SubElement(ratings, 'rating')
if c.author:
rater = ET.SubElement(rating, 'rater')
rater.text = c.author
if c.text:
com = ET.SubElement(rating, 'comment')
com.text = c.text
crits = ET.SubElement(rating, 'criterion')
if c.rate:
crit = ET.SubElement(crits, 'criteria')
critname = ET.SubElement(crit, 'name')
critname.text = 'Overall'
critstars = ET.SubElement(crit, 'stars')
critstars.text = c.rate.split('/')[0]
return header + ET.tostring(doc, encoding='UTF-8', pretty_print=True).decode('utf-8')
class CapRecipe(Capability):
"""
......
# -*- coding: utf-8 -*-
# Copyright(C) 2018 Julien Veyssier
#
# This file is part of weboob.
#
# weboob is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# weboob 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 Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with weboob. If not, see <http://www.gnu.org/licenses/>.
from __future__ import unicode_literals
import base64
import re
import lxml.etree as ET
import requests
from weboob.capabilities.base import empty
__all__ = ['recipe_to_krecipes_xml']
def recipe_to_krecipes_xml(recipe, author=None):
"""
Export recipe to KRecipes XML string
"""
sauthor = u''
if not empty(recipe.author):
sauthor += '%s@' % recipe.author
if author is None:
sauthor += 'Cookboob'
else:
sauthor += author
header = u'<?xml version="1.0" encoding="UTF-8" ?>\n'
initial_xml = '''\
<krecipes version='2.0-beta2' lang='fr' xmlns:xsi='http://www.w3.org/2001/XMLSchema-instance' xsi:noNamespaceSchemaLocation='krecipes.xsd'>
<krecipes-recipe id='1'>
</krecipes-recipe>
</krecipes>'''
doc = ET.fromstring(initial_xml)
xrecipe = doc.find('krecipes-recipe')
desc = ET.SubElement(xrecipe, 'krecipes-description')
title = ET.SubElement(desc, 'title')
title.text = recipe.title
authors = ET.SubElement(desc, 'author')
authors.text = sauthor
eyield = ET.SubElement(desc, 'yield')
if not empty(recipe.nb_person):
amount = ET.SubElement(eyield, 'amount')
if len(recipe.nb_person) == 1:
amount.text = '%s' % recipe.nb_person[0]
else:
mini = ET.SubElement(amount, 'min')
mini.text = u'%s' % recipe.nb_person[0]
maxi = ET.SubElement(amount, 'max')
maxi.text = u'%s' % recipe.nb_person[1]
etype = ET.SubElement(eyield, 'type')
etype.text = 'persons'
if not empty(recipe.preparation_time):
preptime = ET.SubElement(desc, 'preparation-time')
preptime.text = '%02d:%02d' % (recipe.preparation_time / 60, recipe.preparation_time % 60)
if not empty(recipe.picture_url) and recipe.picture_url != '':
data = requests.get(recipe.picture_url).content
datab64 = base64.encodestring(data)[:-1]
pictures = ET.SubElement(desc, 'pictures')
pic = ET.SubElement(pictures, 'pic', {'format': 'JPEG', 'id': '1'})
pic.text = ET.CDATA(datab64)
if not empty(recipe.ingredients):
ings = ET.SubElement(xrecipe, 'krecipes-ingredients')
pat = re.compile('^[0-9,.]*')
for i in recipe.ingredients:
sname = u'%s' % i
samount = ''
sunit = ''
first_nums = pat.match(i).group()
if first_nums != '':
samount = first_nums
sname = i.lstrip('0123456789 ')
ing = ET.SubElement(ings, 'ingredient')
am = ET.SubElement(ing, 'amount')
am.text = samount
unit = ET.SubElement(ing, 'unit')
unit.text = sunit
name = ET.SubElement(ing, 'name')
name.text = sname
if not empty(recipe.instructions):
instructions = ET.SubElement(xrecipe, 'krecipes-instructions')
instructions.text = recipe.instructions
if not empty(recipe.comments):
ratings = ET.SubElement(xrecipe, 'krecipes-ratings')
for c in recipe.comments:
rating = ET.SubElement(ratings, 'rating')
if c.author:
rater = ET.SubElement(rating, 'rater')
rater.text = c.author
if c.text:
com = ET.SubElement(rating, 'comment')
com.text = c.text
crits = ET.SubElement(rating, 'criterion')
if c.rate:
crit = ET.SubElement(crits, 'criteria')
critname = ET.SubElement(crit, 'name')
critname.text = 'Overall'
critstars = ET.SubElement(crit, 'stars')
critstars.text = c.rate.split('/')[0]
return header + ET.tostring(doc, encoding='UTF-8', pretty_print=True).decode('utf-8')
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment