�
��f� � �j � S SK J r S SKrS SKrS SKrS SKrS SKrS SKrS SKr S SK
r " S S5 r S r g)� )�globNc �^ � \ rS rSrSS jrS rS rS rS rSS jr S r
S r S
r S r
S rS
rg)� QuirkChecker� c � � Xl X l / U l U R 5 U R 5 U l SU l g )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�paths �5/usr/lib/python3/dist-packages/Quirks/quirkapplier.py�__init__�QuirkChecker.__init__! s9 � � �
� ���� � �!�!�#� � 0� 0� 2���!=�� � c � � / U l [ R R U R 5 ( a� [ [ R R
U R S5 5 Hh n[ R R U5 ( d M) [ R " SU- 5 U R U5 nU =R U-
sl Mj U R $ [ R " SU R - 5 U R $ )�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�quirkss r r �!QuirkChecker.get_quirks_from_path) s� � ��� �
�7�7�=�=��)�)� *� *��"�'�'�,�,�t�'7�'7��=�>���7�7�>�>�!�$�$��M�M�,��"2�3� !� 6� 6�q� 9�F��L�L�F�*�L� ?� �|�|��
�M�M�-��0@�0@�@� A��|�|�r c �` � [ R R U5 nUR 5 $ )r )�Quirks� quirkreader� ReadQuirk�
get_quirks)r �
quirk_files r r �!QuirkChecker.get_quirks_from_file7 s) � � �'�'�1�1�*�=�
��$�$�&�&r c �^ � [ R R 5 nUR 5 $ )z Get system info for the quirk)r$ � quirkinfo� QuirkInfo� get_dmi_info)r �
quirk_infos r r � QuirkChecker.get_system_info= s% � ��%�%�/�/�1�
��&�&�(�(r c �� � SnUR R 5 H� nUR U H� nU R R U5 ( d M% U R R U5 U:w d MF [ UR U 5 S:: d Md [
R " SU R R U5 < SU< 35 g M� [
R " S5 U$ )z See if tags match system infoT� zFailure to match z with F�Success)�
match_tags�keysr
�get�lenr r )r �quirk�result�tag�vals r � matches_tags�QuirkChecker.matches_tagsB s� � ����#�#�(�(�*�C� �'�'��,���%�%�)�)�#�.�.�4�3D�3D�3H�3H��3M�QT�3T���(�(��-�.�!�3��M�M�#'�#4�#4�#8�#8��#=�s�#D� E� � -� +� �
�
�i� ��
r c �j � U R GH nU R R 5 UR Vs/ s H o3R 5 R 5 PM" sn; d MZ [
R " SUR - 5 U R U5 ( at U( a6 [
R " SUR - 5 U R U5 M� [
R " SUR - 5 U R U5 GM [
R " S5 GM gs snf )z)Process quirks and do something with themzProcessing quirk %szApplying quirk %szUnapplying quirk %szQuirk doesn't matchN) r
r �lowerr �stripr r �idr; �info� _apply_quirk�_unapply_quirk)r �enabler7 �xs r �
_check_quirks�QuirkChecker._check_quirksO s� � ��\�\�E��}�}�"�"�$�E�M�M�(R�M�q���� � �):�M�(R�R��
�
� 3�e�h�h� >�?��$�$�U�+�+� � � � �%8�5�8�8�%C�D� �)�)�%�0� � � �%:�U�X�X�%E�F� �+�+�E�2��M�M�"8�9� "��(Rs �'D0c �& � U R S5 g)z Enable all quirks for a handlerTN�rF �r s r �
enable_quirks�QuirkChecker.enable_quirks_ s � � ���4� r c �& � U R S5 g)z Disable all quirks for a handlerFNrI rJ s r �disable_quirks�QuirkChecker.disable_quirksc s � � ���5�!r c � � U R < SU R < SUR R 5 R SS5 < S3$ )z$Return the path to the X config filez/10-�-� z.conf)r r r@ r> �replace)r r7 s r �_get_destination_path�"QuirkChecker._get_destination_pathg s8 � �%)�%;�%;��
�
�u�x�x�~�~� /� 7� 7��S� A�C� Cr c � � U R U5 n[ R " SSS9nUR UR 5 UR 5 [ R R UR 5 n[ R " UR 5 [ R " UR 5 [ R " SU- 5 UR U5 g! [ a [ R " S5 gf = f)z!Get the xorg snippet and apply it�wF)�mode�deletez Creating %szError during write()T)rT �tempfile�NamedTemporaryFile�write� x_snippet�close�xkit�
xorgparser�Parser�namer r �
globaldictr �unlink�IOError� exception)r r7 � destination�tmp_file�tmp_xkits r rB �QuirkChecker._apply_quirkl s� � � �0�0��7� ��.�.�C��F�����u���'������?�?�)�)�(�-�-�8���
�
�h�)�)�*�
� � �(�-�-� � � �M�M�-�+�5� 6� �N�N�;� '� �� � � � � � 4� 5�� �s �<*C' �' D
� D
c �� � U R U5 n[ R " SU- 5 [ R " U5 g! [
[ 4 a [ R " S5 gf = f)z%Remove the file with the xorg snippetzRemoving %s ...zCannot unlink destinationFT)rT r r r rd �OSErrorre rf )r r7 rg s r rC �QuirkChecker._unapply_quirk� sd � � �0�0��7� ��
�
�'�+�5�6� � �I�I�k� "� �� ��!� � � � � 9� :�� �s �A �&A,�+A,)r r
r
r r N)z/usr/share/jockey/quirks)T)�__name__�
__module__� __qualname__�__firstlineno__r r r r r; rF rK rN rT rB rC �__static_attributes__� r r r r s: � �>� � '� )�
�:� !�"�C�
�( r r c � � [ SSS9n U R 5 U R 5 [ [ R
R
[ 5 5 g)N�nvidiaz /home/alberto/oem/jockey/quirks)r r )r rK rN �printr r �abspath�__file__)�as r �mainrz � s? � ��X�$E�F�A��O�O������ �"�'�'�/�/�8�
%�&�$
r )
r r �sysrZ r � xkit.xutilsr_ �xkit.xorgparser�Quirks.quirkreaderr$ �Quirks.quirkinfor rz rs r r �<module>r� s4 ��( � �
� � � � � � �k� k�\
r