ytmate/setup.py at main · IMV-IN/ytmate · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
def setup():
    #Imports
    import subprocess as sbp
    import platform as plf
    #Check wheather running windows or not
    def is_platform_windows():
        return plf.system() == "Windows"
#####################################################################################################
    #Dependencies
    libs = ["ffmpeg-python", "pathlib"]
    for lib in libs:
        sbp.run(["pip", "install", "{}".format(lib)])
    sbp.run(["python", "-m", "pip", "install", "git+https://github.com/pytube/pytube"])
    if is_platform_windows():
        sbp.run(["winget", "install", "ffmpeg"])
    else:
        #Not very comfortable with having used sudo
        sbp.run(["sudo", "apt", "update", "-y"])
        sbp.run(["sudo", "apt", "install", "ffmpeg", "-y"])
#####################################################################################################
    from pathlib import Path
    innerTube = None

    #Python Version
    index = str(plf.python_version()).find(".")

    #Platform ke according Path
    if is_platform_windows():
        # Get the user directory
        user_directory = Path.home()
        pyVersion = str(plf.python_version())[:index]+str(plf.python_version())[index+1:index+3]
        innerTube = str(user_directory)+"\AppData\Local\Programs\Python\Python{}\Lib\site-packages\pytube\innertube.py".format(pyVersion)
    else:
        pyVersion = str(plf.python_version())[:index+3]
        innerTube = "/usr/local/python/{}/lib/python{}/site-packages/pytube/innertube.py".format(plf.python_version(),pyVersion)
#####################################################################################################
    #Prepare for swapping
    file = open(innerTube, "r")
    lines = file.readlines()
    file_new = open("temp.py","w")
    index=0
    for line in lines:
        index=index+1
        if index == 223:
            file_new.write("    def __init__(self, client='ANDROID', use_oauth=False, allow_cache=True):\n")
            continue
        if index == 48:
            file_new.write("                    'clientVersion': '19.08.35',\n")
            continue
        if index == 61:
            file_new.write("                    'clientVersion': '19.08.35',\n")
            continue
        if index == 88:
            file_new.write("                    'clientVersion': '19.08.35',\n")
            continue
        if index == 102:
            file_new.write("                    'clientVersion': '19.08.35',\n")
            continue
        if index == 141:
            file_new.write("                    'clientVersion': '6.41',\n")
            continue
        if index == 128:
            file_new.write("                    'clientVersion': '6.40.52',\n")
            continue
        file_new.write(line)
    file.close()
    file_new.close()
#####################################################################################################
    #Do the swap
    fileOut = open(innerTube, "w")
    fileIn = open("temp.py","r")
    lines = fileIn.readlines()
    for line in lines:
        fileOut.write(line)
    fileIn.close()
    fileOut.close()
#####################################################################################################
    #Exit
    sbp.run(["rm", "temp.py"])
    if is_platform_windows():
        sbp.run("cls")
    else:
        sbp.run("clear")
    print("Setup Finished !!!")
#####################################################################################################

if __name__ == "__main__":
    setup()