offers

aleksandr-rakov · Dec 15, 2013 · 1c4d7c4 · 1c4d7c4
1 parent c3e4d54
commit 1c4d7c4
Show file tree

Hide file tree

Showing 2 changed files with 42 additions and 20 deletions.
diff --git a/pyayml/__init__.py b/pyayml/__init__.py
@@ -242,20 +242,20 @@ def __init__(self):
     def get_xml(self):
         return HEADER+etree.tostring(self.root,encoding="windows-1251",pretty_print=True)
 
-    def set_shop(self,show_data):
-        shop = etree.SubElement(self.root, 'shop')
-        self.shop=shop
+    def set_shop(self,shop_data):
+        shop_tag = etree.SubElement(self.root, 'shop')
+        self.shop=shop_tag
 
         for key in SHOP_TAGS:
-            if key in show_data:
-                etree.SubElement(shop, key).text = show_data[key]
+            if key in shop_data:
+                etree.SubElement(shop_tag, key).text = shop_data[key]
 
     def set_currencies(self,currencies_data):
-        currencies = etree.SubElement(self.shop, 'currencies')
-        self.currencies=currencies
+        currencies_tag = etree.SubElement(self.shop, 'currencies')
+        self.currencies=currencies_tag
         for currency in currencies_data:
             #xxx check id, rate, plus
-            etree.SubElement(currencies, 'currency', rate=currency['rate'], id=currency['id'])
+            etree.SubElement(currencies_tag, 'currency', rate=currency['rate'], id=currency['id'])
 
     def set_categories(self, categories_data):
         categories_tag = etree.SubElement(self.shop, 'categories')
@@ -266,14 +266,10 @@ def set_categories(self, categories_data):
                 etree.SubElement(categories_tag, 'category', id=category['id'], text = category['name'])
 
 
-    # def add_offers(self, shop):
-    #     offers = etree.SubElement(shop,'offers')
-    #     for product in Product.objects.filter(active=True):
-    #         offer = etree.SubElement(offers,'offer', id=str(product.id), available="true")
-    #         etree.SubElement(offer,'url').text =  YML_CONFIG['url'] + product.get_absolute_url()
-    #         etree.SubElement(offer,'price').text =  str(product.get_price())
-    #         etree.SubElement(offer,'currencyId').text =  product.get_currency()
-    #         etree.SubElement(offer,'categoryId').text =  str(product.category.id)
-    #         etree.SubElement(offer,'picture').text = YML_CONFIG['url'] + product.head_image.url
-    #         etree.SubElement(offer,'delivery').text = "true"
-    #         etree.SubElement(offer,'name').text = product.get_name()
+    def set_offers(self, offers_data):
+        offers_tag = etree.SubElement(self.shop,'offers')
+        for offer in offers_data:
+            offer_tag = etree.SubElement(offers_tag,'offer', id=offer['id'], available="true")
+            for key in OFFERS_TAGS['']:
+                if key in offer:
+                    etree.SubElement(offer_tag, key).text = offer[key]
diff --git a/pyayml/tests.py b/pyayml/tests.py
@@ -16,15 +16,41 @@ def test_001(self):
             },
             'currencies':[
                 {'id':'RUR','rate':'CBFR'},
+                {'id':'USD','rate':'CBFR'},
             ],
             'categories': [
                 {'id':'1','name':u'АЛЛ Продуктс'},
                 {'id':'2','name':'sub cat products','parentId':'1'},
-            ]
+            ],
+            'offers':[
+                {
+                    'id':'1',
+                    'url':'http://mail.ru/1212',
+                    'price':'1234',
+                    'currencyId':'RUR',
+                    'categoryId':'1',
+                    'picture':'http://mail.ru/1212.jpg',
+                    'name':'product1',
+                    'vendor':'AND',
+                    'description':'some text here',
+                },
+                {
+                    'id':'2',
+                    'url':'http://mail.ru/1212',
+                    'price':'1234',
+                    'currencyId':'RUR',
+                    'categoryId':'2',
+                    'picture':'http://mail.ru/1212.jpg',
+                    'name':'product1',
+                    'vendor':'AND',
+                    'description':'some text here',
+                },
+            ],
         }
         obj.set_shop(data['shop'])
         obj.set_currencies(data['currencies'])
         obj.set_categories(data['categories'])
+        obj.set_offers(data['offers'])
 
         print obj.get_xml()