Current File : //lib/python2.7/site-packages/salt/modules/rdp.pyc
�
���^c @@ s6 d Z d d l m Z m Z m Z d d l Z d d l Z d d l Z d d l m
Z
y& d d l m Z
d d l Z e Z Wn e k
r� e Z n Xe j e � Z d � Z d � Z d � Z d � Z d
� Z d � Z e
e � e d � � Z e
e � d
� � Z e
e � d � � Z e
e � d � � Z d S( u'
Manage RDP Service on Windows servers
i ( t absolute_importt unicode_literalst print_functionN( t depends( t errorc C@ s t j j j � r d St d f S( u'
Only works on Windows systems
u rdpu Module only works on Windows.( t saltt utilst platformt
is_windowst False( ( ( s4 /usr/lib/python2.7/site-packages/salt/modules/rdp.pyt __virtual__ s c C@ s9 t j d | � } | s" t d f St | j d � � Sd S( uP
return from the input string the return code of the powershell command
u ReturnValue\s*: (\d*)u' Could not parse PowerShell return code.i N( t ret searchR t intt group( t stringt regex( ( s4 /usr/lib/python2.7/site-packages/salt/modules/rdp.pyt _parse_return_code_powershell" s
c C@ s, d } t d d j | | � d d d t �S( u�
Create a Win32_TerminalServiceSetting WMI Object as $RDP and execute the
command cmd returns the STDOUT of the command
u� $RDP = Get-WmiObject -Class Win32_TerminalServiceSetting -Namespace root\CIMV2\TerminalServices -Computer . -Authentication 6 -ErrorAction Stopu cmd.runu {0} ; {1}t shellu
powershellt python_shell( t __salt__t formatt True( t cmdt rdp( ( s4 /usr/lib/python2.7/site-packages/salt/modules/rdp.pyt _psrdp. s c C@ s t t d � � d k S( uw
Enable RDP the service on the server
CLI Example:
.. code-block:: bash
salt '*' rdp.enable
u $RDP.SetAllowTsConnections(1,1)i ( R R ( ( ( s4 /usr/lib/python2.7/site-packages/salt/modules/rdp.pyt enable: s c C@ s t t d � � d k S( uy
Disable RDP the service on the server
CLI Example:
.. code-block:: bash
salt '*' rdp.disable
u $RDP.SetAllowTsConnections(0,1)i ( R R ( ( ( s4 /usr/lib/python2.7/site-packages/salt/modules/rdp.pyt disableI s c C@ s" t t d � j � � } | d k S( uw
Show if rdp is enabled on the server
CLI Example:
.. code-block:: bash
salt '*' rdp.status
u echo $RDP.AllowTSConnectionsi ( R
R t strip( t out( ( s4 /usr/lib/python2.7/site-packages/salt/modules/rdp.pyt statusX s c
C@ s� t � } t j } i d t j 6d t j 6d t j 6} i
d t j 6d t j 6d t j 6d t j 6d t j
6d t j 6d
t j 6d t j
6d t j 6d
t j 6} x� t j | � D]� } t j | | d t j � p� d } t j | | d t j � } t j | | d t j � } | j | d � } | j | d � }
| d pAd } i |
d 6| d 6| d d 6| d 6| d 6} | r�| r�| j | � q�q� | j | � q� W| s�t j d � n t | d d � �S( u9
List information about the sessions.
.. versionadded:: 2016.11.0
:param logged_in_users_only: If True, only return sessions with users logged in.
:return: A list containing dictionaries of session information.
CLI Example:
.. code-block:: bash
salt '*' rdp.list_sessions
u consoleu citrixu rdpu activeu connectedu
connect_queryu shadowu disconnectedu idleu listenu resetu downu initu SessionIdu unknownu WinStationNameu Disconnectedu connection_statusu protocolu
session_idu stationu useru No sessions found.t keyc S@ s | d S( Nu
session_id( ( t k( ( s4 /usr/lib/python2.7/site-packages/salt/modules/rdp.pyt <lambda>� s N( t listt win32tst WTS_CURRENT_SERVER_HANDLEt WTS_PROTOCOL_TYPE_CONSOLEt WTS_PROTOCOL_TYPE_ICAt WTS_PROTOCOL_TYPE_RDPt WTSActivet WTSConnectedt WTSConnectQueryt WTSShadowt WTSDisconnectedt WTSIdlet WTSListent WTSResett WTSDownt WTSInitt WTSEnumerateSessionst WTSQuerySessionInformationt WTSUserNamet Nonet WTSClientProtocolTypet WTSConnectStatet gett appendt _LOGt warningt sorted(
t logged_in_users_onlyt rett servert protocolst statusest sessiont usert protocol_idt status_idt protocolt connection_statust stationt connection_info( ( s4 /usr/lib/python2.7/site-packages/salt/modules/rdp.pyt
list_sessionsg s<
c C@ sk t � } t � } g | D] } | d | k r | ^ q } | rN | d } n | sg t j d | � n | S( u2
Get information about a session.
.. versionadded:: 2016.11.0
:param session_id: The numeric Id of the session.
:return: A dictionary of session information.
CLI Example:
.. code-block:: bash
salt '*' rdp.get_session session_id
salt '*' rdp.get_session 99
u
session_idi u No session found for id: %s( t dictRJ R: R; ( t
session_idR> t sessionst itemRB ( ( s4 /usr/lib/python2.7/site-packages/salt/modules/rdp.pyt get_session� s )
c C@ sE y t j t j | t � Wn$ t k
r@ } t j d | � t SXt S( uI
Disconnect a session.
.. versionadded:: 2016.11.0
:param session_id: The numeric Id of the session.
:return: A boolean representing whether the disconnect succeeded.
CLI Example:
.. code-block:: bash
salt '*' rdp.disconnect_session session_id
salt '*' rdp.disconnect_session 99
u& Error calling WTSDisconnectSession: %s( R# t WTSDisconnectSessionR$ R t
PyWinErrorR: R R ( RL R ( ( s4 /usr/lib/python2.7/site-packages/salt/modules/rdp.pyt disconnect_session� s c C@ sE y t j t j | t � Wn$ t k
r@ } t j d | � t SXt S( uI
Initiate the logoff of a session.
.. versionadded:: 2016.11.0
:param session_id: The numeric Id of the session.
:return: A boolean representing whether the logoff succeeded.
CLI Example:
.. code-block:: bash
salt '*' rdp.logoff_session session_id
salt '*' rdp.logoff_session 99
u" Error calling WTSLogoffSession: %s( R# t WTSLogoffSessionR$ R RQ R: R R ( RL R ( ( s4 /usr/lib/python2.7/site-packages/salt/modules/rdp.pyt logoff_session� s ( t __doc__t
__future__R R R t loggingR t salt.utils.platformR t salt.utils.decoratorsR t
pywintypesR RQ R# R t _HAS_WIN32TS_DEPENDENCIESt ImportErrorR t getLoggert __name__R: R
R R R R R RJ RO RR RT ( ( ( s4 /usr/lib/python2.7/site-packages/salt/modules/rdp.pyt <module> s.
2
Mini Shell