mirror of
https://github.com/hastagAB/Awesome-Python-Scripts.git
synced 2024-11-23 20:11:07 +00:00
Added File Sharing Telegram Bot
This commit is contained in:
parent
537132d254
commit
52193b64ff
21
File-Sharing-Bot/LICENSE
Normal file
21
File-Sharing-Bot/LICENSE
Normal file
|
@ -0,0 +1,21 @@
|
||||||
|
MIT License
|
||||||
|
|
||||||
|
Copyright (c) 2018 Darshan Patel
|
||||||
|
|
||||||
|
Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||||
|
of this software and associated documentation files (the "Software"), to deal
|
||||||
|
in the Software without restriction, including without limitation the rights
|
||||||
|
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||||
|
copies of the Software, and to permit persons to whom the Software is
|
||||||
|
furnished to do so, subject to the following conditions:
|
||||||
|
|
||||||
|
The above copyright notice and this permission notice shall be included in all
|
||||||
|
copies or substantial portions of the Software.
|
||||||
|
|
||||||
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||||
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||||
|
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||||
|
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||||
|
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||||
|
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||||
|
SOFTWARE.
|
1
File-Sharing-Bot/Procfile
Normal file
1
File-Sharing-Bot/Procfile
Normal file
|
@ -0,0 +1 @@
|
||||||
|
worker: python bot.py
|
5
File-Sharing-Bot/README.md
Normal file
5
File-Sharing-Bot/README.md
Normal file
|
@ -0,0 +1,5 @@
|
||||||
|
# File-Sharing-Bot
|
||||||
|
File Sharing telegram Bot developed in Python
|
||||||
|
It is like a centerlized file repository where authorized users can share files and files are available to all the users.
|
||||||
|
Functionalities and commands can be seen using /help command.
|
||||||
|
This bot can be directly hosted on Heroku.
|
207
File-Sharing-Bot/bot.py
Normal file
207
File-Sharing-Bot/bot.py
Normal file
|
@ -0,0 +1,207 @@
|
||||||
|
from telegram.ext import Updater, CommandHandler, MessageHandler, Filters
|
||||||
|
import logging
|
||||||
|
import os
|
||||||
|
import telegram
|
||||||
|
import shutil
|
||||||
|
# Enable logging
|
||||||
|
logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
|
||||||
|
level=logging.INFO)
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
#list of authorized users
|
||||||
|
#create a list of telegram usernames to authorise them, 0th username is admin.
|
||||||
|
username_list = []
|
||||||
|
# Define a few command handlers. These usually take the two arguments bot and
|
||||||
|
# update. Error handlers also receive the raised TelegramError object in error.
|
||||||
|
def start(bot, update):
|
||||||
|
"""Send a message when the command /start is issued."""
|
||||||
|
reply = "Welcome to World of Automation. \nI am a bot developed by a Lazy Programmer.\nSend /help command to see what i can do."
|
||||||
|
update.message.reply_text(reply)
|
||||||
|
|
||||||
|
|
||||||
|
def help(bot, update):
|
||||||
|
"""Send a message when the command /help is issued."""
|
||||||
|
admin = update.message.from_user.username
|
||||||
|
if admin == username_list[0]:
|
||||||
|
reply = '''Send /get folder_name/file_name.extension to receive a file.
|
||||||
|
\nSend /ls folder_name to show list of files.
|
||||||
|
\nSend /put folder_name/file_name.extension to upload last sent file.
|
||||||
|
\nSend /mkdir folder_name to create a Folder.
|
||||||
|
\nSend /remove folder_name/filename.extension to delete a file.
|
||||||
|
\nSend /adduser username to give access.
|
||||||
|
\nSend /removeuser username to revoke access.
|
||||||
|
\nSend /showuser to show list of users
|
||||||
|
'''
|
||||||
|
else:
|
||||||
|
reply = '''Send /get folder_name/file_name.extension to receive a file.
|
||||||
|
\nSend /ls folder_name to show list of files.
|
||||||
|
\nSend /put folder_name/file_name.extension to upload last sent file.
|
||||||
|
\nSend /mkdir folder_name to create a Folder.
|
||||||
|
'''
|
||||||
|
update.message.reply_text(reply)
|
||||||
|
|
||||||
|
|
||||||
|
def get(bot, update):
|
||||||
|
"""Send requested file."""
|
||||||
|
username = update.message.from_user.username
|
||||||
|
if(username not in username_list):
|
||||||
|
update.message.reply_text("You are not Authorized.")
|
||||||
|
return
|
||||||
|
file = update.message.text.split(" ")[-1]
|
||||||
|
if(file == "/send"):
|
||||||
|
update.message.reply_text("Invalid File name.")
|
||||||
|
else:
|
||||||
|
reply = "Findind and Sending a requested file to you. Hold on..."
|
||||||
|
update.message.reply_text(reply)
|
||||||
|
path = os.getcwd()+'/'+file
|
||||||
|
if (os.path.exists(path)):
|
||||||
|
bot.send_document(chat_id=update.message.chat_id,document=open(path, 'rb'), timeout = 100)
|
||||||
|
else:
|
||||||
|
update.message.reply_text("File not Found.")
|
||||||
|
|
||||||
|
def ls(bot, update):
|
||||||
|
"""Show files in requested directory."""
|
||||||
|
username = update.message.from_user.username
|
||||||
|
if(username not in username_list):
|
||||||
|
update.message.reply_text("You are not Authorized.")
|
||||||
|
return
|
||||||
|
file = update.message.text.split(" ")[-1]
|
||||||
|
if(file == "/show"):
|
||||||
|
update.message.reply_text("Invalid Directory name.")
|
||||||
|
else:
|
||||||
|
reply = "Findind and Sending a list of files to you. Hold on..."
|
||||||
|
update.message.reply_text(reply)
|
||||||
|
path = os.getcwd()+'/'+file
|
||||||
|
if (os.path.exists(path)):
|
||||||
|
update.message.reply_text(os.listdir(path))
|
||||||
|
else:
|
||||||
|
update.message.reply_text("Directory not Found.")
|
||||||
|
|
||||||
|
def put(bot, update):
|
||||||
|
f = open(str(os.getcwd())+"/file", "r")
|
||||||
|
file_id = f.read()
|
||||||
|
f.close
|
||||||
|
if file_id == "":
|
||||||
|
update.message.reply_text("You didn't upload file.")
|
||||||
|
else:
|
||||||
|
new_file = bot.get_file(file_id)
|
||||||
|
message = update.message.text.split(" ")
|
||||||
|
path = message[-1]
|
||||||
|
if len(path) < 1:
|
||||||
|
update.message.reply_text("Enter Path correctly.")
|
||||||
|
else:
|
||||||
|
new_file.download(os.getcwd()+'/'+path)
|
||||||
|
update.message.reply_text("File Stored.")
|
||||||
|
|
||||||
|
|
||||||
|
def mkdir(bot, update):
|
||||||
|
message = update.message.text.split(" ")
|
||||||
|
if len(message) < 1 or message[-1] == "/mkdir":
|
||||||
|
update.message.reply_text("Invalid Syntax. Refer syntax in help section.")
|
||||||
|
return
|
||||||
|
path = os.getcwd() + "/" + message[-1]
|
||||||
|
os.mkdir(path)
|
||||||
|
update.message.reply_text("Folder Created.")
|
||||||
|
|
||||||
|
|
||||||
|
def echo(bot, update):
|
||||||
|
"""Echo the user message."""
|
||||||
|
if update.message.document:
|
||||||
|
file_id = update.message.document.file_id
|
||||||
|
f = open(str(os.getcwd())+"/file", "w")
|
||||||
|
f.write(file_id)
|
||||||
|
f.close
|
||||||
|
update.message.reply_text("Received.Now send file name and location to store. using /put command")
|
||||||
|
else:
|
||||||
|
reply = "Invalid Input."
|
||||||
|
update.message.reply_text(reply)
|
||||||
|
|
||||||
|
def error(bot, update, error):
|
||||||
|
"""Log Errors caused by Updates."""
|
||||||
|
logger.warning('Update "%s" caused error "%s"', update, error)
|
||||||
|
|
||||||
|
def add_user(bot, update):
|
||||||
|
admin = update.message.from_user.username
|
||||||
|
if admin == username_list[0]:
|
||||||
|
username = update.message.text.split(" ")[-1]
|
||||||
|
username_list.append(username)
|
||||||
|
update.message.reply_text("User added.")
|
||||||
|
else:
|
||||||
|
update.message.reply_text("You are not Authorized.")
|
||||||
|
|
||||||
|
def show_user(bot, update):
|
||||||
|
admin = update.message.from_user.username
|
||||||
|
if admin == username_list[0]:
|
||||||
|
update.message.reply_text(username_list)
|
||||||
|
else:
|
||||||
|
update.message.reply_text("You are not Authorized.")
|
||||||
|
|
||||||
|
def remove_user(bot, update):
|
||||||
|
admin = update.message.from_user.username
|
||||||
|
if admin == username_list[0]:
|
||||||
|
username = update.message.text.split(" ")[-1]
|
||||||
|
username_list.remove(username)
|
||||||
|
update.message.reply_text("User Removed.")
|
||||||
|
else:
|
||||||
|
update.message.reply_text("You are not Authorized.")
|
||||||
|
|
||||||
|
def remove(bot, update):
|
||||||
|
admin = update.message.from_user.username
|
||||||
|
if admin == username_list[0]:
|
||||||
|
filename = update.message.text.split(" ")[-1]
|
||||||
|
os.remove(os.getcwd()+ "/" + filename)
|
||||||
|
update.message.reply_text("File Removed.")
|
||||||
|
else:
|
||||||
|
update.message.reply_text("You are not Authorized.")
|
||||||
|
|
||||||
|
def rmdir(bot, update):
|
||||||
|
admin = update.message.from_user.username
|
||||||
|
if admin == username_list[0]:
|
||||||
|
filename = update.message.text.split(" ")[-1]
|
||||||
|
shutil.rmtree(os.getcwd()+ "/" + filename)
|
||||||
|
update.message.reply_text("Folder Removed.")
|
||||||
|
else:
|
||||||
|
update.message.reply_text("You are not Authorized.")
|
||||||
|
|
||||||
|
def main():
|
||||||
|
"""Start the bot."""
|
||||||
|
# Create the EventHandler and pass it your bot's token.
|
||||||
|
TOKEN = os.environ['TOKEN']
|
||||||
|
updater = Updater(TOKEN)
|
||||||
|
|
||||||
|
# Get the dispatcher to register handlers
|
||||||
|
dp = updater.dispatcher
|
||||||
|
|
||||||
|
# on different commands - answer in Telegram
|
||||||
|
dp.add_handler(CommandHandler("start", start))
|
||||||
|
dp.add_handler(CommandHandler("help", help))
|
||||||
|
dp.add_handler(CommandHandler("get", get))
|
||||||
|
dp.add_handler(CommandHandler("ls", ls))
|
||||||
|
dp.add_handler(CommandHandler("put", put))
|
||||||
|
dp.add_handler(CommandHandler("mkdir", mkdir))
|
||||||
|
|
||||||
|
#admin functionalities
|
||||||
|
dp.add_handler(CommandHandler("adduser", add_user))
|
||||||
|
dp.add_handler(CommandHandler("showuser", show_user))
|
||||||
|
dp.add_handler(CommandHandler("removeUser", remove_user))
|
||||||
|
dp.add_handler(CommandHandler("remove", remove))
|
||||||
|
dp.add_handler(CommandHandler("rmdir", rmdir))
|
||||||
|
|
||||||
|
# on noncommand i.e message - echo the message on Telegram
|
||||||
|
dp.add_handler(MessageHandler(Filters.document, echo))
|
||||||
|
|
||||||
|
# log all errors
|
||||||
|
dp.add_error_handler(error)
|
||||||
|
|
||||||
|
# Start the Bot
|
||||||
|
updater.start_polling()
|
||||||
|
|
||||||
|
# Run the bot until you press Ctrl-C or the process receives SIGINT,
|
||||||
|
# SIGTERM or SIGABRT. This should be used most of the time, since
|
||||||
|
# start_polling() is non-blocking and will stop the bot gracefully.
|
||||||
|
updater.idle()
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
main()
|
1
File-Sharing-Bot/requirements.txt
Normal file
1
File-Sharing-Bot/requirements.txt
Normal file
|
@ -0,0 +1 @@
|
||||||
|
python-telegram-bot==10.1.0
|
1
File-Sharing-Bot/runtime.txt
Normal file
1
File-Sharing-Bot/runtime.txt
Normal file
|
@ -0,0 +1 @@
|
||||||
|
python-3.6.0
|
Loading…
Reference in New Issue
Block a user