ovosimpatico/CLI-TikTok
1
0
Fork 0
mirror of https://github.com/ovosimpatico/CLI-TikTok.git synced 2026-01-15 11:52:52 -03:00
Code Issues Packages Projects Releases Wiki Activity

PEP8 Compliance

Browse Source
This commit is contained in:
nanometer5088
2023-01-28 21:27:46 -03:00
parent cb59a11cfa
commit b586be171e
7 changed files with 172 additions and 113 deletions
Download Patch File Download Diff File Expand all files Collapse all files

76
main.py
View File

@@ -1,35 +1,46 @@
# Introduction and pre-test
from src.init import init
import sys
from src.init import init
a = init()
if a == -1:
print("Dependencies installed successfully.\nOpen the program again\n")
sys.exit()
# Import libraries and required functions after sucessful pre-test
from src.functions import listas
from src.streaming import playbackrandom, playback
import os
import subprocess
from src.byuser import getLinks, streamuser
from src.downloader import downloadtiktoks
from src.byuser import streamuser, getLinks
import os, subprocess
from src.functions import listas
from src.streaming import playback, playbackrandom
# Needlessly big code to simply prompt the user which action they want to do
try:
question = int(input("""Do you want to download or watch tiktoks?
question = int(
input(
"""Do you want to download or watch tiktoks?
(1) Download
(2) Watch
"""))
"""
)
)
os.system("cls || clear")
## Download
if question == 1:
downloadquestion = int(input("""Do you want to download your liked videos or a creator?
downloadquestion = int(
input(
"""Do you want to download your liked videos or a creator?
(1) Liked Videos
(2) Creator
"""))
"""
)
)
os.system("cls || clear")
## Download liked videos
@@ -40,9 +51,13 @@ try:
## Download creator
if downloadquestion == 2:
print('Due to specific limitations of the current data method, downloading by creator will only get the latest 30 videos.')
print('This limitation is being actively researched, any contributions will be welcome.')
username = str(input('Enter the tiktok username here: '))
print(
"Due to specific limitations of the current data method, downloading by creator will only get the latest 30 videos."
)
print(
"This limitation is being actively researched, any contributions will be welcome."
)
username = str(input("Enter the tiktok username here: "))
links = getLinks(username)
downloadtiktoks(links)
sys.exit()
@@ -50,21 +65,28 @@ try:
## Stream
if question == 2:
watchquestion = int(input("""Do you want to watch your liked videos or a creator?
watchquestion = int(
input(
"""Do you want to watch your liked videos or a creator?
(1) Liked Videos
(2) Creator
"""))
"""
)
)
os.system("cls || clear")
## Stream liked videos
if watchquestion == 1:
randomquestion = int(input("""Do you want to watch the tiktoks in randomized order?
randomquestion = int(
input(
"""Do you want to watch the tiktoks in randomized order?
(1) Yes
(2) No
"""))
"""
)
)
os.system("cls || clear")
## Stream liked videos randomized
@@ -83,9 +105,13 @@ try:
## Stream creator
if watchquestion == 2:
print('Due to specific limitations of the current data method, watching by creator will only get the latest 30 videos.')
print('This limitation is being actively researched, any contributions will be welcome.')
username = str(input('Enter the tiktok username here: '))
print(
"Due to specific limitations of the current data method, watching by creator will only get the latest 30 videos."
)
print(
"This limitation is being actively researched, any contributions will be welcome."
)
username = str(input("Enter the tiktok username here: "))
streamuser(username)
sys.exit()
@@ -98,12 +124,16 @@ except ValueError:
# Error handling for missing Likes.txt file
except FileNotFoundError:
print("The 'Likes.txt' file was not found. Make sure it is in the program folder and try again.")
print(
"The 'Likes.txt' file was not found. Make sure it is in the program folder and try again."
)
# Error handling for MPV media player or MPV not found in PATH
except subprocess.CalledProcessError:
os.system("cls || clear")
print("Mpv media player was not found on your system path. Make sure it's installed and try again.")
print(
"Mpv media player was not found on your system path. Make sure it's installed and try again."
)
# Error handling for exiting the code with CTRL + C
except KeyboardInterrupt:

1
setup.py
View File

@@ -1,2 +1,3 @@
import os
os.system("pip install -r requirements.txt --user")

18
src/byuser.py
View File

