filesystem-bot/database bot.py

import discord
from discord.ext import commands
from time import sleep
import os
import datetime
import wget
import threading
from zipfile import ZipFile
from pathlib import Path
import shutil


def closefile():
    sleep(300)
    if os.path.exists(finalfilepath):
        os.remove(finalfilepath)
    else:
        pass

def closefile1():
    sleep(300)
    if os.path.exists(finalfilepath1):
        os.remove(finalfilepath1)
    else:
        pass
    


client = commands.Bot(command_prefix='.')


@client.command() 
async def addcharacter(ctx):
    await ctx.send("adding to database please wait")
    try:
        await ctx.message.attachments[0].save("newcharacter.zip")
                                         
        zf = ZipFile('newcharacter.zip', 'r')
        zf.extractall('public/singlecharacter')
        zf.close()
        if os.path.exists("newcharacter.zip"):
            os.remove("newcharacter.zip")
        else:
            pass
        if os.path.exists("public/characters.zip"):
            os.remove("public/characters.zip")
        else:
            pass
        shutil.make_archive('public/characters', 'zip', 'public/singlecharacter')
        await ctx.send("added to database")
    except Exception:
        await ctx.send("something went wrong! this probably means you didn't attach a file or discord api is broken")

@client.command() 
async def addcourtroom(ctx):
    await ctx.send("adding to database please wait")
    try:
        await ctx.message.attachments[0].save("newcourtroom.zip")
                                         
        zf = ZipFile('newcourtroom.zip', 'r')
        zf.extractall('public/singlecourtroom')
        zf.close()
        if os.path.exists("newcourtroom.zip"):
            os.remove("newcourtroom.zip")
        else:
            pass
        if os.path.exists("public/courtroom.zip"):
            os.remove("public/courtroom.zip")
        else:
            pass
        shutil.make_archive('public/courtrooms', 'zip', 'public/singlecourtroom')
        await ctx.send("added to database")
    except Exception:
        await ctx.send("something went wrong! this probably means you didn't attach a file or discord api is broken")

@client.command()     
async def addsound(ctx):
    await ctx.send("adding to database please wait")
    try:
        await ctx.message.attachments[0].save("newsound.zip")
                                         
        zf = ZipFile('newsound.zip', 'r')
        zf.extractall('public/singlesound')
        zf.close()
        if os.path.exists("newsound.zip"):
            os.remove("newsound.zip")
        else:
            pass
        if os.path.exists("public/sounds.zip"):
            os.remove("public/sounds.zip")
        else:
            pass
        shutil.make_archive('public/sounds', 'zip', 'public/singlesound')
        await ctx.send("added to database")
    except Exception:
        await ctx.send("something went wrong! this probably means you didn't attach a file or discord api is broken")

@client.command()     
async def setip(ctx):
    ip = userInput = ctx.message.content[7:]
    with open("public/serverlist.txt", "w+") as hisc:
        hisc.write(str(ip)+":best bois courthouse")
    await ctx.send("changed ip to "+str(ip))

@client.command()     
async def getchrasset(ctx):
    weburl = ctx.message.content[13:]
    filedirweburl = weburl.replace("%20", " ")
    if os.path.isfile('public/singlecharacter/'+str(filedirweburl)):
        betterweburl = weburl.replace(" ", "%20")
        await ctx.send("http://fierce-push.auto.playit.gg:47746/singlecharacter/"+str(betterweburl))
    else:
        await ctx.send("oops! that file does not exist make sure its spelled correctly")

@client.command()     
async def getcortasset(ctx):
    weburl = ctx.message.content[14:]
    filedirweburl = weburl.replace("%20", " ")
    if os.path.isfile('public/singlecharacter/'+str(filedirweburl)):
        betterweburl = weburl.replace(" ", "%20")
        await ctx.send("http://fierce-push.auto.playit.gg:47746/singlecourtroom/"+str(betterweburl))
    else:
        await ctx.send("oops! that file does not exist make sure its spelled correctly")
        
