Subtitle-Synchronizer/subsync.py at master · shamiul94/Subtitle-Synchronizer · 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
100
101
102
103
104
105
106
107
# @author: Shamiul Hasan (Heisenberg), BUET CSE
# Date: 3:45 PM (BDT), Oct 12, 2019
# Email: shamiulhasan93@gmail.com
# github: shamiul94


import os
import sys
from argparse import ArgumentParser

parser = ArgumentParser()

parser.add_argument("-f", "--file", type=str, dest="filename", required=True,
                    help="input subtitle file name")

parser.add_argument("-t", "--time",
                    required=True, type=int, dest="diff", default=0,
                    help="input positive (speed up subtitle) or negative (speed down subtitle) number of seconds")

args = parser.parse_args()
dir_path = str(os.path.dirname(os.path.realpath(__file__)))
diff = args.diff
filename = dir_path + '/' + args.filename

with open(filename) as f:
    content = f.readlines()

content = [x for x in content]

fw = open(filename, "w")
for x in content:
    if x.strip().__contains__('-->'):
        line = x.strip().split(' --> ')
        first = line[0]
        second = line[1]

        firstTimeWithoutMS = first.split(',')[0]
        firstMS = first.split(',')[1]
        secondTimeWithoutMS = second.split(',')[0]
        secondMS = second.split(',')[1]

        firstHour = int(firstTimeWithoutMS.split(':')[0])
        firstMinute = int(firstTimeWithoutMS.split(':')[1])
        firstSecond = int(firstTimeWithoutMS.split(':')[2])

        secondHour = int(secondTimeWithoutMS.split(':')[0])
        secondMinute = int(secondTimeWithoutMS.split(':')[1])
        secondSecond = int(secondTimeWithoutMS.split(':')[2])

        firstTime = firstHour * 60 * 60 + firstMinute * 60 + firstSecond + diff
        secondTime = secondHour * 60 * 60 + secondMinute * 60 + secondSecond + diff

        if firstTime < 0 or secondTime < 0:
            print('\n')
            print('#' * 50)
            print('Synchronization Not Possible. Your desired output goes under 00:00:00. File Unchanged.')
            print('#' * 50)
            print('First 10 lines of the original file is:\n')
            print('*'*50)
            fw.close()
            fw = open(filename, "w")
            for i in range(10):
                print(content[i])
            for y in content:
                fw.write(y)
            fw.close()
            sys.exit()

        firstHour = int(firstTime / 3600)
        firstMinute = int((firstTime % 3600) / 60)
        firstSecond = int(((firstTime % 3600) % 60))

        secondHour = int(secondTime / 3600)
        secondMinute = int((secondTime % 3600) / 60)
        secondSecond = int(((secondTime % 3600) % 60))

        # format = 02:39:53,500 --> 02:39:54,859

        finalLine = str(firstHour).zfill(2) + ':' \
                    + str(firstMinute).zfill(2) + ':' \
                    + str(firstSecond).zfill(2) + ',' \
                    + firstMS + ' --> ' \
                    + str(secondHour).zfill(2) + ':' \
                    + str(secondMinute).zfill(2) + ':' \
                    + str(secondSecond).zfill(2) + ',' + secondMS

        fw.write(finalLine + '\n')

    else:
        fw.write(x)

fw.close()

print('\n')
print('#' * 50)
print('Subtitle File Successfully Synchronized!!!')
print('#' * 50)

print('\nFirst 10 lines of the generated file is: \n')
print('*' * 50)

N = 10
with open(filename) as f:
    content = f.readlines()

for i in range(N):
    print(content[i])