Odoo18-Base/addons/hw_drivers/iot_handlers/drivers/DisplayDriver_L.py
2025-01-06 10:57:38 +07:00

203 lines
7.9 KiB
Python

# -*- coding: utf-8 -*-
# Part of Odoo. See LICENSE file for full copyright and licensing details.
import jinja2
import json
import logging
import netifaces as ni
import os
import socket
import subprocess
import time
import werkzeug
import urllib3
from odoo import http
from odoo.addons.hw_drivers.browser import Browser, BrowserState
from odoo.addons.hw_drivers.connection_manager import connection_manager
from odoo.addons.hw_drivers.driver import Driver
from odoo.addons.hw_drivers.main import iot_devices
from odoo.addons.hw_drivers.tools import helpers
from odoo.addons.hw_drivers.tools.helpers import Orientation
from odoo.tools.misc import file_path
path = os.path.realpath(os.path.join(os.path.dirname(__file__), '../../views'))
loader = jinja2.FileSystemLoader(path)
jinja_env = jinja2.Environment(loader=loader, autoescape=True)
jinja_env.filters["json"] = json.dumps
pos_display_template = jinja_env.get_template('pos_display.html')
_logger = logging.getLogger(__name__)
class DisplayDriver(Driver):
connection_type = 'display'
def __init__(self, identifier, device):
super(DisplayDriver, self).__init__(identifier, device)
self.device_type = 'display'
self.device_connection = 'hdmi'
self.device_name = device['name']
self.owner = False
self.customer_display_data = {}
self.url, self.orientation = helpers.load_browser_state()
if self.device_identifier != 'distant_display':
self._x_screen = device.get('x_screen', '0')
self.browser = Browser(
self.url or 'http://localhost:8069/point_of_sale/display/' + self.device_identifier,
self._x_screen,
os.environ.copy(),
)
self.update_url(self.load_url())
self._actions.update({
'update_url': self._action_update_url,
'display_refresh': self._action_display_refresh,
})
self.set_orientation(self.orientation)
@classmethod
def supported(cls, device):
return True # All devices with connection_type == 'display' are supported
@classmethod
def get_default_display(cls):
displays = list(filter(lambda d: iot_devices[d].device_type == 'display', iot_devices))
return len(displays) and iot_devices[displays[0]]
def run(self):
while self.device_identifier != 'distant_display' and not self._stopped.is_set() and "pos_customer_display" not in self.url:
time.sleep(60)
if self.url != 'http://localhost:8069/point_of_sale/display/' + self.device_identifier and self.browser.state != BrowserState.KIOSK:
# Refresh the page every minute
self.browser.refresh()
def update_url(self, url=None):
self.url = (
url
or helpers.load_browser_state()[0]
or 'http://localhost:8069/point_of_sale/display/' + self.device_identifier
)
browser_state = BrowserState.KIOSK if "/pos-self/" in self.url else BrowserState.FULLSCREEN
self.browser.open_browser(self.url, browser_state)
def load_url(self):
url = None
if helpers.get_odoo_server_url():
# disable certifiacte verification
urllib3.disable_warnings()
http = urllib3.PoolManager(cert_reqs='CERT_NONE')
try:
response = http.request(
'GET',
"%s/iot/box/%s/display_url" % (helpers.get_odoo_server_url(), helpers.get_mac_address())
)
if response.status == 200:
data = json.loads(response.data.decode('utf8'))
url = data[self.device_identifier]
except json.decoder.JSONDecodeError:
url = response.data.decode('utf8')
except Exception:
pass
return url
def _action_update_url(self, data):
if self.device_identifier != 'distant_display':
self.update_url(data.get('url'))
def _action_display_refresh(self, data):
if self.device_identifier != 'distant_display':
self.browser.refresh()
def set_orientation(self, orientation=Orientation.NORMAL):
if self.device_identifier == 'distant_display':
# Avoid calling xrandr if no display is connected
return
if type(orientation) is not Orientation:
raise TypeError("orientation must be of type Orientation")
subprocess.run(['xrandr', '-o', orientation.value], check=True)
subprocess.run([file_path('hw_drivers/tools/sync_touchscreen.sh'), str(int(self._x_screen) + 1)], check=False)
helpers.save_browser_state(orientation=orientation)
class DisplayController(http.Controller):
@http.route('/hw_proxy/customer_facing_display', type='json', auth='none', cors='*')
def customer_facing_display(self, action, pos_id=None, access_token=None, data=None):
display = self.ensure_display()
if action in ['open', 'open_kiosk']:
origin = helpers.get_odoo_server_url()
if action == 'open_kiosk':
url = f"{origin}/pos-self/{pos_id}?access_token={access_token}"
display.set_orientation(Orientation.RIGHT)
else:
url = f"{origin}/pos_customer_display/{pos_id}/{access_token}"
display.update_url(url)
return {'status': 'opened'}
if action == 'close':
helpers.unlink_file('browser-url.conf')
helpers.unlink_file('screen-orientation.conf')
display.browser.disable_kiosk_mode()
display.update_url()
return {'status': 'closed'}
if action == 'set':
display.customer_display_data = data
return {'status': 'updated'}
if action == 'get':
return {'status': 'retrieved', 'data': display.customer_display_data}
if action == 'rotate_screen':
display.set_orientation(Orientation(data))
return {'status': 'rotated'}
def ensure_display(self):
display: DisplayDriver = DisplayDriver.get_default_display()
if not display:
raise werkzeug.exceptions.ServiceUnavailable(description="No display connected")
return display
@http.route(['/point_of_sale/display', '/point_of_sale/display/<string:display_identifier>'], auth='none')
def display(self, display_identifier=None):
interfaces = ni.interfaces()
display_ifaces = []
for iface_id in interfaces:
if 'wlan' in iface_id or 'eth' in iface_id:
iface_obj = ni.ifaddresses(iface_id)
ifconfigs = iface_obj.get(ni.AF_INET, [])
essid = helpers.get_ssid()
for conf in ifconfigs:
if conf.get('addr'):
display_ifaces.append({
'iface_id': iface_id,
'essid': essid,
'addr': conf.get('addr'),
'icon': 'sitemap' if 'eth' in iface_id else 'wifi',
})
default_display = DisplayDriver.get_default_display()
if not display_identifier and default_display != 0:
display_identifier = default_display.device_identifier
return pos_display_template.render({
'title': "Odoo -- Point of Sale",
'breadcrumb': 'POS Client display',
'display_ifaces': display_ifaces,
'display_identifier': display_identifier,
'hostname': socket.gethostname(),
'pairing_code': connection_manager.pairing_code,
})
@http.route('/point_of_sale/iot_devices', type='json', auth='none', methods=['POST'])
def get_iot_devices(self):
iot_device = [{
'name': iot_devices[device].device_name,
'type': iot_devices[device].device_type,
} for device in iot_devices]
return json.dumps({'iot_device_status': iot_device})