mirror of
https://github.com/nikdoof/tvorganise.git
synced 2025-12-13 06:42:16 +00:00
Removed unused or unneeded chunks of code and comments
This commit is contained in:
120
tvorganise.py
120
tvorganise.py
@@ -31,116 +31,6 @@ from filename_config import tv_regex
|
||||
# end configs
|
||||
##############################################
|
||||
|
||||
def colour(text,colour="red"):
|
||||
nocolour=False
|
||||
if nocolour: # Colour no supported, return plain text
|
||||
return text
|
||||
#end if
|
||||
|
||||
c = {'red':'[31m',
|
||||
'green':'[32m',
|
||||
'blue':'[34m',
|
||||
}
|
||||
CLR=chr(27)+'[0m'
|
||||
if not colour in c.keys():
|
||||
raise ValueError("Invalid colour")
|
||||
else:
|
||||
return chr(27)+c[colour] + text + CLR
|
||||
#end if
|
||||
#end colour
|
||||
|
||||
class ProgressBar:
|
||||
"""From http://code.activestate.com/recipes/168639/"""
|
||||
def __init__(self, minValue = 0, maxValue = 10, totalWidth=12):
|
||||
self.progBar = "[]" # This holds the progress bar string
|
||||
self.min = minValue
|
||||
self.max = maxValue
|
||||
self.span = maxValue - minValue
|
||||
self.width = totalWidth
|
||||
self.amount = 0 # When amount == max, we are 100% done
|
||||
self.updateAmount(0) # Build progress bar string
|
||||
|
||||
def updateAmount(self, newAmount = 0):
|
||||
if newAmount < self.min: newAmount = self.min
|
||||
if newAmount > self.max: newAmount = self.max
|
||||
self.amount = newAmount
|
||||
|
||||
# Figure out the new percent done, round to an integer
|
||||
diffFromMin = float(self.amount - self.min)
|
||||
percentDone = (diffFromMin / float(self.span)) * 100.0
|
||||
percentDone = round(percentDone)
|
||||
percentDone = int(percentDone)
|
||||
|
||||
# Figure out how many hash bars the percentage should be
|
||||
allFull = self.width - 2
|
||||
numHashes = (percentDone / 100.0) * allFull
|
||||
numHashes = int(round(numHashes))
|
||||
|
||||
# build a progress bar with hashes and spaces
|
||||
self.progBar = "[" + '#'*numHashes + ' '*(allFull-numHashes) + "]"
|
||||
|
||||
# figure out where to put the percentage, roughly centered
|
||||
percentPlace = (len(self.progBar) / 2) - len(str(percentDone))
|
||||
percentString = str(percentDone) + "%"
|
||||
|
||||
# slice the percentage into the bar
|
||||
self.progBar = (self.progBar[0:percentPlace] + percentString
|
||||
+ self.progBar[percentPlace+len(percentString):])
|
||||
|
||||
def __str__(self):
|
||||
return str(self.progBar)
|
||||
|
||||
def copy_with_prog(src_file, dest_file, overwrite = False, block_size = 512):
|
||||
if not overwrite:
|
||||
if os.path.isfile(dest_file):
|
||||
raise IOError("File exists, not overwriting")
|
||||
|
||||
# Open src and dest files, get src file size
|
||||
src = open(src_file, "rb")
|
||||
dest = open(dest_file, "wb")
|
||||
|
||||
src_size = os.stat(src_file).st_size
|
||||
|
||||
# Set progress bar
|
||||
prgb = ProgressBar(totalWidth = 79, maxValue = src_size)
|
||||
|
||||
# Start copying file
|
||||
cur_block_pos = 0 # a running total of current position
|
||||
while True:
|
||||
cur_block = src.read(block_size)
|
||||
|
||||
# Update progress bar
|
||||
prgb.updateAmount(cur_block_pos)
|
||||
cur_block_pos += block_size
|
||||
|
||||
sys.stdout.write(
|
||||
'\r%s\r' % str(prgb)
|
||||
)
|
||||
|
||||
# If it's the end of file
|
||||
if not cur_block:
|
||||
# ..write new line to prevent messing up terminal
|
||||
print # print line break to clear progress bar
|
||||
break
|
||||
else:
|
||||
# ..if not, write the block and continue
|
||||
dest.write(cur_block)
|
||||
#end while
|
||||
|
||||
# Close files
|
||||
src.close()
|
||||
dest.close()
|
||||
|
||||
# Check output file is same size as input one!
|
||||
dest_size = os.stat(dest_file).st_size
|
||||
|
||||
if dest_size != src_size:
|
||||
raise IOError(
|
||||
"New file-size does not match original (src: %s, dest: %s)" % (
|
||||
src_size, dest_size)
|
||||
)
|
||||
|
||||
|
||||
def findFiles(args):
|
||||
"""
|
||||
Takes a list of files/folders, grabs files inside them. Does not recurse
|
||||
@@ -260,9 +150,8 @@ def main():
|
||||
|
||||
# Warn if no files are found, then exit
|
||||
if len(files) == 0:
|
||||
print colour('No files found','red')
|
||||
print 'No files found'
|
||||
sys.exit(0)
|
||||
#end if files == 0
|
||||
|
||||
for name in files:
|
||||
oldfile = os.path.join(name['filepath'], name['filename']) + "." + name['ext']
|
||||
@@ -286,7 +175,6 @@ def main():
|
||||
os.rename(oldfile, newfile)
|
||||
except Exception, errormsg:
|
||||
print "[!] Error moving file! %s" % (errormsg)
|
||||
#end try
|
||||
else:
|
||||
print "[*] Copying file"
|
||||
try:
|
||||
@@ -295,14 +183,8 @@ def main():
|
||||
print "[!] Error copying file! %s" % (errormsg)
|
||||
else:
|
||||
print "[*] ..done"
|
||||
#end try
|
||||
#end if same_partition
|
||||
#end if not file_exists
|
||||
else:
|
||||
print "Skipping file"
|
||||
#end if ans
|
||||
#end for name in files
|
||||
#end main
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
|
||||
Reference in New Issue
Block a user