3 Y�jo�@s�dZddddgZddlZddlZddlZddlmZddlmZdd lm Z dd l mZddlmZddl mZGd d�dej�Zedd��Zedd��Zdd�ZdS)z>This module contains decorators for use with and without D-Bus�FirewallDBusException�handle_exceptions�dbus_handle_exceptions�dbus_service_method�N)� DBusException)� decorator)�config)� FirewallError)�errors)�logc@seZdZdZdejjZdS)rz%s.ExceptionN)�__name__� __module__�__qualname__�__doc__r�dbusZDBUS_INTERFACEZ_dbus_error_name�rr� /usr/lib/python3.6/decorators.pyr+scOsdy |||�Stk rD}ztjtj��tj|�WYdd}~Xntk r^tj�YnXdS)zTDecorator to handle exceptions and log them. Used if not conneced to D-Bus. N)r r�debug1� traceback� format_exc�error� Exception� exception)�func�args�kwargsrrrrr/s cOs�y |||�Stk r�}zdtjt|��}|tjtjtjtjgkrRtj t|��ntj tj��tj t|��tt|��WYdd}~XnZtk r�}z |�WYdd}~Xn6tk r�}ztj�tt|��WYdd}~XnXdS)z�Decorator to handle exceptions, log and report them into D-Bus :Raises DBusException: on a firewall error code problems. N)r �get_code�strr ZALREADY_ENABLEDZNOT_ENABLEDZZONE_ALREADY_SETZALREADY_SETrZwarningrrrrrrrr)rrrr�codeZexrrrr<s cOs|jdd�tjj||�S)zAdd sender argument for D-BusZsender_keywordZsender)� setdefaultrZservice�method)rrrrrrTs)r�__all__rZdbus.servicerZdbus.exceptionsrrZfirewallrZfirewall.errorsr r Zfirewall.core.loggerrrrrrrrrr�s