a
�<�en � @ s� d dl mZ d dlmZ ddlmZ ddlmZmZm Z ddl
m Z m Z m
Z
mZmZ ddlmZ dd lmZ G d
d � d e�ZG d d
� d
e�ZG dd� de�ZG dd� de�ZG dd� de�ZdS )� )� ASSETS_PATH)�download_image� )�Node)�Line� Rectangle�UnknownElement)�Button�Text�Image� TextEntry� ButtonHover� )�Template)�Pathc sV e Zd Zd� fdd� Zdd� Zedd� �Zed�d d
�Ze d�d d �Z
d
d� Z � Z S )�Framer c s� t � �|� � �� \� _� _� �� � _i � _|� _|� _ |t
d|� � � _ � j j ddd� � j j ddd� � fdd�� j
D �� _d S )N�frameT)�parents�exist_okc s g | ]}t |�jr� �|��qS � )r Zvisible�create_element)�.0�child��selfr �I/Users/pranavzagade/PycharmProjects/LuciaVision/tkdesigner/figma/frame.py�
<listcomp> s
�z"Frame.__init__.<locals>.<listcomp>)�super�__init__�size�width�height�colorZbg_color�counter�
figma_file� output_pathr � assets_path�mkdir�children�elements)r �noder$ r% Z
frameCount�� __class__r r r
s
�zFrame.__init__c C s� |d � � �� }|d � � �� }td|� d|� d�� |dkr�| j�td�d | jt< |d }| j�|�}| jd
| jt � d � }t ||� |�
| j�}t|| || jt � d �S |d
k�r | j�t d�d | jt < |d }| j�|�}| jd| jt � d � }t ||� |�
| j�}t || |�S |dv �r�| j�t d�d | jt < |d }| j�|�}| jd| jt � d � }t ||� |�
| j�}t || || jt � d �S |dk�r| j�t
d�d | jt
< |d }| j�|�}| jd| jt
� d � }t ||� |�
| j�}t
|| || jt
� d �S |dk�s,|dk�r6t|| �S |dk�sJ|dk�rTt|| �S |dk�rht|| �S td|� d�� t|| �S d S )N�name�typezCreating Element { name: z, type: z }�buttonr r �idZbutton_z.png)Zid_Z buttonhoverZ
button_hover_)ZtextboxZtextareaZentry_�imageZimage_� rectangle�line�textzElement with the name: `z9` cannot be parsed. Would be displayed as Black Rectangle)�strip�lower�printr# �getr r$ � get_imager& r � relative_tor
r r r r r
r )r �elementZ element_nameZ element_type�item_idZ image_urlZ
image_pathr r r r # s| ��� �
�
�
�
�
�
�
�zFrame.create_elementc C s | j �d�S )Nr( �r* r8 r r r r r( s s zFrame.children��returnc s` zF| j d d d � � fdd�dD �^}}}}d|d�|d�|d��W S tyZ Y d S 0 d
S ) z4Returns HEX form of element RGB color (str)
Zfillsr r" c s g | ]}t � �|d �d ��qS )r � )�intr8 )r �i�r" r r r } � z Frame.color.<locals>.<listcomp>Zrgba�#Z02Xz#FFFFFFN)r* � Exception)r �r�g�b�_r rC r r" x s z Frame.colorc C s* | j d }|d }|d }t|�t|�fS )zCReturns element dimensions as width (int) and height (int)
ZabsoluteBoundingBoxr r! )r* rA )r �bboxr r! r r r r � s
z
Frame.sizec C s t |�}|j| | j| jd�S )N)�windowr) r& )r Zrenderr) r& )r �template�tr r r �to_code� s
�z
Frame.to_code)r )
�__name__�
__module__� __qualname__r r �propertyr( �strr" �tupler rO �
__classcell__r r r+ r r s P
r c s e Zd Z� fdd�Z� ZS )�Groupc s t � �|� d S �N�r r �r r* r+ r r r � s zGroup.__init__�rP rQ rR r rV r r r+ r rW � s rW c s e Zd Z� fdd�Z� ZS )� Componentc s t � �|� d S rX rY rZ r+ r r r � s zComponent.__init__r[ r r r+ r r\ � s r\ c s e Zd Z� fdd�Z� ZS )� ComponentSetc s t � �|� d S rX rY rZ r+ r r r � s zComponentSet.__init__r[ r r r+ r r] � s r] c s. e Zd Z� fdd�Zeed�dd��Z� ZS )�Instancec s t � �|� d S rX rY rZ r+ r r r � s zInstance.__init__r> c C s | j �d� d S )NZ componentIdr= r r r r � component_id� s zInstance.component_id)rP rQ rR r rS rT r_ rV r r r+ r r^ � s r^ N)� constantsr �utilsr r* r Zvector_elementsr r r Zcustom_elementsr r
r r r
Zjinja2r �pathlibr r rW r\ r] r^ r r r r �<module> s