mirror of
https://github.com/Xevion/truefile.git
synced 2026-01-31 08:26:25 -06:00
added coloring, final filtering fixed
needs fixing for document types, but that should be in a commandline flag instead
This commit is contained in:
+10
-5
@@ -2,6 +2,7 @@ import click
|
||||
import os
|
||||
import imghdr
|
||||
import sys
|
||||
import colorama
|
||||
|
||||
@click.group()
|
||||
def cli():
|
||||
@@ -13,21 +14,25 @@ def what(*args, **kwargs):
|
||||
except PermissionError:
|
||||
return None
|
||||
|
||||
def unequal(x, y):
|
||||
return x != y
|
||||
|
||||
@cli.command()
|
||||
@click.option('-s', '--showall', is_flag=True, help='Show all files regardless of detected extension mishaps', default=False)
|
||||
@click.option('-s', '--showall', is_flag=True, help='Show all files regardless of detected extension mishaps')
|
||||
def detect(showall):
|
||||
colorama.init()
|
||||
path = os.getcwd()
|
||||
# Filter and map files down to absolute paths for FILES (not directories)
|
||||
files = map(lambda file : file, [file for file in os.listdir(path)]) # abspath of all files in curdir
|
||||
files = filter(os.path.isfile, files) # file not directory
|
||||
files = zip(files, map(what, files)) # get scan of file type
|
||||
files = filter(lambda x : x[1] is not None, files) # filter for only images
|
||||
files = list(map(lambda x : (x[0], os.path.splitext(x[0])[1], x[1]), files)) # find just the file extension and put into tuple
|
||||
files = list(map(lambda x : (x[0], os.path.splitext(x[0])[1], '.' + x[1]), files)) # find just the file extension and put into tuple
|
||||
if not showall:
|
||||
files = list(filter(lambda x : x[1].replace('jpg', 'jpeg') != x[2], files)) # filter for differing extensions
|
||||
files = list(filter(lambda x : unequal(x[1], x[2]), files)) # filter for differing extensions
|
||||
largest = max(map(lambda x : len(x[0]), files)) # get length of the longest file name
|
||||
print('\n'.join(map(lambda x : f'{x[0].ljust(largest)} | {x[1]} -> {x[2]}', files))) # print with proper spacing
|
||||
|
||||
print('\n'.join(map(lambda x : f'{x[0].ljust(largest)} | {colorama.Fore.RED if unequal(x[1], x[2]) else colorama.Fore.GREEN}{x[1]} -> {x[2]}{colorama.Fore.RESET}', files))) # print with proper spacing
|
||||
print(colorama.Style.RESET_ALL, end='')
|
||||
|
||||
@cli.command()
|
||||
def rename():
|
||||
|
||||
Reference in New Issue
Block a user