File: //usr/lib/python3/dist-packages/Quirks/__pycache__/quirkapplier.cpython-310.pyc
o
��f� � @ sf d dl m Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZ d dl
Z G dd� d�Z dd� Z dS )� )�globNc @ sh e Zd Zddd�Zdd� Zdd� Zdd � Zd
d � Zdd
d�Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dS ) � QuirkChecker�/usr/share/jockey/quirksc C s. || _ || _g | _| �� | �� | _d| _d S )Nz/usr/share/X11/xorg.conf.d)�_handler� quirks_path�_quirks�get_quirks_from_path�get_system_info� _system_info�_xorg_conf_d_path)�self�handler�path� r �5/usr/lib/python3/dist-packages/Quirks/quirkapplier.py�__init__! s
zQuirkChecker.__init__c C s| g | _ tj�| j�r3ttj�| jd��D ]}tj�|�r/t� d| � | �
|�}| j |7 _ q| j S t� d| j � | j S )�5check all the files in a directory looking for quirks�*z
Parsing %sz%s does not exist) r �osr �isdirr r �join�isfile�logging�debug�get_quirks_from_file)r �f�quirksr r r r ) s
��z!QuirkChecker.get_quirks_from_pathc C s t j�|�}|�� S )r )�Quirks� quirkreader� ReadQuirk�
get_quirks)r �
quirk_filer r r r 7 s z!QuirkChecker.get_quirks_from_filec C s t j�� }|�� S )z Get system info for the quirk)r � quirkinfo� QuirkInfo� get_dmi_info)r �
quirk_infor r r r = s
z QuirkChecker.get_system_infoc C s� d}|j �� D ]2}|j | D ]*}| j�|�r8| j�|�|kr8t|j | �dkr8t�d| j�|�|f � dS qqt�d� |S )z See if tags match system infoT� zFailure to match %s with %sF�Success)�
match_tags�keysr
�get�lenr r )r �quirk�result�tag�valr r r � matches_tagsB s ���
zQuirkChecker.matches_tagsTc C s� | j D ]?}| j�� dd� |jD �v rBt�d|j � | �|�r=|r/t�d|j � | � |� qt�d|j � | �
|� qt�d� qdS )z)Process quirks and do something with themc S s g | ]}|� � �� �qS r )�lower�strip)�.0�xr r r �
<listcomp>R s z.QuirkChecker._check_quirks.<locals>.<listcomp>zProcessing quirk %szApplying quirk %szUnapplying quirk %szQuirk doesn't matchN) r r r1 r
r r �idr0 �info� _apply_quirk�_unapply_quirk)r �enabler, r r r �
_check_quirksO s
��zQuirkChecker._check_quirksc C � | � d� dS )z Enable all quirks for a handlerTN�r; �r r r r �
enable_quirks_ � zQuirkChecker.enable_quirksc C r< )z Disable all quirks for a handlerFNr= r> r r r �disable_quirksc r@ zQuirkChecker.disable_quirksc C s d| j | j|j�� �dd�f S )z$Return the path to the X config filez%s/10-%s-%s.conf� �-)r r r6 r1 �replace)r r, r r r �_get_destination_pathg s �z"QuirkChecker._get_destination_pathc C s� | � |�}tjddd�}|�|j� |�� tj�|j �}t
� |j � t
�|j � zt
� d| � |�|� W dS tyG t
�d� Y dS w )z!Get the xorg snippet and apply it�wF)�mode�deletez Creating %szError during write()T)rE �tempfile�NamedTemporaryFile�write� x_snippet�close�xkit�
xorgparser�Parser�namer r �
globaldictr �unlink�IOError� exception)r r, � destination�tmp_file�tmp_xkitr r r r8 l s
�
�zQuirkChecker._apply_quirkc C sL | � |�}t�d| � zt�|� W dS ttfy% t�d� Y dS w )z%Remove the file with the xorg snippetzRemoving %s ...zCannot unlink destinationFT)rE r r r rS �OSErrorrT rU )r r, rV r r r r9 � s
�
�zQuirkChecker._unapply_quirkN)r )T)�__name__�
__module__� __qualname__r r r r r0 r; r? rA rE r8 r9 r r r r r s
r c C s0 t ddd�} | �� | �� ttj�t�� dS )N�nvidiaz /home/alberto/oem/jockey/quirks)r r )r r? rA �printr r �abspath�__file__)�ar r r �main� s
rb )
r r �sysrI r � xkit.xutilsrN �xkit.xorgparser�Quirks.quirkreaderr �Quirks.quirkinfor rb r r r r �<module> s n