File: //snap/core20/2669/usr/share/subiquity/console_conf/controllers/__pycache__/identity.cpython-38.pyc
U
K��^m � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ d dlm Z m
Z
d dl m Z d dl
mZmZ d dlmZmZ e�d�Zdd � Zd
d � Zd d
� Zdd� ZdZdZdd� Zdd� Z G dd� de�Z dS )� N)�BaseController)�
host_key_info�get_ips_standalone)�SnapdConnection)�disable_console_conf� run_command)� IdentityView� LoginViewz!console_conf.controllers.identityc C s� d} z"t | d��}|�� }W 5 Q R X W n tk
r< Y dS X d}t�|�D ]:}|�d�\}}}|dkrv|dkrv dS |dkrL|} q�qL|S )z6 For a ubuntu-core system, return its version or None z/usr/lib/os-release�rN�=ZIDz ubuntu-coreZ
VERSION_ID)�open�read�FileNotFoundError�shlex�split� partition)�path�fpZcontent�version�line�key�_�value� r �9/usr/share/subiquity/console_conf/controllers/identity.py�get_core_version s r c C s t dd�} | �d��� d d S )z Check if device is managed � �/run/snapd.socketzv2/system-info�resultZmanaged)r �get�json)�conr r r � get_managed6 s
r" c C s8 zt �| �}W n tk
r$ Y dS X |j�dd�d S )Nr �,� r )�pwd�getpwnam�KeyErrorZpw_gecosr )�username�infor r r � get_realname< s
r* c C s` t dd�} | �d��� d D ]>}d|kr*q |d }d| }tj�|�r |t|�|d� S q dS ) z Get device owner, if any r r zv2/usersr r( z/home/)r( �realname�homedirN)r r r �osr �isdirr* )r! �userr( r, r r r �get_device_ownerD s
� r0 z�Ubuntu Core {version} on {first_ip} ({tty_name})
{host_key_info}
To login:
{sshcommands}
Personalize your account at https://login.ubuntu.com.
z�Ubuntu Core {version} on <no ip address> ({tty_name})
You cannot log in until the system has an IP address. (Is there
supposed to be a DHCP server running on your network?)
Personalize your account at https://login.ubuntu.com.
c C s� d}|D ]}|d||f 7 }qt �d�dd � }t� p8d}t|�dkr^| �tj|||d�� n$|d }| �tj|t� |||d�� d S )N�
z ssh %s@%s
r � �16)� sshcommands�tty_namer )r4 r r5 �first_ipr ) r- �ttynamer �len�write�login_details_tmpl_no_ip�format�login_details_tmplr )r r( �ipsr4 Zipr5 r r6 r r r �write_login_detailsi s$
�
�r> c C s� t � } t� }t|�dkr^| d kr,td� dS t�d�dd � }t� pFd}ttj||d�� dS | d kr|td�d� |��� nt
t j | d |� dS )
Nr zdevice managed without user� r2 r3 )r5 r z device managed without user @ {}z, r( )
r0 r r8 �printr- r7 r r: r; �joinr> �sys�stdout)�ownerr= r5 r r r r � write_login_details_standalone{ s"
�rE c sD e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d
� Zd d � Z� Z S )
�IdentityControllerc s t � �|� |jj| _d S �N)�super�__init__�
base_modelZidentity�model)�self�app�� __class__r r rI � s zIdentityController.__init__c C st | j �t| j| �� t� rpt� }|rh| j�|� tj� |d d�}t
dd|g�}|j � dd��
� | jj_| �� d S )Nr, z.ssh/authorized_keysz
ssh-keygenz-lf�
r )�ui�set_bodyr rK r" r0 �add_userr- r rA r rC �replace�
splitlinesr/ Z fingerprints�login)rL Z device_ownerZkey_file�cpr r r �start_ui� s ��zIdentityController.start_uic
C s | j jr$||d�}| j�|� d}n�| jj�� tdddd|g�}|jdkrpt | j
j t �rl| j
j �
d|j� d S t�|j�}||d d�}tjd
d d � d
}| j�|� g }| jjj}|�� D ]}|�|j� q�t|d��} t| |d |� W 5 Q R X | �� d S )N)r+ r( z .subiquity/login-details.txtZsnapz create-userz--sudoerz--jsonr zCreating user failed:r( z/run/console-confT)�exist_okz#/run/console-conf/login-details.txt�w) �opts�dry_runrK rS rM Z
urwid_loopZ draw_screenr �
returncode�
isinstancerQ Zbodyr Zsnap_create_user_failed�stderrr �loadsrC r- �makedirsrJ �network�get_all_netdevs�extendZactual_global_ip_addressesr r> rV )
rL Zemailr Zlogin_details_pathrW �datar= � net_modelZdevr r r r �
identity_done� s> � �
� �
z IdentityController.identity_donec C s | j jd kr| j�� d S rG )rK r/ rM Z prev_screen�rL r r r �cancel� s zIdentityController.cancelc C sL d}| j �|� | jjj}|�� }t| j| j| |�}d|j _
| j � |� d S )NzConfiguration Completer? ) rQ Z
set_headerrM rJ rb rc r r[ rK Z_wZfocus_positionrR )rL �titlerf ZifacesZ
login_viewr r r rV � s
zIdentityController.loginc C s | j jst� | j�� d S rG )r[ r\ r rM �exitrh r r r �
login_done� s z IdentityController.login_done)
�__name__�
__module__� __qualname__rI rX rg ri rV rl �
__classcell__r r rN r rF � s
" rF ) r Zloggingr- r% r rB Zsubiquitycore.controllerr Zsubiquitycore.sshr r Zsubiquitycore.snapdr Zsubiquitycore.utilsr r Zconsole_conf.ui.viewsr r Z getLogger�logr r" r* r0 r< r: r> rE rF r r r r �<module> s(