mirror of
https://github.com/SpiceBot/SpiceBot.git
synced 2026-11-11 06:00:59 -05:00
96 lines
3.4 KiB
Python
96 lines
3.4 KiB
Python
#!/usr/bin/env python
|
|
# coding=utf-8
|
|
from __future__ import unicode_literals, absolute_import, print_function, division
|
|
|
|
# sopel imports
|
|
import sopel.module
|
|
from sopel.config.types import StaticSection, ValidatedAttribute
|
|
|
|
import os
|
|
import pip
|
|
|
|
import spicemanip
|
|
|
|
from sopel_modules.SpiceBot_SBTools import service_manip, sopel_triggerargs, command_permissions_check, bot_logging, spicebot_reload
|
|
|
|
|
|
class SpiceBot_Update_MainSection(StaticSection):
|
|
gitrepo = ValidatedAttribute('gitrepo', default="https://github.com/SpiceBot/SpiceBot")
|
|
gitbranch = ValidatedAttribute('gitbranch', default="master")
|
|
|
|
|
|
def configure(config):
|
|
config.define_section("SpiceBot_Update", SpiceBot_Update_MainSection, validate=False)
|
|
config.SpiceBot_Update.configure_setting('gitrepo', 'SpiceBot_Update git repo to install')
|
|
config.SpiceBot_Update.configure_setting('gitbranch', 'SpiceBot_Update git branch to install')
|
|
|
|
|
|
def setup(bot):
|
|
bot_logging(bot, 'SpiceBot_Update', "Initial Setup processing")
|
|
bot.config.define_section("SpiceBot_Update", SpiceBot_Update_MainSection, validate=False)
|
|
|
|
|
|
def shutdown(bot):
|
|
pass
|
|
|
|
|
|
@sopel.module.nickname_commands('update')
|
|
def nickname_comand_update(bot, trigger):
|
|
|
|
if not command_permissions_check(bot, trigger, ['admins', 'owner', 'OP', 'ADMIN', 'OWNER']):
|
|
bot.say("I was unable to process this Bot Nick command due to privilege issues.")
|
|
return
|
|
|
|
triggerargs, triggercommand = sopel_triggerargs(bot, trigger, 'nickname_command')
|
|
|
|
if not len(triggerargs):
|
|
commandused = 'nodeps'
|
|
else:
|
|
commandused = spicemanip.main(triggerargs, 1).lower()
|
|
|
|
if commandused not in ['deps', 'nodeps']:
|
|
bot.say("Please specify deps or nodeps")
|
|
return
|
|
|
|
triggerargs = spicemanip.main(triggerargs, '2+', 'list')
|
|
|
|
bot_logging(bot, 'SpiceBot_Update', "Received command from " + trigger.nick + " to update from Github and restart")
|
|
bot.osd("Received command from " + trigger.nick + " to update from Github and restart. Be Back Soon!", bot.channels.keys())
|
|
|
|
if commandused == 'nodeps':
|
|
spicebot_update(bot, "False")
|
|
if commandused == 'deps':
|
|
spicebot_update(bot, "True")
|
|
|
|
# service_manip(bot, bot.nick, 'restart', 'SpiceBot_Update')
|
|
spicebot_reload(bot, 'SpiceBot_Update')
|
|
|
|
|
|
def spicebot_update(bot, deps="False"):
|
|
|
|
pipcommand = "sudo pip3 install --upgrade"
|
|
if deps == "False":
|
|
pipcommand += " --no-deps"
|
|
pipcommand += " --force-reinstall"
|
|
pipcommand += " git+" + str(bot.config.SpiceBot_Update.gitrepo) + "@" + str(bot.config.SpiceBot_Update.gitbranch)
|
|
|
|
bot_logging(bot, 'SpiceBot_Update', "Running `" + pipcommand + "`")
|
|
for line in os.popen(pipcommand).read().split('\n'):
|
|
bot_logging(bot, 'SpiceBot_Update', " " + line)
|
|
|
|
stock_modules_begone(bot)
|
|
|
|
|
|
def stock_modules_begone(bot):
|
|
|
|
# Remove stock modules, if present
|
|
main_sopel_dir = os.path.dirname(os.path.abspath(sopel.__file__))
|
|
modules_dir = os.path.join(main_sopel_dir, 'modules')
|
|
stockdir = os.path.join(modules_dir, "stock")
|
|
if not os.path.exists(stockdir) or not os.path.isdir(stockdir):
|
|
os.system("sudo mkdir " + stockdir)
|
|
for pathname in os.listdir(modules_dir):
|
|
path = os.path.join(modules_dir, pathname)
|
|
if (os.path.isfile(path) and path.endswith('.py') and not path.startswith('_')):
|
|
os.system("sudo mv " + path + " " + stockdir)
|