62 lines
2.7 KiB
Python
62 lines
2.7 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
# Part of Odoo. See LICENSE file for full copyright and licensing details.
|
||
|
|
||
|
from cups import Connection as cups_connection
|
||
|
from re import sub
|
||
|
from threading import Lock
|
||
|
|
||
|
from odoo.addons.hw_drivers.interface import Interface
|
||
|
|
||
|
conn = cups_connection()
|
||
|
PPDs = conn.getPPDs()
|
||
|
cups_lock = Lock() # We can only make one call to Cups at a time
|
||
|
|
||
|
class PrinterInterface(Interface):
|
||
|
_loop_delay = 120
|
||
|
connection_type = 'printer'
|
||
|
printer_devices = {}
|
||
|
|
||
|
def get_devices(self):
|
||
|
discovered_devices = {}
|
||
|
with cups_lock:
|
||
|
printers = conn.getPrinters()
|
||
|
devices = conn.getDevices()
|
||
|
for printer_name, printer in printers.items():
|
||
|
path = printer.get('device-uri', False)
|
||
|
if printer_name != self.get_identifier(path):
|
||
|
printer.update({'supported': True}) # these printers are automatically supported
|
||
|
device_class = 'network'
|
||
|
if 'usb' in printer.get('device-uri'):
|
||
|
device_class = 'direct'
|
||
|
printer.update({'device-class': device_class})
|
||
|
printer.update({'device-make-and-model': printer_name}) # give name setted in Cups
|
||
|
printer.update({'device-id': ''})
|
||
|
devices.update({printer_name: printer})
|
||
|
for path, device in devices.items():
|
||
|
identifier = self.get_identifier(path)
|
||
|
device.update({'identifier': identifier})
|
||
|
device.update({'url': path})
|
||
|
device.update({'disconnect_counter': 0})
|
||
|
discovered_devices.update({identifier: device})
|
||
|
self.printer_devices.update(discovered_devices)
|
||
|
# Deal with devices which are on the list but were not found during this call of "get_devices"
|
||
|
# If they aren't detected 3 times consecutively, remove them from the list of available devices
|
||
|
for device in list(self.printer_devices):
|
||
|
if not discovered_devices.get(device):
|
||
|
disconnect_counter = self.printer_devices.get(device).get('disconnect_counter')
|
||
|
if disconnect_counter >= 2:
|
||
|
self.printer_devices.pop(device, None)
|
||
|
else:
|
||
|
self.printer_devices[device].update({'disconnect_counter': disconnect_counter + 1})
|
||
|
return dict(self.printer_devices)
|
||
|
|
||
|
def get_identifier(self, path):
|
||
|
allowed_characters = '[^a-zA-Z0-9_-]'
|
||
|
if 'uuid=' in path:
|
||
|
identifier = sub(allowed_characters, '', path.split('uuid=')[1])
|
||
|
elif 'serial=' in path:
|
||
|
identifier = sub(allowed_characters, '', path.split('serial=')[1])
|
||
|
else:
|
||
|
identifier = sub(allowed_characters, '', path)
|
||
|
return identifier
|