@@ -1,12 +1,16 @@
import requests
import atoma
import sys
import atoma
import requests
def info(username):
response = requests.get(f'https://proxitok.pussthecat.org/@{username}/rss')
response = requests.get(f"https://proxitok.pussthecat.org/@{username}/rss")
if response.status_code == 404:
print("Something went wrong while getting the information. Make sure the username was correctly inserted and try again.")
print(
"Something went wrong while getting the information. Make sure the username was correctly inserted and try again."
)
sys.exit()
if str(response.content) == "b''":
print("The specified account does not exist.")
@@ -16,14 +20,15 @@ def info(username):
# if feed.description == None:
# print("This account does not have a bio.")
# else:
#print(feed.description) ## TIKTOK BIO
# print(feed.description) ## TIKTOK BIO
def getLinks(username):
feed = info(username)
linklist = []
for i in range(len(feed.items)):
linklist.append('https://www.tiktok.com' + feed.items[i].link)
linklist.append("https://www.tiktok.com" + feed.items[i].link)
return linklist
@@ -35,4 +40,5 @@ def streamuser(username):
for i in range(len(links)):
from src.streaming import mpv
mpv(links[i])

29
src/downloader.py
View File

@@ -1,29 +1,32 @@
from src.functions import detect_dead_link
from src.functions import url_redirection
import os
from src.functions import detect_dead_link, url_redirection
from yt_dlp import YoutubeDL
import os
def downloader(url):
ydl_opts = {
'ignoreerrors': True,
'format': 'bestvideo*+bestaudio/best',
'outtmpl': os.getcwd()+'/video/by-creator/%(creator)s/%(id)s.%(ext)s'
"ignoreerrors": True,
"format": "bestvideo*+bestaudio/best",
"outtmpl": os.getcwd()
+ "/video/by-creator/%(creator)s/%(id)s.%(ext)s",
}
YoutubeDL(ydl_opts).download(url)
def downloadtiktoks(urls):
index = -1
errorcount = []
a = input(f"""
a = input(
f"""
**WARNING**
This action will download up to {len(urls)} tiktoks in CLI-TIkTok/video/
Ensure you have enough free space before proceeding!
\n\t\tPress ENTER to proceed...
""")
if a != '':
print('Operation canceled.')
"""
)
if a != "":
print("Operation canceled.")
return
while True:
try:
@@ -32,12 +35,14 @@ def downloadtiktoks(urls):
if detect_dead_link(url) == True:
downloader(url)
print('')
print("")
else:
errorcount.append(urls)
except IndexError:
print("The tiktoks were downloaded")
return
if len(errorcount) != 0:
print(f'\n{len(errorcount)} video(s) failed to download.\nThe video(s) were likely banned or removed from the platform.')
print(
f"\n{len(errorcount)} video(s) failed to download.\nThe video(s) were likely banned or removed from the platform."
)
break

37
src/functions.py
View File

@@ -1,12 +1,13 @@
import requests
import random
import os
import random
import requests
def listas():
#Retrieves tiktok links and dates from Likes.txt
# Retrieves tiktok links and dates from Likes.txt
i = 0
arquivo = open('Likes.txt', 'r', encoding='utf-8')
arquivo = open("Likes.txt", "r", encoding="utf-8")
linhas = arquivo.readlines()
tamanho = len(linhas)
arquivo.close()
@@ -14,7 +15,7 @@ def listas():
vet = []
listalinks = []
listadatas = []
arquivo = open('Likes.txt', 'r', encoding='utf-8')
arquivo = open("Likes.txt", "r", encoding="utf-8")
while i <= tamanho:
lelinha = arquivo.readline().rstrip()
@@ -23,25 +24,25 @@ def listas():
arquivo.close()
for i in range(2, len(vet), 3):
listalinks.append(vet[i-1][1])
listalinks.append(vet[i - 1][1])
for i in range(2, len(vet), 3):
listadatas.append(vet[i-2][1] + ' ' + vet[i-2][2])
listadatas.append(vet[i - 2][1] + " " + vet[i - 2][2])
return listalinks, listadatas
#Unused function - Might be useful in future iterations of the project
#https://github.com/nanometer5088/CLI-TikTok/commit/ad589d7b324042ee85a270625df3ad9f6f82ab8a
# Unused function - Might be useful in future iterations of the project
# https://github.com/nanometer5088/CLI-TikTok/commit/ad589d7b324042ee85a270625df3ad9f6f82ab8a
def removevideo():
if os.path.exists(os.getcwd()+"/video/video"):
os.remove(os.getcwd()+"/video/video")
if os.path.exists(os.getcwd() + "/video/video"):
os.remove(os.getcwd() + "/video/video")
def detect_dead_link(url):
#Detects if the video is available to be streamed or downloaded.
dead_url_start_with = 'https://www.tiktok.com/@/video'
#No user on the @ means video was removed, taken down or isn't in a video format
# Detects if the video is available to be streamed or downloaded.
dead_url_start_with = "https://www.tiktok.com/@/video"
# No user on the @ means video was removed, taken down or isn't in a video format
if dead_url_start_with in url:
return False # Means do not proceed with the download
else:
@@ -49,12 +50,14 @@ def detect_dead_link(url):
def url_redirection(url):
#Tiktok links from the Likes.txt are shortened. They need to be redirected to the final link, which is done here.
headers = {'User-Agent': 'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/110.0.0.0 Safari/537.36'}
# Tiktok links from the Likes.txt are shortened. They need to be redirected to the final link, which is done here.
headers = {
"User-Agent": "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/110.0.0.0 Safari/537.36"
}
response = requests.get(url, headers=headers)
return response.url
def randomvideo(urls):
#Chooses a random video from Likes.txt - Optional Feature
# Chooses a random video from Likes.txt - Optional Feature
return random.randint(0, (len(urls) - 1))

