Das deutsche Python-Forum

Code: Alles auswählen

fn_regex = re.compile(r"([\-0-9]+)")

Code: Alles auswählen

r"^(.*[^\-0-9])?(\-?[0-9]+)$"

Code: Alles auswählen

#!/usr/bin/env python
# -*- coding: UTF-8 -*-

"""
umbenennen einer Render Image Sequenz

kopiert alle Dateien einer Image Sequenz von SOURCE_PATH nach DEST_PATH und
nummeriert dabei die Dateien neu durch.

Anwendungsfall - negative Zeitleiste
------------------------------------
In 3dsmax ist es z.B. möglich die Zeitleiste in den negativen Bereich zu
erweitern. In diesem Falle haben die einzelnen Dateien allerdings ungünstige
Namen.
z.B.:
    Frame-003.png
    Frame-002.png
    Frame-001.png
    Frame-003.png
    Frame0000.png
    Frame0001.png
    Frame0002.png
Zum einen kann diese Sequenz nicht in einem eingelesen werden (z.B. mit
VirtualDub) und zum anderen würde der negative Bereich umgekehrt eingelesen
werden.

Das Skript macht nun aus der Sequenz folgendes:
    Frame-003.png -> Frame_0000.png
    Frame-002.png -> Frame_0001.png
    Frame-001.png -> Frame_0002.png
    Frame-003.png -> Frame_0003.png
    Frame0000.png -> Frame_0004.png
    Frame0001.png -> Frame_0005.png
    Frame0002.png -> Frame_0006.png


Anwendungsfall - fehlende Frames
--------------------------------
Wenn innerhalb einer Image Sequenz Frames fehlen (oder aber auch nur jedes x'te
Frame absichtlich rausgerechnet hat) kann man ebenfalls die Sequenz nicht mit
VirtualDub eingelesen werden.
In dem Falle, sollte das Skript auch wieder eine korrekte Nummerierung
vornehmen. ACHTUNG: Das ist nicht getestet!


Created by Jens Diemer

license:
    GNU General Public License v3
    http://www.opensource.org/licenses/gpl-license.php
"""

import os, sys, re, shutil, time

__version__ = "v2.0"


def make_file_dict(filelist, basename):
    """
    >>> basename = "base"
    >>> filelist = ["base-2.png", "base-1.png", "base-0.png", "base01.png", ]
    >>> make_file_dict(filelist, basename)
    {0: 'base-0.png', 1: 'base01.png', -1: 'base-1.png', -2: 'base-2.png'}
    """
    pos = len(basename)
    file_dict = {}
    for filename in filelist:
        fn, ext = os.path.splitext(filename)
        fileno = int(fn[pos:])
        file_dict[fileno] = filename

    return file_dict


def file_copy(file_dict, basename, source_path, dest_path):
    start_time = time.time()
    filecount = len(file_dict)
    for no, key in enumerate(sorted(file_dict)):
        filename = file_dict[key]
        fn, ext = os.path.splitext(filename)

        src = os.path.join(source_path, filename)
        new_filename = "%s_%04i%s" % (basename, no, ext)

        print "%s -> %s" % (filename, new_filename)

        dst = os.path.join(dest_path, new_filename)

        current_time = int(time.time())
        elapsed = float(current_time-start_time)        # Vergangene Zeit
        estimated = elapsed / (no+1) * (filecount+1)    # Geschäzte Zeit
        if estimated>60:
            time_info = "%.2f/%.2fmin %.2fmin" % (
                elapsed/60.0, estimated/60.0, (estimated-elapsed)/60.0
            )
        else:
            time_info = "%i/%isec %isec" % (
                elapsed, estimated, estimated-elapsed
            )

        print "copy %s/%s %s left (%s)..." % (
            no+1, filecount, filecount-(no+1), time_info
        ),
        shutil.copy2(src, dst)
        print "OK"
        print

def file_sequenz_copy(source_path, dest_path):
    assert os.path.isdir(source_path)
    assert os.path.isdir(dest_path)

    print "Read filenames...",
    filelist = os.listdir(source_path)
    print "OK"

    print "Analyse baename...",
    # Gemeinsamen Basis Namen finden
    basename = os.path.commonprefix(filelist)
    print "OK:", basename

    print "Generate file dictionary...",
    file_dict = make_file_dict(filelist, basename)
    print "OK"

    print "Create destination path...",
    try:
        destination = os.path.join(dest_path, basename)
        os.mkdir(destination)
    except Exception, e:
        print "mkdir error:", e
    else:
        print "OK"

    print
    print "_"*80
    print "START FILE COPY"
    print

    # Dateien kopieren
    file_copy(file_dict, basename, source_path, destination)


def _test():
    print "_"*80
    print "doctest:"
    import doctest
    doctest.testmod()
    print "-"*80
    print "(END)"

if __name__ == "__main__":
    #~ _test()

    file_sequenz_copy(
        r"W:\3dsmax_rendering\Preview01",
        r"D:\Previews"
    )