관리-도구
편집 파일: fw_nm.cpython-36.pyc
3 @)�f � @ s d Z ddddddddgZd d lZd d lmZ yejd d � W n ek rT dZY n8X yd dlmZ dZW n e eej fk r� dZY nX d a d dl m Z d dlmZ d dlmZ d d lZdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d� Z d"d� Z d#d� Z d S )$z(Functions for NetworkManager interaction�check_nm_imported�nm_is_imported�nm_get_zone_of_connection�nm_set_zone_of_connection�nm_get_connections� nm_get_connection_of_interface�nm_get_bus_name�nm_get_dbus_interface� N)�GLib�NMz1.0F)r T)�errors)� FirewallError)�logc C s t sttjd��dS )zNCheck function to raise a MISSING_IMPORT error if the import of NM failed zgi.repository.NM = 1.0N)� _nm_importedr r ZMISSING_IMPORT� r r �/usr/lib/python3.6/fw_nm.pyr 0 s c C s t S )znReturns true if NM has been properly imported @return True if import was successful, False otherwirse )r r r r r r 6 s c C s t stjjd�a t S )z�Returns the NM client object or None if the import of NM failed @return NM.Client instance if import was successful, None otherwise N)� _nm_clientr ZClient�newr r r r � nm_get_client< s r c C s� t � t� j| �}|dkr dS |j� }|dkr2dS y |j� tjjtjjB @ rPdS W n t k rr |j � rndS Y nX |j � }|dkr�d}|S )z�Get zone of connection from NM @param connection name @return zone string setting of connection, empty string if not set, None if connection is unknown N� ) r r �get_connection_by_uuid�get_setting_connection� get_flagsr �SettingsConnectionFlags� NM_GENERATED� NM_VOLATILE�AttributeError� get_unsavedZget_zone)� connection�con� setting_con�zoner r r r E s$ c C sV t � t� j|�}|dkr dS |j� }|dkr2dS | dkr>d} |jd| � |jdd�S )zSet the zone for a connection @param zone name @param connection name @return True if zone was set, else False NFr r! T)r r r r Z set_propertyZcommit_changes)r! r r r r r r r c s c C s~ | j � |j � t� t� j� }xX|D ]P}|j� r4q&|j� }|j� }|j� }|||< x |D ]}|j� }|rZ|| |< qZW q&W dS )znGet active connections from NM @param connections return dict @param connections_name return dict N) �clearr r �get_active_connections�get_vpnZget_id�get_uuid� get_devices� get_ip_iface) Z connectionsZconnections_nameZactive_connections� active_con�nameZuuidZdevices�dev�ip_ifacer r r r x s c C s� t � g } x�t� j� D ]|}|j� r$qy&|j� }|j� tjjtjj B @ rHwW n t k rh |j � rdwY nX x&|j � D ]}|j � }|rt| j|� qtW qW | S )zGGet active interfaces from NM @returns list of interface names )r r r# r$ �get_connectionr r r r r r r r& r' �append)Zactive_interfacesr( r r* r+ r r r �nm_get_interfaces� s$ r. c C s6 g }x,t � D ]"}t|�}| t|�kr |j|� q W |S )N)r. r r r- )r! Z interfaces� interfaceZconnr r r �nm_get_interfaces_in_zone� s r0 c C s< t � x0t� j� D ]"}|j� }|dkr(q|| kr|S qW dS )zzGet device from NM which has the given IP interface @param interface name @returns NM.Device instance or None N)r r r& r' )r/ �devicer+ r r r �nm_get_device_by_ip_iface� s r2 c C sx t � t| �}|dkrdS |j� }|dkr.dS y |j� }|j� tjj@ rLdS W n tk rn |j � rjdS Y nX |j � S )z�Get connection from NM that is using the interface @param interface name @returns connection that is using interface or None N) r r2 Zget_active_connectionr, r r r r r r r% )r/ r1 r( r r r r r � s c C sR t sd S y&tj� } | jtjtj�}|j}~~ |S tk rL t j d� Y nX d S )Nz(Failed to get bus name of NetworkManager) r �dbusZ SystemBusZ get_objectr �DBUS_INTERFACEZ DBUS_PATHZbus_name� Exceptionr Zdebug2)Zbus�objr) r r r r � s c C s t sdS tjS )Nr )r r r4 r r r r r � s ) �__doc__�__all__ZgiZ gi.repositoryr Zrequire_version� ValueErrorr r � ImportError�Errorr Zfirewallr Zfirewall.errorsr Zfirewall.core.loggerr r3 r r r r r r r. r0 r2 r r r r r r r �<module> s@