locast2plex/main.py at main · tgorgdotcom/locast2plex · 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
89
90
91
92
93
94
95
96
97
98
99
# pylama:ignore=E722,E303,E302,E305
import os
import sys
import time
import platform
import argparse
import pathlib
from multiprocessing import Process

import lib.locast_service as locast_service
import lib.ssdp_server as ssdp_server
import lib.tuner_interface as tuner_interface
import lib.epg2xml as epg2xml
import lib.stations as stations
import lib.location as location
from lib.user_config import get_config
from lib.l2p_tools import clean_exit, get_version_str


if sys.version_info.major == 2 or sys.version_info < (3, 6):
    print('Error: Locast2Plex requires python 3.6+.')
    sys.exit(1)


def get_args():
    parser = argparse.ArgumentParser(description='Fetch TV from locast.', epilog='')
    parser.add_argument('--config_file', dest='cfg', type=str, default=None, help='')
    return parser.parse_args()


# Startup Logic
if __name__ == '__main__':

    os.chdir(os.path.dirname(os.path.abspath(__file__)))

    print("Initiating Locast2Plex v" + get_version_str())

    # Gather args
    args = get_args()

    # Get Operating system
    opersystem = platform.system()

    # set to directory of script
    script_dir = pathlib.Path(os.path.dirname(os.path.abspath(__file__)))

    # Open Configuration File
    print("Opening and Verifying Configuration File.")
    config = get_config(script_dir, opersystem, args)

    print("Getting user location...")
    location_info = location.DMAFinder(config)

    locast = locast_service.LocastService(location_info.location)

    if (not locast.login(config["main"]["locast_username"], config["main"]["locast_password"])) or (not locast.validate_user()):
        print("Invalid Locast Login Credentials. Exiting...")
        clean_exit(1)


    try:
        fcc_cache_dir = pathlib.Path(config["main"]["cache_dir"]).joinpath("stations")
        if not fcc_cache_dir.is_dir():
            fcc_cache_dir.mkdir()

        print("Starting First time Stations refresh...")
        stations.refresh_dma_stations_and_channels(config, locast, location_info.location)

        print("Starting Stations thread...")
        stations_server = Process(target=stations.stations_process, args=(config, locast, location_info.location))
        stations_server.daemon = True
        stations_server.start()

        print("Starting device server on " + config["main"]['plex_accessible_ip'] + ":" + config["main"]['plex_accessible_port'])
        tuner_interface.start(config, locast, location_info.location)

        if not config['main']['disable_ssdp']:
            print("Starting SSDP server...")
            ssdp_server = Process(target=ssdp_server.ssdp_process, args=(config,))
            ssdp_server.daemon = True
            ssdp_server.start()

        print("Starting First time EPG refresh...")
        epg2xml.generate_epg_file(config, location_info.location)

        print("Starting EPG thread...")
        epg_server = Process(target=epg2xml.epg_process, args=(config, location_info.location))
        epg_server.daemon = True
        epg_server.start()

        print("Locast2Plex is now online.")

        # wait forever
        while True:
            time.sleep(3600)

    except KeyboardInterrupt:
        print('^C received, shutting down the server')
        clean_exit()