78 lines
1.9 KiB
Python
78 lines
1.9 KiB
Python
|
#! /usr/bin/env python
|
||
|
|
|
||
|
|
import sys
|
||
|
|
import os
|
||
|
|
|
||
|
|
|
||
|
|
def main(src, dst, sort_by_size):
|
||
|
|
if src.endswith("/"):
|
||
|
|
src = src[:-1]
|
||
|
|
if dst.endswith("/"):
|
||
|
|
dst = dst[:-1]
|
||
|
|
|
||
|
|
# parse the dst tree
|
||
|
|
dst_tree = {}
|
||
|
|
for root, dnames, fnames in os.walk(dst):
|
||
|
|
root = root.replace(dst, "", 1)
|
||
|
|
for fname in fnames:
|
||
|
|
path = os.path.join(root, fname)
|
||
|
|
try:
|
||
|
|
dst_tree[fname].add(path)
|
||
|
|
except KeyError:
|
||
|
|
dst_tree[fname] = set()
|
||
|
|
dst_tree[fname].add(path)
|
||
|
|
|
||
|
|
# parse the src tree
|
||
|
|
filelist = []
|
||
|
|
for root, dnames, fnames in os.walk(src):
|
||
|
|
root = root.replace(src, "", 1)
|
||
|
|
for fname in fnames:
|
||
|
|
path = os.path.join(root, fname)
|
||
|
|
paths = dst_tree.get(fname)
|
||
|
|
|
||
|
|
if not paths:
|
||
|
|
# file not found
|
||
|
|
try:
|
||
|
|
size = os.path.getsize(os.path.join(src, path[1:]))
|
||
|
|
except OSError:
|
||
|
|
size = 0L
|
||
|
|
filelist.append((size, path))
|
||
|
|
else:
|
||
|
|
# fname found
|
||
|
|
if path in paths:
|
||
|
|
# exact match
|
||
|
|
continue
|
||
|
|
else:
|
||
|
|
# partial match
|
||
|
|
# TODO
|
||
|
|
continue
|
||
|
|
|
||
|
|
if sort_by_size:
|
||
|
|
filelist.sort(reverse=True)
|
||
|
|
|
||
|
|
for size, path in filelist:
|
||
|
|
# convert size to human readable
|
||
|
|
human = ""
|
||
|
|
for base in ["KiB", "MiB", "GiB"]:
|
||
|
|
size /= 1024
|
||
|
|
if size < 1024:
|
||
|
|
human = "{0:6.1f}{1}".format(size, base)
|
||
|
|
break
|
||
|
|
|
||
|
|
print("{0} {1}".format(human, path))
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
try:
|
||
|
|
src, dst = sys.argv[1], sys.argv[2]
|
||
|
|
except IndexError:
|
||
|
|
print("invalid arguments count")
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
try:
|
||
|
|
sort_by_size = sys.argv[3]
|
||
|
|
except IndexError:
|
||
|
|
sort_by_size = False
|
||
|
|
|
||
|
|
main(src, dst, bool(sort_by_size))
|