42
src/init.py
View File

@@ -1,25 +1,32 @@
from os import system
import sys
from os import system
def init():
#Intro for the user
# Intro for the user
system("cls || clear")
input("Welcome to CLI TikTok, an open-source TikTok archiver and viewer!\nPress ENTER to proceed")
input(
"Welcome to CLI TikTok, an open-source TikTok archiver and viewer!\nPress ENTER to proceed"
)
# Detect and install libraries - If they aren't installed,
# the user is prompted to make the auto-installation.
try:
import yt_dlp
import requests
import atoma
import requests
import yt_dlp
system("cls || clear")
except ModuleNotFoundError:
system("cls || clear")
input("""
input(
"""
The program detected dependencies are not installed
Press ENTER to install the necessary libraries
(You will need to open the program again afterwards)
""")
"""
)
system("pip install -r requirements.txt --user")
system("cls || clear")
return -1
@@ -28,11 +35,15 @@ def init():
# If the user does not have internet access, warns him the software won't work properly and quit.
try:
import requests
data = requests.get("https://raw.githubusercontent.com/nanometer5088/CLI-TikTok/main/VERSION")
version = open('VERSION', 'r', encoding='utf=8')
data = requests.get(
"https://raw.githubusercontent.com/nanometer5088/CLI-TikTok/main/VERSION"
)
version = open("VERSION", "r", encoding="utf=8")
if version.readline().rstrip() > (data.text):
system("cls || clear")
input("""
input(
"""
There's a new version available!
Updates bring performance and feature improvements!
@@ -40,10 +51,15 @@ def init():
https://github.com/nanometer5088/CLI-TikTok/archive/refs/heads/main.zip
Press ENTER to proceed
""")
"""
)
version.close()
except requests.exceptions.ConnectionError:
print("CLI-TikTok detected your device isn't connected to the internet")
print("This software requires a reliable and uncensored internet connection to properly work")
print(
"CLI-TikTok detected your device isn't connected to the internet"
)
print(
"This software requires a reliable and uncensored internet connection to properly work"
)
print("Please try again with an internet connection")
sys.exit()

20
src/streaming.py
View File

@@ -1,32 +1,30 @@
from src.functions import detect_dead_link
from src.functions import url_redirection
import os
import subprocess
from src.functions import detect_dead_link, url_redirection
from yt_dlp import YoutubeDL
import subprocess
import os
def getVideoInfo(url):
ydl_opts = {
"quiet": True,
"simulate": True,
"forceurl": True
}
ydl_opts = {"quiet": True, "simulate": True, "forceurl": True}
return YoutubeDL(ydl_opts).extract_info(url)["url"]
def mpv(url):
subprocess.check_output(f'mpv "{getVideoInfo(url)}"', shell=True)
def playbackrandom(urls, datas):
while True:
os.system("cls || clear")
from src.functions import randomvideo
randomvideo = randomvideo(urls)
url = url_redirection(urls[randomvideo])
if detect_dead_link(url) == True:
print(f'\nVideo Liked - {datas[randomvideo]}\n')
print(f"\nVideo Liked - {datas[randomvideo]}\n")
mpv(url)
@@ -39,7 +37,7 @@ def playback(urls, datas):
url = url_redirection(urls[randomvideo])
if detect_dead_link(url) == True:
print(f'\nVideo Liked - {datas[randomvideo]}\n')
print(f"\nVideo Liked - {datas[randomvideo]}\n")
mpv(url)
except IndexError:
print("All tiktoks were played.")