chore:docker

.gitattributes

@@ -1 +1,2 @@
+*.sh linguist-language=Python
 Dockerfile linguist-language=Python

.github/workflows/main.yml

@@ -50,22 +50,22 @@ jobs:
       #       open_ffmpeg = False
       #     print(open_ffmpeg)')" >> $GITHUB_ENV
       - name: Set up Chrome
-        if: env.OPEN_DRIVER == 'True' || env.OPEN_DRIVER == 'true'
+        if: env.OPEN_DRIVER == 'True'
         uses: browser-actions/setup-chrome@latest
         with:
           chrome-version: stable
       - name: Download chrome driver
-        if: env.OPEN_DRIVER == 'True' || env.OPEN_DRIVER == 'true'
+        if: env.OPEN_DRIVER == 'True'
         uses: nanasess/setup-chromedriver@master
       # - name: Install FFmpeg
-      #   if: env.OPEN_FFMPEG == 'True' || env.OPEN_FFMPEG == 'true'
+      #   if: env.OPEN_FFMPEG == 'True'
       #   run: sudo apt-get update && sudo apt-get install -y ffmpeg
       - name: Install pipenv
         run: pip3 install --user pipenv
       - name: Install dependecies
         run: pipenv --python 3.13 && pipenv install
       - name: Install selenium
-        if: env.OPEN_DRIVER == 'True' || env.OPEN_DRIVER == 'true'
+        if: env.OPEN_DRIVER == 'True'
         run: pipenv install selenium
       - name: Update
         run: pipenv run dev

Dockerfile

@@ -1,6 +1,6 @@
 FROM python:3.13 AS builder
 
-ARG LITE=false
+ARG LITE=False
 
 WORKDIR /app
 
@@ -9,16 +9,16 @@ COPY Pipfile* ./
 RUN pip install -i https://mirrors.aliyun.com/pypi/simple pipenv
 
 RUN PIPENV_VENV_IN_PROJECT=1 pipenv install --deploy\
-  && if [ "$LITE" = false ]; then pipenv install selenium; fi
+  && if [ "$LITE" = False ]; then pipenv install selenium; fi
 
 
 FROM python:3.13-slim
 
 ARG APP_WORKDIR=/iptv
-ARG LITE=false
+ARG LITE=False
 
 ENV APP_WORKDIR=$APP_WORKDIR
-ENV PIPENV_VENV_IN_PROJECT=1
+ENV LITE=$LITE
 ENV PATH="/.venv/bin:$PATH"
 
 WORKDIR $APP_WORKDIR
@@ -40,7 +40,7 @@ RUN echo "deb https://mirrors.aliyun.com/debian/ bookworm main contrib non-free
 
 RUN apt-get update && apt-get install -y --no-install-recommends cron
 
