U
    } el                     @   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
   %/var/www/html/new_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   s0   t jd|  ddd }t|d d }dS )Nz=ffprobe -v quiet -show_streams -select_streams v:0 -of json ""T)shellstreamsr   gGq  @)
subprocesscheck_outputdecodejsonloads)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d
d}t|}W 5 Q R X |d |d< td
d}|	t
| |  d S )N.r   r   z
2m/2m_hls/.tszecho '2m/2m_hls/z.ts'  >> result.m3u8z.ts' >> live.txtall_files.jsonr+names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_name4   s    
r:   c              	   C   sV   d|  }t dd}t|}W 5 Q R X ||d< t dd}|t| |  d S )N@zref.jsonr$   r7   r&   )r-   r   r.   r/   r0   r1   )r   r   Z	jsonFile1Zdata_refr
   r
   r   save_changed_fileF   s    
r<   c              	      sB  t dd}t|}W 5 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 )Nr#   r$   r!   r   02c                    s   g | ]} t | d  qS r"   r)   .0ir6   r
   r   
<listcomp>_   s     zfix_flag.<locals>.<listcomp>r   
   z2m/2m_hls/*.tsc                 S   s   g | ]}| d d qS /rH   r'   rC   r2   r
   r
   r   rF   a   s     r%   3c                    s   g | ]} t | d  qS r@   rA   rB   rE   r
   r   rF   e   s        c                 S   s   g | ]}| d d qS rI   rK   rL   r
   r
   r   rF   g   s     c                    s   g | ]}| kr|qS r
   r
   rL   )	new_filesr
   r   rF   k   s      )r-   r   r.   ranger(   globremove)r   flagr8   r9   r4   r5   ZliZ
files_pathr
   )rO   r6   r   fix_flagS   s*    *

*
rT   c                 C   sF   |  dd }|dkr|d S |dkr0t||S |dkrBt||S d S )N_r   1r"   r?   rM   )r'   rT   )r   rS   r7   r
   r
   r   get_right_tso   s    
rW   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
   )rC   Zts_videor
   r
   r   rF      s     z$check_ts_in_temp.<locals>.<listcomp>z	temp/*.tsc                 S   s   g | ]}| d d qS rI   rK   )rC   Zvideor
   r
   r   rF      s     r   r   )rQ   )r7   Z	ts_videosZ
new_videosr
   r
   r   check_ts_in_temp   s    rX   )loggingr*   rQ   sys	importlibr   coloredlogsconcurrent.futures
concurrentr   r   r   r   r   r   r   my_tokenr   r   r   r:   r<   rT   rW   rX   r
   r
   r
   r   <module>   s(   