from __future__ import unicode_literals import frappe, json from frappe.frappeclient import FrappeClient from frappe.desk.form.load import get_attachments from six import string_types @frappe.whitelist() def call_hub_method(method, params=None): connection = get_hub_connection() if isinstance(params, string_types): params = json.loads(params) params.update({ 'cmd': 'hub.hub.api.' + method }) response = connection.post_request(params) return response def map_fields(items): field_mappings = get_field_mappings() table_fields = [d.fieldname for d in frappe.get_meta('Item').get_table_fields()] hub_seller = frappe.db.get_value('Hub Settings' , 'Hub Settings', 'company_email') for item in items: for fieldname in table_fields: item.pop(fieldname, None) for mapping in field_mappings: local_fieldname = mapping.get('local_fieldname') remote_fieldname = mapping.get('remote_fieldname') value = item.get(local_fieldname) item.pop(local_fieldname, None) item[remote_fieldname] = value item['doctype'] = 'Hub Item' item['hub_seller'] = hub_seller item.pop('attachments', None) return items @frappe.whitelist() def get_valid_items(search_value=''): items = frappe.get_list( 'Item', fields=["*"], filters={ 'item_name': ['like', '%' + search_value + '%'], 'publish_in_hub': 0 }, order_by="modified desc" ) valid_items = filter(lambda x: x.image and x.description, items) def prepare_item(item): item.source_type = "local" item.attachments = get_attachments('Item', item.item_code) return item valid_items = map(prepare_item, valid_items) return valid_items @frappe.whitelist() def publish_selected_items(items_to_publish): items_to_publish = json.loads(items_to_publish) if not len(items_to_publish): frappe.throw('No items to publish') for item in items_to_publish: item_code = item.get('item_code') frappe.db.set_value('Item', item_code, 'publish_in_hub', 1) frappe.get_doc({ 'doctype': 'Hub Tracked Item', 'item_code': item_code, 'hub_category': item.get('hub_category'), 'image_list': item.get('image_list') }).insert(ignore_if_duplicate=True) items = map_fields(items_to_publish) try: item_sync_preprocess() load_base64_image_from_items(items) # TODO: Publish Progress connection = get_hub_connection() connection.insert_many(items) item_sync_postprocess({ 'status': 'Success', 'stats': len(items) }) except Exception as e: frappe.log_error(message=e, title='Hub Sync Error') def item_sync_preprocess(): hub_seller = frappe.db.get_value("Hub Settings", "Hub Settings", "company_email") response = call_hub_method('add_hub_seller_activity', { 'hub_seller': hub_seller, 'activity_details': json.dumps({ 'subject': 'Publishing items', 'status': 'Success' }) }) if response: frappe.db.set_value("Hub Settings", "Hub Settings", "sync_in_progress", 1) return response else: frappe.throw('Unable to update remote activity') def item_sync_postprocess(sync_details): hub_seller = frappe.db.get_value("Hub Settings", "Hub Settings", "company_email") response = call_hub_method('add_hub_seller_activity', { 'hub_seller': hub_seller, 'activity_details': json.dumps({ 'subject': 'Publishing items:' + sync_details['status'], 'content': str(sync_details['stats']) + ' items synced.' }) }) if response: frappe.db.set_value('Hub Settings', 'Hub Settings', 'last_sync_datetime', frappe.utils.now()) else: frappe.throw('Unable to update remote activity') frappe.db.set_value('Hub Settings', 'Hub Settings', 'sync_in_progress', 0) def load_base64_image_from_items(items): import io, base64, urllib, os from frappe.utils.file_manager import get_file_path for item in items: file_path = item['image'] file_name = os.path.basename(file_path) if file_path.startswith('http'): url = file_path file_path = os.path.join('/tmp', file_name) urllib.urlretrieve(url, file_path) else: file_path = os.path.abspath(get_file_path(file_path)) with io.open(file_path, 'rb') as f: image_data = json.dumps({ 'file_name': file_name, 'base64': base64.b64encode(f.read()) }) item['image'] = image_data def get_hub_connection(): read_only = True if frappe.db.exists('Data Migration Connector', 'Hub Connector'): hub_connector = frappe.get_doc('Data Migration Connector', 'Hub Connector') # full rights to user who registered as hub_seller if hub_connector.username == frappe.session.user: read_only = False if not read_only: hub_connection = hub_connector.get_connection() return hub_connection.connection # read-only connection if read_only: hub_connection = FrappeClient(frappe.conf.hub_url) return hub_connection def get_field_mappings(): return []