convert.py

import sys
import os.path
import glob

# Get input and output folders as parameters

# if __name__ == '__main__':
#     if len(sys.argv) < 3:
#         pass
#     else:
#         print 'Usage:'   
    
try:
    dir_in = sys.argv[1]
    dir_out = sys.argv[2]
except IndexError:
    print('======================================================\n' +
          '=== Not enough parameters. Check the command line. ===\n' +
          '===      Should be input_folder output_folder      ===\n' +
          '======================================================')
    sys.exit(1)

file_in_list = glob.glob(dir_in + '/*.txt')

def timestamp_convert(timestamp):
    """Converts time stamp to CUE time stamp
    Converts time stamp from HH:MM:ss,ms to ММ:SS:MS
    """
    timestamp = timestamp[3:].replace(',', ':')
    if len(timestamp) == 2:
        timestamp = timestamp + ':00:00'
    elif len(timestamp) == 5:
        timestamp = timestamp + ':00'
    else:
        timestamp = timestamp + '0'
    return timestamp


for j in file_in_list:
    file_in_path = file_in_list[j]
    file_in_name = os.path.basename(file_in_path)

    # Open and read input file
    with open(file_in_path, 'r') as file_in:
        file_in_content = file_in.read()

    # Initial cursor position
    cursor_pos = file_in_content.find('00:00:00 ')

    # Number of strings in file to convert
    string_count = file_in_content.count('\n00:')


    def string_parser(cursor_pos):
        """ String parser
        Parses string at current cursor position
        """
        timestamp = ''
        singer = ''
        song_title = ''
        track_title = ''

        while file_in_content[cursor_pos] != ' ':
            timestamp += file_in_content[cursor_pos]
            cursor_pos += 1
        cursor_pos += 1
        time = timestamp_convert(timestamp)

        if file_in_content[cursor_pos] != '•':
            song_flag = False
            while file_in_content[cursor_pos] != '\n':
                track_title += file_in_content[cursor_pos]
                cursor_pos += 1
            cursor_pos += 1
        else:
            song_flag = True
            cursor_pos += 2
            while file_in_content[cursor_pos] != '-':
                singer += file_in_content[cursor_pos]
                cursor_pos += 1
            singer = singer.rstrip()
            cursor_pos += 2
            while file_in_content[cursor_pos] != '\n':
                song_title += file_in_content[cursor_pos]
                cursor_pos += 1
            cursor_pos += 1
            track_title = '• ' + singer + ' - ' + song_title
        return time, song_flag, track_title, singer, cursor_pos

    issue_number = file_in_name[:3]
    issue_name = file_in_name[4:-10]
    notification = issue_number + ': .txt >'

    # Here's an initial cyrillic 'A'
    issue_title = 'А-' + issue_number + '. ' + issue_name
    issue_author = 'Борис Гребенщиков'
    performer = issue_author

    cue_header ='''TITLE "{issue_title}"'
    'PERFORMER "{performer}"'
    'FILE "{issue_title}" WAVE'''

    file_out_content = ''.join(cue_header.format())

    for i in range(string_count):
        track_number = '{:0>2}'.format(str(i + 1))
        time = string_parser(cursor_pos)[0]
        song_flag = string_parser(cursor_pos)[1]
        track_title = string_parser(cursor_pos)[2]
        singer = string_parser(cursor_pos)[3]
        if song_flag == True:
            performer = singer
        else:
            performer = issue_author
        cue_item = [
            '  TRACK ', track_number, ' AUDIO\n',
            '    TITLE "', track_title, '"\n',
            '    PERFORMER "', performer, '"\n',
            '    INDEX 01 ', time, '\n']
        file_out_content = file_out_content + ''.join(cue_item)
        cursor_pos = string_parser(cursor_pos)[4]

    # Create output file
    file_out_name = issue_title + '.cue'
    file_out_path = dir_out + '/' + file_out_name

    with open(file_out_path, 'w') as file_out:
        file_out.write(file_out_content)
    notification = notification + ' .cue succeed'
    print(notification)

    file_in.close()
    file_out.close()

print('=== ' + str(j + 1) + ' file(s) converted ===')

process.py

#!/usr/bin/env python

import os, os.path
import sys


def convert_timestamp(timestamp):
    """
    Converts time stamp from HH:MM:ss,ms to ММ:SS:MS
    """
    _, a, b = timestamp.split(':', maxsplit=2)
    return ':'.join([a, b[:2], '00'])


def process(fname):
    with open(fname, 'r') as f:
        for line in f:
            if line.startswith('0'):
                ts, rest = line.split(maxsplit=1)
                timestamp = convert_timestamp(ts)
                if rest.startswith('•'):
                    _, track = rest.split(maxsplit=1)
                else:
                    track = rest
                print(timestamp, track)


if __name__ == '__main__':
    if len(sys.argv) < 3:
        print('Usage: ...')
    else:
        input_dir = sys.argv[1]
        output_dir = sys.argv[2]
        for f in os.listdir(input_dir):
            if f.endswith('.txt'):
                process(os.path.join(input_dir, f))