-RUN if [ "$LITE" = false ]; then apt-get install -y --no-install-recommends chromium chromium-driver; fi \
+RUN if [ "$LITE" = False ]; then apt-get install -y --no-install-recommends chromium chromium-driver; fi \
   && apt-get clean && rm -rf /var/lib/apt/lists/*
 
 RUN (crontab -l ; \
@@ -49,5 +49,10 @@ RUN (crontab -l ; \
 
 EXPOSE 8000
 
-CMD . /.venv/bin/activate && service cron start && python $APP_WORKDIR/main.py \
-  && gunicorn -w 4 -b 0.0.0.0:8000 main:app
+COPY entrypoint.sh /iptv_entrypoint.sh
+
+COPY config /iptv_config
+
+RUN chmod +x /iptv_entrypoint.sh
+
+ENTRYPOINT /iptv_entrypoint.sh

Pipfile

@@ -6,11 +6,11 @@ verify_ssl = true
 [scripts]
 dev = "python main.py"
 ui = "python tkinter_ui/tkinter_ui.py"
-docker_run = "docker run -v config:/iptv/config -v output:/iptv/output -d -p 8000:8000 guovern/iptv"
-docker_run_lite = "docker run -v config:/iptv-lite/config -v output:/iptv-lite/output -d -p 8000:8000 guovern/iptv:lite"
+docker_run = "docker run -v ./config:/iptv/config -v ./output:/iptv/output -d -p 8000:8000 guovern/iptv"
+docker_run_lite = "docker run -v ./config:/iptv_lite/config -v ./output:/iptv_lite/output -d -p 8000:8000 guovern/iptv:lite"
 tkinter_build = "pyinstaller tkinter_ui/tkinter_ui.spec"
 docker_build = "docker buildx build --platform linux/amd64,linux/arm64,linux/arm/v7 --build-arg APP_WORKDIR=/iptv -t guovern/iptv ."
-docker_build_lite = "docker buildx build --platform linux/amd64,linux/arm64,linux/arm/v7 --build-arg APP_WORKDIR=/iptv-lite --build-arg LITE=true -t guovern/iptv:lite ."
+docker_build_lite = "docker buildx build --platform linux/amd64,linux/arm64,linux/arm/v7 --build-arg APP_WORKDIR=/iptv_lite --build-arg LITE=True -t guovern/iptv:lite ."
 
 [dev-packages]
 requests = "*"

README.md

@@ -198,7 +198,7 @@ docker run -v /etc/docker/config:/iptv/config -v /etc/docker/output:/iptv/output
 - iptv:lite：
 
 ```bash
-docker run -v /etc/docker/config:/iptv-lite/config -v /etc/docker/output:/iptv-lite/output -d -p 8000:8000 guovern/iptv:lite
+docker run -v /etc/docker/config:/iptv_lite/config -v /etc/docker/output:/iptv_lite/output -d -p 8000:8000 guovern/iptv:lite
 ```
 
 3. 更新结果：

README_en.md

@@ -153,7 +153,7 @@ pipenv run ui
 
 ### Method 4: Docker
 
-- iptv (Full version): Higher performance requirements, slower update speed, high stability and success rate. Set open_driver = False to switch to the iptv-lite running mode (recommended for hotel sources, multicast sources, and online searches)
+- iptv (Full version): Higher performance requirements, slower update speed, high stability and success rate. Set open_driver = False to switch to the lite running mode (recommended for hotel sources, multicast sources, and online searches)
 - iptv:lite (Condensed version): Lightweight, low performance requirements, fast update speed, stability uncertain (recommend using this version for the subscription source)
 
 It's recommended to try each one and choose the version that suits you
@@ -200,7 +200,7 @@ docker run -v /etc/docker/config:/iptv/config -v /etc/docker/output:/iptv/output
 - iptv:lite：
 
 ```bash
-docker run -v /etc/docker/config:/iptv-lite/config -v /etc/docker/output:/iptv-lite/output -d -p 8000:8000 guovern/iptv:lite
+docker run -v /etc/docker/config:/iptv_lite/config -v /etc/docker/output:/iptv_lite/output -d -p 8000:8000 guovern/iptv:lite
 ```
 
 3. Update results:

docs/tutorial.md

@@ -245,7 +245,7 @@ docker run -v /etc/docker/config:/iptv/config -v /etc/docker/output:/iptv/output
 - iptv:lite：
 
 ```bash
-docker run -v /etc/docker/config:/iptv-lite/config -v /etc/docker/output:/iptv-lite/output -d -p 8000:8000 guovern/iptv:lite
+docker run -v /etc/docker/config:/iptv_lite/config -v /etc/docker/output:/iptv_lite/output -d -p 8000:8000 guovern/iptv:lite
 ```
 
 3. 更新结果：

docs/tutorial_en.md

@@ -197,7 +197,7 @@ pipenv run ui
 
 ### Method 4: Docker
 
-- iptv (Full version): Higher performance requirements, slower update speed, high stability and success rate. Set open_driver = False to switch to the iptv-lite running mode (recommended for hotel sources, multicast sources, and online searches)
+- iptv (Full version): Higher performance requirements, slower update speed, high stability and success rate. Set open_driver = False to switch to the lite running mode (recommended for hotel sources, multicast sources, and online searches)
 - iptv:lite (Condensed version): Lightweight, low performance requirements, fast update speed, stability uncertain (recommend using this version for the subscription source)
 
 It's recommended to try each one and choose the version that suits you
@@ -244,7 +244,7 @@ docker run -v /etc/docker/config:/iptv/config -v /etc/docker/output:/iptv/output
 - iptv:lite：
 
 ```bash
-docker run -v /etc/docker/config:/iptv-lite/config -v /etc/docker/output:/iptv-lite/output -d -p 8000:8000 guovern/iptv:lite
+docker run -v /etc/docker/config:/iptv_lite/config -v /etc/docker/output:/iptv_lite/output -d -p 8000:8000 guovern/iptv:lite
 ```
 
 3. Update results:

driver/setup.py

@@ -1,4 +1,10 @@
-from selenium import webdriver
+from utils.config import config
+
+if config.open_driver:
+    try:
+        from selenium import webdriver
+    except:
+        pass
 
 
 def setup_driver(proxy=None):

driver/utils.py

@@ -6,6 +6,13 @@ from utils.retry import (
 from time import sleep
 import re
 from bs4 import BeautifulSoup
+from utils.config import config
+
+if config.open_driver:
+    try:
+        from selenium.webdriver.common.by import By
+    except:
+        pass
 
 
 def get_soup_driver(url):
@@ -33,8 +40,6 @@ def search_submit(driver, name):
     """
     Input key word and submit with driver
     """
-    from selenium.webdriver.common.by import By
-
     search_box = locate_element_with_retry(driver, (By.XPATH, '//input[@type="text"]'))
     if not search_box:
         return

entrypoint.sh

@@ -0,0 +1,17 @@
+#!/bin/bash
+
+for file in /iptv_config/*; do
+  filename=$(basename "$file")
+  target_file="$APP_WORKDIR/config/$filename"
+  if [ ! -e "$target_file" ]; then
+    cp -r "$file" "$target_file"
+  fi
+done
+
+. /.venv/bin/activate
+
+service cron start
+
+python $APP_WORKDIR/main.py
+
+gunicorn -w 4 -b 0.0.0.0:8000 main:app

updates/hotel/request.py

@@ -22,6 +22,12 @@ from updates.subscribe import get_channels_by_subscribe_urls
 from collections import defaultdict
 import updates.fofa.fofa_map as fofa_map
 
+if config.open_driver:
+    try:
+        from selenium.webdriver.common.by import By
+    except:
+        pass
+
 
 async def get_channels_by_hotel(callback=None):
     """
@@ -32,8 +38,6 @@ async def get_channels_by_hotel(callback=None):
     proxy = None
     open_proxy = config.open_proxy
     open_driver = config.open_driver
-    if open_driver:
-        from selenium.webdriver.common.by import By
     page_num = config.hotel_page_num
     region_list = config.hotel_region_list
     if "all" in region_list or "ALL" in region_list or "全部" in region_list:

updates/multicast/request.py

@@ -26,6 +26,12 @@ from urllib.parse import parse_qs
 from collections import defaultdict
 from .update_tmp import get_multicast_region_result_by_rtp_txt
 
+if config.open_driver:
+    try:
+        from selenium.webdriver.common.by import By
+    except:
+        pass
+
 
 async def get_channels_by_multicast(names, callback=None):
     """
@@ -36,8 +42,6 @@ async def get_channels_by_multicast(names, callback=None):
     proxy = None
     open_proxy = config.open_proxy
     open_driver = config.open_driver
-    if open_driver:
-        from selenium.webdriver.common.by import By
     page_num = config.multicast_page_num
     if open_proxy:
         proxy = await get_proxy(pageUrl, best=True, with_test=True)

updates/online_search/request.py

@@ -24,6 +24,12 @@ from tqdm.asyncio import tqdm_asyncio
 from concurrent.futures import ThreadPoolExecutor
 from requests_custom.utils import get_soup_requests, close_session
 
+if config.open_driver:
+    try:
+        from selenium.webdriver.common.by import By
+    except:
+        pass
+
 
 async def get_channels_by_online_search(names, callback=None):
     """
@@ -36,8 +42,6 @@ async def get_channels_by_online_search(names, callback=None):
     proxy = None
     open_proxy = config.open_proxy
     open_driver = config.open_driver
-    if open_driver:
-        from selenium.webdriver.common.by import By
     page_num = config.online_search_page_num
     if open_proxy:
         proxy = await get_proxy(pageUrl, best=True, with_test=True)

utils/config.py

@@ -270,7 +270,9 @@ class ConfigManager:
 
     @property
     def open_driver(self):
-        return self.config.getboolean("Settings", "open_driver", fallback=True)
+        return not os.environ.get("LITE") and self.config.getboolean(
+            "Settings", "open_driver", fallback=True
+        )
 
     @property
     def hotel_page_num(self):

utils/retry.py

@@ -2,9 +2,12 @@ from time import sleep
 from utils.config import config
 
 if config.open_driver:
-    from selenium.webdriver.support.ui import WebDriverWait
-    from selenium.webdriver.support import expected_conditions as EC
-    from selenium.common.exceptions import TimeoutException
+    try:
+        from selenium.webdriver.support.ui import WebDriverWait
+        from selenium.webdriver.support import expected_conditions as EC
+        from selenium.common.exceptions import TimeoutException
+    except:
+        pass
 
 max_retries = 2