Prateeksha Singh | 6495d53 | 2018-08-01 16:38:39 +0530 | [diff] [blame] | 1 | from __future__ import unicode_literals |
Prateeksha Singh | a8065ff | 2018-08-31 22:41:26 +0530 | [diff] [blame] | 2 | |
| 3 | import frappe |
| 4 | import json |
| 5 | import io |
| 6 | import base64 |
| 7 | import os |
| 8 | import requests |
| 9 | |
| 10 | from frappe import _ |
Prateeksha Singh | 6495d53 | 2018-08-01 16:38:39 +0530 | [diff] [blame] | 11 | from frappe.frappeclient import FrappeClient |
Prateeksha Singh | a525d12 | 2018-08-19 22:31:33 +0530 | [diff] [blame] | 12 | from frappe.desk.form.load import get_attachments |
Faris Ansari | 95e4141 | 2018-08-30 11:46:14 +0530 | [diff] [blame] | 13 | from frappe.utils.file_manager import get_file_path |
Suraj Shetty | 3cd0c54 | 2018-08-21 16:29:06 +0530 | [diff] [blame] | 14 | from six import string_types |
Prateeksha Singh | 6495d53 | 2018-08-01 16:38:39 +0530 | [diff] [blame] | 15 | |
Prateeksha Singh | 810ae01 | 2018-09-01 01:14:29 +0530 | [diff] [blame^] | 16 | current_user = frappe.session.user |
| 17 | |
Prateeksha Singh | 6495d53 | 2018-08-01 16:38:39 +0530 | [diff] [blame] | 18 | @frappe.whitelist() |
Prateeksha Singh | a8065ff | 2018-08-31 22:41:26 +0530 | [diff] [blame] | 19 | def register_seller(**kwargs): |
| 20 | settings = frappe.get_single('Marketplace Settings') |
| 21 | settings.update(kwargs) |
| 22 | settings.users = [] |
| 23 | |
| 24 | validate_registerer() |
Prateeksha Singh | a8065ff | 2018-08-31 22:41:26 +0530 | [diff] [blame] | 25 | message = settings.register() |
| 26 | |
| 27 | if message.get('email'): |
| 28 | settings.registered = 1 |
| 29 | |
Prateeksha Singh | 747c464 | 2018-09-01 00:12:17 +0530 | [diff] [blame] | 30 | settings.append('users', { |
Prateeksha Singh | 810ae01 | 2018-09-01 01:14:29 +0530 | [diff] [blame^] | 31 | 'user': current_user, |
| 32 | 'name': current_user, |
Prateeksha Singh | 747c464 | 2018-09-01 00:12:17 +0530 | [diff] [blame] | 33 | 'username': kwargs.get('username'), |
| 34 | 'password': message.get('password') |
| 35 | }) |
Prateeksha Singh | a8065ff | 2018-08-31 22:41:26 +0530 | [diff] [blame] | 36 | |
Prateeksha Singh | 747c464 | 2018-09-01 00:12:17 +0530 | [diff] [blame] | 37 | user_emails = kwargs.get('users').strip()[:-1].split(', ') |
| 38 | for email in user_emails: |
| 39 | settings.append('users', {'user': email, 'name': email}) |
| 40 | |
| 41 | settings.insert() |
Prateeksha Singh | a8065ff | 2018-08-31 22:41:26 +0530 | [diff] [blame] | 42 | |
| 43 | return message |
| 44 | |
| 45 | def validate_registerer(): |
Prateeksha Singh | 810ae01 | 2018-09-01 01:14:29 +0530 | [diff] [blame^] | 46 | if current_user == 'Administrator': |
Prateeksha Singh | a8065ff | 2018-08-31 22:41:26 +0530 | [diff] [blame] | 47 | frappe.throw(_('Please login as another user to register on Marketplace')) |
| 48 | |
| 49 | if 'System Manager' not in frappe.get_roles(): |
| 50 | frappe.throw(_('Only users with System Manager role can register on Marketplace'), frappe.PermissionError) |
| 51 | |
Prateeksha Singh | a8065ff | 2018-08-31 22:41:26 +0530 | [diff] [blame] | 52 | @frappe.whitelist() |
Prateeksha Singh | 6495d53 | 2018-08-01 16:38:39 +0530 | [diff] [blame] | 53 | def call_hub_method(method, params=None): |
| 54 | connection = get_hub_connection() |
| 55 | |
Suraj Shetty | 3cd0c54 | 2018-08-21 16:29:06 +0530 | [diff] [blame] | 56 | if isinstance(params, string_types): |
Prateeksha Singh | 6495d53 | 2018-08-01 16:38:39 +0530 | [diff] [blame] | 57 | params = json.loads(params) |
| 58 | |
| 59 | params.update({ |
| 60 | 'cmd': 'hub.hub.api.' + method |
| 61 | }) |
| 62 | |
| 63 | response = connection.post_request(params) |
| 64 | return response |
| 65 | |
Prateeksha Singh | caadd8a | 2018-08-21 19:59:15 +0530 | [diff] [blame] | 66 | def map_fields(items): |
| 67 | field_mappings = get_field_mappings() |
| 68 | table_fields = [d.fieldname for d in frappe.get_meta('Item').get_table_fields()] |
| 69 | |
Faris Ansari | f9a66c1 | 2018-08-31 16:15:06 +0530 | [diff] [blame] | 70 | hub_seller = frappe.db.get_value('Marketplace Settings' , 'Marketplace Settings', 'company_email') |
Prateeksha Singh | caadd8a | 2018-08-21 19:59:15 +0530 | [diff] [blame] | 71 | |
| 72 | for item in items: |
| 73 | for fieldname in table_fields: |
| 74 | item.pop(fieldname, None) |
| 75 | |
| 76 | for mapping in field_mappings: |
| 77 | local_fieldname = mapping.get('local_fieldname') |
| 78 | remote_fieldname = mapping.get('remote_fieldname') |
| 79 | |
| 80 | value = item.get(local_fieldname) |
| 81 | item.pop(local_fieldname, None) |
| 82 | item[remote_fieldname] = value |
| 83 | |
| 84 | item['doctype'] = 'Hub Item' |
| 85 | item['hub_seller'] = hub_seller |
Prateeksha Singh | 40f7c46 | 2018-08-24 16:38:34 +0530 | [diff] [blame] | 86 | item.pop('attachments', None) |
Prateeksha Singh | caadd8a | 2018-08-21 19:59:15 +0530 | [diff] [blame] | 87 | |
| 88 | return items |
| 89 | |
Prateeksha Singh | 6495d53 | 2018-08-01 16:38:39 +0530 | [diff] [blame] | 90 | @frappe.whitelist() |
| 91 | def get_valid_items(search_value=''): |
| 92 | items = frappe.get_list( |
| 93 | 'Item', |
| 94 | fields=["*"], |
| 95 | filters={ |
Faris Ansari | 00f6aef | 2018-08-30 18:54:28 +0530 | [diff] [blame] | 96 | 'disabled': 0, |
Prateeksha Singh | 6495d53 | 2018-08-01 16:38:39 +0530 | [diff] [blame] | 97 | 'item_name': ['like', '%' + search_value + '%'], |
| 98 | 'publish_in_hub': 0 |
| 99 | }, |
| 100 | order_by="modified desc" |
| 101 | ) |
| 102 | |
| 103 | valid_items = filter(lambda x: x.image and x.description, items) |
| 104 | |
Prateeksha Singh | a525d12 | 2018-08-19 22:31:33 +0530 | [diff] [blame] | 105 | def prepare_item(item): |
Prateeksha Singh | 6495d53 | 2018-08-01 16:38:39 +0530 | [diff] [blame] | 106 | item.source_type = "local" |
Prateeksha Singh | a525d12 | 2018-08-19 22:31:33 +0530 | [diff] [blame] | 107 | item.attachments = get_attachments('Item', item.item_code) |
Prateeksha Singh | 6495d53 | 2018-08-01 16:38:39 +0530 | [diff] [blame] | 108 | return item |
| 109 | |
Prateeksha Singh | 02c176c | 2018-08-29 14:27:47 +0530 | [diff] [blame] | 110 | valid_items = map(prepare_item, valid_items) |
Prateeksha Singh | a525d12 | 2018-08-19 22:31:33 +0530 | [diff] [blame] | 111 | |
Prateeksha Singh | 6495d53 | 2018-08-01 16:38:39 +0530 | [diff] [blame] | 112 | return valid_items |
| 113 | |
| 114 | @frappe.whitelist() |
| 115 | def publish_selected_items(items_to_publish): |
| 116 | items_to_publish = json.loads(items_to_publish) |
| 117 | if not len(items_to_publish): |
Prateeksha Singh | 0a60d1c | 2018-08-19 19:39:00 +0530 | [diff] [blame] | 118 | frappe.throw('No items to publish') |
Prateeksha Singh | 6495d53 | 2018-08-01 16:38:39 +0530 | [diff] [blame] | 119 | |
Prateeksha Singh | 40f7c46 | 2018-08-24 16:38:34 +0530 | [diff] [blame] | 120 | for item in items_to_publish: |
| 121 | item_code = item.get('item_code') |
Prateeksha Singh | 6495d53 | 2018-08-01 16:38:39 +0530 | [diff] [blame] | 122 | frappe.db.set_value('Item', item_code, 'publish_in_hub', 1) |
| 123 | |
Prateeksha Singh | 0a60d1c | 2018-08-19 19:39:00 +0530 | [diff] [blame] | 124 | frappe.get_doc({ |
| 125 | 'doctype': 'Hub Tracked Item', |
| 126 | 'item_code': item_code, |
Prateeksha Singh | 40f7c46 | 2018-08-24 16:38:34 +0530 | [diff] [blame] | 127 | 'hub_category': item.get('hub_category'), |
| 128 | 'image_list': item.get('image_list') |
Faris Ansari | 811a2de | 2018-08-28 17:17:29 +0530 | [diff] [blame] | 129 | }).insert(ignore_if_duplicate=True) |
Prateeksha Singh | 0a60d1c | 2018-08-19 19:39:00 +0530 | [diff] [blame] | 130 | |
Prateeksha Singh | caadd8a | 2018-08-21 19:59:15 +0530 | [diff] [blame] | 131 | |
Prateeksha Singh | 40f7c46 | 2018-08-24 16:38:34 +0530 | [diff] [blame] | 132 | items = map_fields(items_to_publish) |
Prateeksha Singh | caadd8a | 2018-08-21 19:59:15 +0530 | [diff] [blame] | 133 | |
Prateeksha Singh | 6495d53 | 2018-08-01 16:38:39 +0530 | [diff] [blame] | 134 | try: |
Prateeksha Singh | a6f184b | 2018-08-30 02:30:40 +0530 | [diff] [blame] | 135 | item_sync_preprocess(len(items)) |
Faris Ansari | 811a2de | 2018-08-28 17:17:29 +0530 | [diff] [blame] | 136 | load_base64_image_from_items(items) |
Prateeksha Singh | caadd8a | 2018-08-21 19:59:15 +0530 | [diff] [blame] | 137 | |
| 138 | # TODO: Publish Progress |
| 139 | connection = get_hub_connection() |
| 140 | connection.insert_many(items) |
| 141 | |
Prateeksha Singh | a6f184b | 2018-08-30 02:30:40 +0530 | [diff] [blame] | 142 | item_sync_postprocess() |
Prateeksha Singh | 6495d53 | 2018-08-01 16:38:39 +0530 | [diff] [blame] | 143 | except Exception as e: |
Prateeksha Singh | 02c176c | 2018-08-29 14:27:47 +0530 | [diff] [blame] | 144 | frappe.log_error(message=e, title='Hub Sync Error') |
Prateeksha Singh | 6495d53 | 2018-08-01 16:38:39 +0530 | [diff] [blame] | 145 | |
Prateeksha Singh | a6f184b | 2018-08-30 02:30:40 +0530 | [diff] [blame] | 146 | def item_sync_preprocess(intended_item_publish_count): |
| 147 | response = call_hub_method('pre_items_publish', { |
| 148 | 'intended_item_publish_count': intended_item_publish_count |
Prateeksha Singh | 6495d53 | 2018-08-01 16:38:39 +0530 | [diff] [blame] | 149 | }) |
| 150 | |
| 151 | if response: |
Faris Ansari | f9a66c1 | 2018-08-31 16:15:06 +0530 | [diff] [blame] | 152 | frappe.db.set_value("Marketplace Settings", "Marketplace Settings", "sync_in_progress", 1) |
Prateeksha Singh | 6495d53 | 2018-08-01 16:38:39 +0530 | [diff] [blame] | 153 | return response |
| 154 | else: |
| 155 | frappe.throw('Unable to update remote activity') |
| 156 | |
Prateeksha Singh | a6f184b | 2018-08-30 02:30:40 +0530 | [diff] [blame] | 157 | def item_sync_postprocess(): |
| 158 | response = call_hub_method('post_items_publish', {}) |
Prateeksha Singh | 6495d53 | 2018-08-01 16:38:39 +0530 | [diff] [blame] | 159 | if response: |
Faris Ansari | f9a66c1 | 2018-08-31 16:15:06 +0530 | [diff] [blame] | 160 | frappe.db.set_value('Marketplace Settings', 'Marketplace Settings', 'last_sync_datetime', frappe.utils.now()) |
Prateeksha Singh | 6495d53 | 2018-08-01 16:38:39 +0530 | [diff] [blame] | 161 | else: |
| 162 | frappe.throw('Unable to update remote activity') |
| 163 | |
Faris Ansari | f9a66c1 | 2018-08-31 16:15:06 +0530 | [diff] [blame] | 164 | frappe.db.set_value('Marketplace Settings', 'Marketplace Settings', 'sync_in_progress', 0) |
Prateeksha Singh | caadd8a | 2018-08-21 19:59:15 +0530 | [diff] [blame] | 165 | |
Faris Ansari | 811a2de | 2018-08-28 17:17:29 +0530 | [diff] [blame] | 166 | |
| 167 | def load_base64_image_from_items(items): |
Faris Ansari | 811a2de | 2018-08-28 17:17:29 +0530 | [diff] [blame] | 168 | for item in items: |
| 169 | file_path = item['image'] |
| 170 | file_name = os.path.basename(file_path) |
Faris Ansari | 8683bd8 | 2018-08-29 16:18:36 +0530 | [diff] [blame] | 171 | base64content = None |
Faris Ansari | 811a2de | 2018-08-28 17:17:29 +0530 | [diff] [blame] | 172 | |
| 173 | if file_path.startswith('http'): |
Faris Ansari | 8683bd8 | 2018-08-29 16:18:36 +0530 | [diff] [blame] | 174 | # fetch content and then base64 it |
Faris Ansari | 811a2de | 2018-08-28 17:17:29 +0530 | [diff] [blame] | 175 | url = file_path |
Faris Ansari | 8683bd8 | 2018-08-29 16:18:36 +0530 | [diff] [blame] | 176 | response = requests.get(url) |
| 177 | base64content = base64.b64encode(response.content) |
Faris Ansari | 811a2de | 2018-08-28 17:17:29 +0530 | [diff] [blame] | 178 | else: |
Faris Ansari | 8683bd8 | 2018-08-29 16:18:36 +0530 | [diff] [blame] | 179 | # read file then base64 it |
Faris Ansari | 811a2de | 2018-08-28 17:17:29 +0530 | [diff] [blame] | 180 | file_path = os.path.abspath(get_file_path(file_path)) |
Faris Ansari | 8683bd8 | 2018-08-29 16:18:36 +0530 | [diff] [blame] | 181 | with io.open(file_path, 'rb') as f: |
| 182 | base64content = base64.b64encode(f.read()) |
Faris Ansari | 811a2de | 2018-08-28 17:17:29 +0530 | [diff] [blame] | 183 | |
Faris Ansari | 8683bd8 | 2018-08-29 16:18:36 +0530 | [diff] [blame] | 184 | image_data = json.dumps({ |
| 185 | 'file_name': file_name, |
| 186 | 'base64': base64content |
| 187 | }) |
Faris Ansari | 811a2de | 2018-08-28 17:17:29 +0530 | [diff] [blame] | 188 | |
| 189 | item['image'] = image_data |
| 190 | |
| 191 | |
Prateeksha Singh | 6495d53 | 2018-08-01 16:38:39 +0530 | [diff] [blame] | 192 | def get_hub_connection(): |
Prateeksha Singh | 810ae01 | 2018-09-01 01:14:29 +0530 | [diff] [blame^] | 193 | settings = frappe.get_single('Marketplace Settings') |
| 194 | marketplace_url = settings.marketplace_url |
| 195 | current_user_records = filter(lambda x: x.user == current_user, settings.users) |
Faris Ansari | 64202ad | 2018-08-29 15:02:39 +0530 | [diff] [blame] | 196 | |
Prateeksha Singh | 810ae01 | 2018-09-01 01:14:29 +0530 | [diff] [blame^] | 197 | if current_user_records: |
| 198 | record = current_user_records[0] |
| 199 | password = frappe.get_doc('Hub User', current_user).get_password() |
| 200 | hub_connection = FrappeClient(marketplace_url, record.user, password) |
Faris Ansari | 64202ad | 2018-08-29 15:02:39 +0530 | [diff] [blame] | 201 | return hub_connection |
Prateeksha Singh | 810ae01 | 2018-09-01 01:14:29 +0530 | [diff] [blame^] | 202 | else: |
| 203 | read_only_hub_connection = FrappeClient(marketplace_url) |
| 204 | return read_only_hub_connection |
| 205 | |
Prateeksha Singh | caadd8a | 2018-08-21 19:59:15 +0530 | [diff] [blame] | 206 | |
| 207 | def get_field_mappings(): |
| 208 | return [] |