B
    cd                 @   s   d dl Z d dlZd dlZd dlZd dlmZ ee  d dlZd dlZd dl	Z	d dl
Z
d dlZd dlmZmZ d dlZd dl	Z	d dlZdZefddZdd Zd	d
 Zdd Zdd Zdd Zdd Zdd ZdS )    N)reload)datetime	timedeltaz.5522894419:AAE9CeONlV2lfmUGqbXGZAHC-l1Z90TMuD4c             C   s   t j|d}|j|| d dS )zf
    Send a message to a telegram user or group specified on chatId
    chat_id must be a number!
    )token)chat_idtextN)telegramZBotZsendMessage)msgr   r   Zbot r
   %/home/fefo/Videos/level1/functions.pysend   s    r   c       
      C   s   t   }t| } t|}t| gdgd dgddg}t|gdgd dgddg}t||tj}t||tjd d }d| }|d | }t   }	|S )Nr         g      ?)timecv2ZimreadZcalcHistZcompareHistZHISTCMP_BHATTACHARYYAZmatchTemplateZTM_CCOEFF_NORMED)
Zimage_1Zimage_2
start_timeZfirst_image_histZsecond_image_histZimg_hist_diffZimg_template_probability_matchZimg_template_diffZcommutative_image_diffend_timer
   r
   r   compare_image   s    

r   c             C   s8   t jd|  ddd }t|d d }t|d S )Nz=ffprobe -v quiet -show_streams -select_streams v:0 -of json ""T)shellstreamsr   Zduration)
subprocesscheck_outputdecodejsonloadsfloat)filenameresultfieldsr
   r
   r   with_ffprobe)   s
    
r    c       	   	   C   s   | dd }|dd  }|d d }t|d }|t| }td|  d| d  td| d  td	d
}t|}W d Q R X |d |d< td	d}|	t
| |  d S )N.r   r   z
2m/2m_hls/z.tszecho '2m/2m_hls/z.ts'  >> result.m3u8zall_files.jsonzr+nameszw+)splitintstrosrenamesystemopenr   loadwritedumpsclose)	filerefresres2res1Zplus_onename	jsonFile6	all_namesr
   r
   r   change_name3   s    
r7   c          	   C   sV   d|  }t dd}t|}W d Q R X ||d< t dd}|t| |  d S )N@zref.jsonzr+r4   zw+)r*   r   r+   r,   r-   r.   )r   r   Z	jsonFile1Zdata_refr
   r
   r   save_changed_fileC   s    
r9   c          	      sB  t dd}t|}W d Q R X | }|dd  }|d d |d dkr`|dd  }|d d |dkrćfdd	tt|d
 t|d dD }td}dd	 |D  |d  krĈ |d  |dkr,fdd	tt|d t|d dD }td}dd	 |D  |d  kr, |d   fdd	|D d S )Nzall_files.jsonzr+r"   r   02c                s   g | ]} t | d  qS )z.ts)r&   ).0i)r3   r
   r   
<listcomp>\   s    zfix_flag.<locals>.<listcomp>r   
   z2m/2m_hls/*.tsc             S   s   g | ]}| d d qS )/rA   )r$   )r=   r/   r
   r
   r   r?   ^   s    r#   3c                s   g | ]} t | d  qS )z.ts)r&   )r=   r>   )r3   r
   r   r?   b   s       c             S   s   g | ]}| d d qS )rB   rA   )r$   )r=   r/   r
   r
   r   r?   d   s    c                s   g | ]}| kr|qS r
   r
   )r=   r/   )	new_filesr
   r   r?   h   s    )r*   r   r+   ranger%   globremove)r   flagr5   r6   r1   r2   ZliZ
files_pathr
   )rE   r3   r   fix_flagP   s*    *

*
rJ   c             C   sF   |  dd }|dkr|d S |dkr0t||S |dkrBt||S d S )N_r   1z.tsr<   rC   )r$   rJ   )r   rI   r4   r
   r
   r   get_right_tsl   s    
rM   c             C   s2   dd t  dD }dd |D }| |kr.dS dS )Nc             S   s   g | ]}|qS r
   r
   )r=   Zts_videor
   r
   r   r?      s    z$check_ts_in_temp.<locals>.<listcomp>z	temp/*.tsc             S   s   g | ]}| d d qS )rB   rA   )r$   )r=   Zvideor
   r
   r   r?      s    r   r   )rG   )r4   Z	ts_videosZ
new_videosr
   r
   r   check_ts_in_temp}   s    rN   )loggingr'   rG   sys	importlibr   coloredlogsconcurrent.futures
concurrentr   r   r   r   r   r   r   my_tokenr   r   r    r7   r9   rJ   rM   rN   r
   r
   r
   r   <module>   s(   
