filesManager.py

#!/usr/bin/env python3



from datetime import datetime
from genericpath import isfile
import os
import time

from colored import *
from pathlib import Path

from Actions import ActionDllFile
from config import Channel, DaysToKeep, SavePath, UtcTOfset, url
from xmlreq import DllXmlReq

starttime = ""
endtime = ""
name = ""


def parsertspuri(rtspuri):
    # starttime=20220311T073025Z&endtime=20220311T073547Z&
    # name=08000000014000713&size=27635176
    #rtspuri = rtspuri.replace('rtsp://' + url + '/Streaming/tracks/" + Channel + "/?', '')
    Delrtspurl = rtspuri.split("/Streaming/tracks/" + Channel + "/?")
    tmpdic = Delrtspurl[1].split("&")
    starttime = tmpdic[0].replace('starttime=', '').replace('T', '-') + "-"
    endtime = tmpdic[1].replace('endtime=', '').replace('T', '-') + "-"

    #if starttime == "20220414-014652Z-":
    #    print(CBLINK2 + CREDBG + "debug : " +
    #          starttime + "   " + endtime + CEND)

    size = int(tmpdic[3].replace('size=', ''))
    name = tmpdic[2].replace('name=', '') + ".mp4"
    #filename = starttime + endtime + name
    print(CGREEN + "Downloading : " + CEND + name)
    print(CGREEN + "size = " + CEND + str(size/1000000) + " M")
    FsName = PrepareDir(starttime[0:8]) + \
        timestamp(starttime[0:15]) + "-" + timestamp(endtime[0:15]) + "-" + name
    #ValidSize(FsName, size)
    PartialDll(FsName, name, size)
    return FsName


def donloadfs(rtspuris):
    print(CBLINK2 + CRED + str(len(rtspuris)) + CEND + " file(s) to Download")
    for rtspuri in rtspuris:
        #rtspuri = GetLocalip(rtspuri)
        print(CGREENBG + "rtspuri= " + rtspuri + CEND)
        DestFs = parsertspuri(rtspuri)
        if not os.path.exists(DestFs):
            ActionDllFile('/ISAPI/ContentMgmt/download?playbackURI=' +
                          rtspuri, DestFs, DllXmlReq(rtspuri))
            print(CBLUE + DestFs + CRED + " Downloaded" + CEND)
    DelOldestDir()


def GetLocalip(rtspuri):
    Delrtspurl = rtspuri.split("/Streaming/tracks/" + Channel + "/?")
    Delrtspurl[0] = 'rtsp://' + url
    return Delrtspurl[0] + "/Streaming/tracks/" + Channel + "/?" + Delrtspurl[1]


def PrepareDir(Day):
    path = SavePath + Day + os.path.sep
    # Check whether the specified path exists or not
    isExist = os.path.exists(path)
    if not isExist:
        # Create a new directory because it does not exist
        os.makedirs(path)

    return path


def ValidSize(Fs, size):
    if os.path.exists(Fs):
        b = os.path.getsize(Fs)
        if b < size:
            try:
                os.remove(Fs)
                print(CRED + "Existing file have not the good size, deleting it" + CEND)
            except:
                print(CRED + "Problem to delete file" + CEND)
        else:
            print(CRED + "Existing file have the good size, pass" + CEND)


def PartialDll(FsName, name, size):
    dir_path = os.path.dirname(os.path.realpath(FsName))
    Files = [f for f in os.listdir(dir_path) if isfile(os.path.join(dir_path, f))]
    
    for Fs in Files:
     if Fs.endswith(name):
         ValidSize(os.path.join(dir_path, Fs), size)


def DelOldestDir():

    # Get list of all dir only in the given directory
    list_of_dir = os.listdir(SavePath)
    # Sort list of dir based on last modification time in ascending order
    list_of_dir = sorted(list_of_dir)
    # Iterate over sorted list of dir and print file path
    # along with last modification time of dir
    """ 
    for dir_name in list_of_dir:
        dir_path = os.path.join(SavePath, dir_name)
        timestamp_str = time.strftime('%m/%d/%Y',
                                time.gmtime(os.path.getmtime(dir_path))) 
        print(timestamp_str, ' -->', dir_name)
    """

    lendirs = len(list_of_dir)
    if lendirs > DaysToKeep:
        for ExtraDir in list_of_dir[:lendirs - DaysToKeep - 1]:
            remove_path(Path(os.path.join(SavePath, ExtraDir)))


def remove_path(path: Path):
    if path.is_file() or path.is_symlink():
        path.unlink()
        return
    for p in path.iterdir():
        remove_path(p)
    path.rmdir()
    
def timestamp(date='20180501'):
    
    UnixTstamp = int(time.mktime(
        datetime.strptime( date, "%Y%m%d-%H%M%S" ).timetuple()
    ))
    
    UnixTstampAdd = UnixTstamp
    
    if UtcTOfset == "yes": 
        UnixTstampAdd = UnixTstamp + int(time.strftime('%z')) * 6 * 6
    
    Result = datetime.fromtimestamp(UnixTstampAdd).strftime('%Y%m%d-%H%M%S')    
    return Result
    
def UTCofssetAdd(tcheck):
    tmpdic = tcheck.split("-")
    Ttcheck = time(tmpdic[1])
     
    if UtcTOfset:
        UtcTOf = time.strftime('%z')
        HourN = UtcTOf[1:2]
        if UtcTOf.startswith("+"):
            Ttcheck = Ttcheck + HourN
        else:
            Ttcheck = Ttcheck - HourN
    return tmpdic[0] + "-" + tmpdic[1]

def Daylight():
    if time.daylight > 0:
        return time.altzone
    else:
        return time.timezone