
    %
i5                         d dl Z d dlZd dlZd dlT g ZddZd Zd Zd Zd Z	 e	              e              e             dS )	    N)*c                    d|z  }	 t          j        | d          }t          |          dk    rd S |D ]}	 |j        t          k    r3t          |dz   |j                   t                              |           n|j        t          k    rt          |dz   |j                   nrt          |dz   |j        z   d           t          j
        t          t          d|          }t                       t          ||dz              t          j        |           # t           j        $ r%}t          |dz   |j        z              Y d }~d }~ww xY wH)	N    r   zHave share with name:z Have generic resource with name:zEnumerating )endz"Couldn't enumerate this resource: )	win32wnetWNetEnumResourcelendwDisplayTypeRESOURCEDISPLAYTYPE_SHAREprintlpRemoteNamepossible_sharesappendRESOURCEDISPLAYTYPE_GENERICWNetOpenEnumRESOURCE_GLOBALNETRESOURCETYPE_ANY_doDumpHandleWNetCloseEnumerrorstrerror)handlelevelindentitemsitemkdetailss          C:\Users\Dell Inspiron 16\Desktop\tws\AgrotaPowerBi\back-agrota-powerbi\mcp-client-agrota\venv\Lib\site-packages\win32/Demos/win32wnet/testwnet.pyr   r   
   s   5[FX*6155u::??E 	X 	XDX%)BBB&#::D<MNNN#**40000'+FFF!CCTEV   
 &>1D4EE3OOOO!.*,<a A GGG!!UQY///+   ? X X XfCCgFVVWWWWWWWWX1Xs   CDE
%EE
c                      t          d           t          j        t          t          dd           } 	 t          |            |                                  n# |                                  w xY wt          d           d S )NzEEnumerating all resources on the network - this may take some time...r   zFinished dumping all resources.)r   r   r   r   r   r   Close)r   s    r    TestOpenEnumr#   (   sr    	
QRRR#$68H!TRRFf	
+,,,,,s   A A,c                     d t          j                                        d          D             } t          j        t
          t          dd           }	 	 t          j        |d          }t          |          dk    rn"d |D             }| 	                    |           K	 |
                                 n# |
                                 w xY wdD ]
}|| vr|c S t          d          )Nc                 F    g | ]}||d                                           S r   )lower).0xs     r    
<listcomp>z)findUnusedDriveLetter.<locals>.<listcomp>4   s9       q	!

       r   r   c                 Z    g | ](}|j         	|j         d                                          )S r&   )lpLocalNamer'   )r(   is     r    r*   z)findUnusedDriveLetter.<locals>.<listcomp>=   s2    MMMq}MAM!$**,,MMMr+   defghijklmnopqrstuvwxyzzAll drive mappings are taken?)win32apiGetLogicalDriveStringssplitr   r   RESOURCE_REMEMBEREDRESOURCETYPE_DISKr	   r
   extendr"   RuntimeError)existingr   r   xtramaybes        r    findUnusedDriveLetterr;   3   s    &=??EEdKK  H #$79JAtTTF	".vq99E5zzQMMeMMMDOOD!!!	"  	*    LLL !
6
7
77s   AB5 5Cc            	      z   t          t                    dk    rt          d           d S t                      dz   } t          D ]}t          d| d|j                   	 t          j        |j        | |j                   n3# t
          j        $ r!}t          d|j	        z              Y d }~gd }~ww xY w	 t          j                            | dz   t          j        | dz             d                   }	 t          d                    |t          j        |                               n6# t
          j        $ r$}t          d	| d
|j	                    Y d }~nd }~ww xY wt          dt          j        |                      t          j        | dd           n# t          j        | dd           w xY wt          j                    }|j        |_        | |_        |j        |_        t          j        |           t          j        | dd           t          j        d|           t          j        | dd            d S d S )Nr   z0Couldn't find any potential shares to connect to:zAttempting connection oftozCouldn't connect: \zUniversal name of '{}' is '{}'z Couldn't get universal name of 'z': z User name for this connection is)r
   r   r   r;   r   r   WNetAddConnection2dwTyper   r   ospathjoinlistdirformatWNetGetUniversalNameWNetGetUserWNetCancelConnection2NETRESOURCEr.   WNetAddConnection3)	localNamesharer   fnamenrs        r    TestConnectionrP   G   sn   
?q  @AAA%''#-I  # #()T5;MNNN	(y%BTUUUU 	 	 	&)99:::HHHH		=GLLT!12:i$>N3O3OPQ3RSSEW4;;y=eDD    
 ? W W WUUU7CSUUVVVVVVVVW4i6KI6V6VWWW+Iq!<<<<I+Iq!<<<< "$$L	",$R((('	1a888 	$Q+++'	1a888 	G# #sN    A>>B.B))B.2>F15D'&F'E6EFE%FF.c                      t          j                    } t          d|            | t          j        d           k    rt          d          d S )NzCurrent global user is z"Default value didn't seem to work!)r   rH   r   r7   )us    r    TestGetUserrS   r   sT    A	
)A
)
)***I!$''''?@@@ ('r+   r&   )
rB   r1   r   winnetwkr   r   r#   r;   rP   rS    r+   r    <module>rV      s    				         X X X X<- - -8 8 8(( ( (VA A A        r+   