@client.command()     
async def getsound(ctx):
    sound = ctx.message.content[10:]
    sounddirweb = sound.replace("%20", " ")
    if os.path.isfile('public/singlesound/'+str(sounddirweb)):
        bettersound = sound.replace(" ", "%20")
        await ctx.send("http://fierce-push.auto.playit.gg:47746/singlesound/"+str(bettersound))
    else:
        await ctx.send("oops! that file does not exist make sure its spelled correctly")

@client.command()     
async def downloadcharacter(ctx):
    filepath = ctx.message.content[19:]
    fullfilepath = 'public/singlecharacter/'+str(filepath)
    if os.path.isdir(fullfilepath):
      await ctx.send("grabbing the character please wait")
      shutil.make_archive(fullfilepath, 'zip', fullfilepath)
      global finalfilepath
      finalfilepath = str(fullfilepath)+'.zip'
      if Path(fullfilepath).stat().st_size > 3850:
          urlpath = finalfilepath.replace(" ", "_")
          os.rename(finalfilepath,urlpath)
          newurlpath = finalfilepath.replace("/public", "")
          await ctx.send("http://fierce-push.auto.playit.gg:47746/"+str(newurlpath))
          await ctx.send("note: this link will become invalid in 5 minutes")
          delfile = threading.Thread(target=closefile)
          delfile.start()
      else:
          await ctx.send(file=discord.File(str(finalfilepath)))  
          if os.path.exists(finalfilepath):
             os.remove(finalfilepath)
          else:
            pass
    else:
        await ctx.send("sorry, but thats not a character in my database")

@client.command()     
async def downloadcourtroom(ctx):
    filepath = ctx.message.content[19:]
    fullfilepath = 'public/singlecourtroom/'+str(filepath)
    if os.path.isdir(fullfilepath):
      await ctx.send("grabbing the courtroom please wait")
      shutil.make_archive(fullfilepath, 'zip', fullfilepath)
      global finalfilepath1
      finalfilepath1 = str(fullfilepath)+'.zip'
      if Path(fullfilepath1).stat().st_size > 3850:
          urlpath = finalfilepath1.replace(" ", "_")
          os.rename(finalfilepath1,urlpath)
          newurlpath = finalfilepath1.replace("/public", "")
          await ctx.send("http://fierce-push.auto.playit.gg:47746/"+str(newurlpath))
          await ctx.send("note: this link will become invalid in 5 minutes")
          delfile = threading.Thread(target=closefile1)
          delfile.start()
      else:
          await ctx.send(file=discord.File(str(finalfilepath)))  
          if os.path.exists(finalfilepath):
             os.remove(finalfilepath)
          else:
            pass
    else:
        await ctx.send("sorry, but thats not a courtroom in my database")

@client.command()
async def listchr(ctx):
    findchr = os.listdir('public/singlecharacter')
    await ctx.send('```'+str(findchr)+'```')

@client.command()
async def listcourt(ctx):
    findchr = os.listdir('public/singlecourtroom')
    await ctx.send('```'+str(findchr)+'```')

@client.command()
async def listsound(ctx):
    soutype = ctx.message.content[11:]
    findsou1 = 'public/singlesound/'+str(soutype)
    if os.path.isdir(findsou1):
        findsou = os.listdir('public/singlesound/'+str(soutype))
        soulenght = len(str(findsou))
        if soulenght > 1990:
            with open("public/soulist.txt", "w+") as souname:
                souname.write(str(findsou))
            await ctx.send(file=discord.File("public/soulist.txt"))  
            if os.path.exists("public/soulist.txt"):
                os.remove("public/soulist.txt")
            else:
                pass
        else:
            await ctx.send('```'+str(findsou)+'```')
    else:
        await ctx.send('thats not a sound option avalible options are "sound", and "general"')

    
client.run('botid')