Sort products when fetching from Woo API
default sort is by 'date desc' which is not always determinate also, combine logic for fetching products
This commit is contained in:
		
							parent
							
								
									ddced3e90c
								
							
						
					
					
						commit
						3d0f5ddd81
					
				
					 3 changed files with 63 additions and 35 deletions
				
			
		| 
						 | 
				
			
			@ -2,7 +2,7 @@
 | 
			
		|||
################################################################################
 | 
			
		||||
#
 | 
			
		||||
#  Rattail -- Retail Software Framework
 | 
			
		||||
#  Copyright © 2010-2021 Lance Edgar
 | 
			
		||||
#  Copyright © 2010-2022 Lance Edgar
 | 
			
		||||
#
 | 
			
		||||
#  This file is part of Rattail.
 | 
			
		||||
#
 | 
			
		||||
| 
						 | 
				
			
			@ -27,6 +27,7 @@ WooCommerce model importers
 | 
			
		|||
from woocommerce import API as WooAPI
 | 
			
		||||
 | 
			
		||||
from rattail import importing
 | 
			
		||||
from rattail_woocommerce.woocommerce.util import get_woocommerce_products
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class ToWooCommerce(importing.Importer):
 | 
			
		||||
| 
						 | 
				
			
			@ -114,23 +115,11 @@ class ProductImporter(ToWooCommerce):
 | 
			
		|||
        Fetch existing products from WooCommerce.
 | 
			
		||||
        """
 | 
			
		||||
        cache = {}
 | 
			
		||||
        page = 1
 | 
			
		||||
        while True:
 | 
			
		||||
            response = self.api.get('products', params={'per_page': 100,
 | 
			
		||||
                                                        'page': page})
 | 
			
		||||
            for product in response.json():
 | 
			
		||||
                data = self.normalize_local_object(product)
 | 
			
		||||
                normal = self.normalize_cache_object(product, data)
 | 
			
		||||
                key = self.get_cache_key(product, normal)
 | 
			
		||||
                cache[key] = normal
 | 
			
		||||
 | 
			
		||||
            # TODO: this seems a bit hacky, is there a better way?
 | 
			
		||||
            link = response.headers.get('Link')
 | 
			
		||||
            if link and 'rel="next"' in link:
 | 
			
		||||
                page += 1
 | 
			
		||||
            else:
 | 
			
		||||
                break
 | 
			
		||||
 | 
			
		||||
        for product in get_woocommerce_products(self.api):
 | 
			
		||||
            data = self.normalize_local_object(product)
 | 
			
		||||
            normal = self.normalize_cache_object(product, data)
 | 
			
		||||
            key = self.get_cache_key(product, normal)
 | 
			
		||||
            cache[key] = normal
 | 
			
		||||
        return cache
 | 
			
		||||
 | 
			
		||||
    def get_single_local_object(self, key):
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
							
								
								
									
										53
									
								
								rattail_woocommerce/woocommerce/util.py
									
										
									
									
									
										Normal file
									
								
							
							
						
						
									
										53
									
								
								rattail_woocommerce/woocommerce/util.py
									
										
									
									
									
										Normal file
									
								
							| 
						 | 
				
			
			@ -0,0 +1,53 @@
 | 
			
		|||
# -*- coding: utf-8; -*-
 | 
			
		||||
################################################################################
 | 
			
		||||
#
 | 
			
		||||
#  Rattail -- Retail Software Framework
 | 
			
		||||
#  Copyright © 2010-2022 Lance Edgar
 | 
			
		||||
#
 | 
			
		||||
#  This file is part of Rattail.
 | 
			
		||||
#
 | 
			
		||||
#  Rattail 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 3 of the License, or (at your option) any later
 | 
			
		||||
#  version.
 | 
			
		||||
#
 | 
			
		||||
#  Rattail 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
 | 
			
		||||
#  Rattail.  If not, see <http://www.gnu.org/licenses/>.
 | 
			
		||||
#
 | 
			
		||||
################################################################################
 | 
			
		||||
"""
 | 
			
		||||
WooCommerce utilities
 | 
			
		||||
"""
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def get_woocommerce_products(api):
 | 
			
		||||
    """
 | 
			
		||||
    Fetch and return all products from Woo API.
 | 
			
		||||
    """
 | 
			
		||||
    products = []
 | 
			
		||||
    page = 1
 | 
			
		||||
    while True:
 | 
			
		||||
 | 
			
		||||
        # TODO: 100 seems to be the max allowed per page?
 | 
			
		||||
        # although docs do not seem to mention a limit..
 | 
			
		||||
        # https://woocommerce.github.io/woocommerce-rest-api-docs/?python#list-all-products
 | 
			
		||||
        response = api.get('products', params={'per_page': 100,
 | 
			
		||||
                                               'page': page,
 | 
			
		||||
                                               'orderby': 'id',
 | 
			
		||||
                                               'order': 'asc'})
 | 
			
		||||
 | 
			
		||||
        products.extend(response.json())
 | 
			
		||||
 | 
			
		||||
        # TODO: this seems a bit hacky, is there a better way?
 | 
			
		||||
        link = response.headers.get('Link')
 | 
			
		||||
        if link and 'rel="next"' in link:
 | 
			
		||||
            page += 1
 | 
			
		||||
        else:
 | 
			
		||||
            break
 | 
			
		||||
 | 
			
		||||
    return products
 | 
			
		||||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue