This commit is contained in:
NikolajDanger
2021-02-01 23:42:58 +01:00
parent 52c84d7ba5
commit 03ba417b7f
2 changed files with 1 additions and 70 deletions

View File

@ -136,11 +136,6 @@ class MiscCog(commands.Cog):
async def downloading(self,ctx):
await self.bedreNetflix.downloading(ctx)
#Calculates how much space is used on the server
@commands.command(aliases = ["dirsize","storage"])
async def directorySizes(self,ctx):
await self.bedreNetflix.directorySizes(ctx)
#Looks up on Wolfram Alpha
@commands.command()
async def wolf(self, ctx, *, content):

View File

@ -1,4 +1,4 @@
import requests, imdb, discord, json, math, time, asyncio, os, concurrent.futures
import requests, imdb, discord, json, math, time, asyncio
from funcs import logThis
radarrURL = "http://localhost:7878/api/v3/"
@ -7,32 +7,6 @@ qbittorrentURL = "http://localhost:1340/api/v2/"
moviePath = "/media/plex/Server/movies/"
showPath = "/media/plex/Server/Shows/"
def getSize(startPath,pathLength):
logThis("Getting the size of "+startPath)
totalSize = 0
for dirpath, dirnames, filenames in os.walk(startPath):
for f in filenames:
fp = os.path.join(dirpath, f)
# skip if it is symbolic link
if not os.path.islink(fp):
print(fp[:60],end="\r")
totalSize += os.path.getsize(fp)
return totalSize
async def formatSize(size):
if size >= 1024**4:
return ("%.1f" % (size/(1024**4))) + "TB"
elif size >= 1024**3:
return ("%.1f" % (size/(1024**3))) + "GB"
elif size >= 1024**2:
return ("%.1f" % (size/(1024**2))) + "MB"
elif size >= 1024:
return ("%.1f" % (size/(1024))) + "KB"
else:
return str(size)+"b"
class BedreNetflix():
def __init__(self,bot):
self.bot = bot
@ -276,41 +250,3 @@ class BedreNetflix():
else:
await ctx.send(messageText)
async def directorySizes(self,ctx):
logThis("Calculating directory size")
message = await ctx.send("```Calculating directory size. This might take a while.```")
startPath = ".."
pathLength = len(startPath)
directories = []
for f in os.listdir(startPath):
fp = os.path.join(startPath, f)
if os.path.isfile(fp):
size = os.path.getsize(fp)
else:
with concurrent.futures.ThreadPoolExecutor() as executor:
future = executor.submit(getSize,fp,pathLength)
size = future.result()
directories.append((str(f),size))
directories.sort(key=lambda tup: tup[1])
directories.reverse()
returnText = "```Calculating directory size. This might take a while.\n"
total = 0
for directory, size in directories:
total += size
sizeText = await formatSize(size)
returnText += "\n"+directory+": "+sizeText
returnText += "\nTotal: "+await formatSize(total)+"```"
await message.edit(content=returnText)
directories.sort(key=lambda tup: tup[1])
directories.reverse()
returnText = "```"
total = 0
for directory, size in directories:
total += size
sizeText = await formatSize(size)
returnText += "\n"+directory+": "+sizeText
returnText += "\nTotal: "+await formatSize(total)+"```"
logThis("Returning final result of dirsize")
await message.edit(content=returnText)