bobloy
6 years ago
parent
6cc32a937c
commit
75aab626f9
@ -1,22 +0,0 @@
|
|||||||
from .fight import Fight
|
|
||||||
|
|
||||||
# def check_folders():
|
|
||||||
# if not os.path.exists("data/Fox-Cogs"):
|
|
||||||
# print("Creating data/Fox-Cogs folder...")
|
|
||||||
# os.makedirs("data/Fox-Cogs")
|
|
||||||
|
|
||||||
# if not os.path.exists("data/Fox-Cogs/fight"):
|
|
||||||
# print("Creating data/Fox-Cogs/fight folder...")
|
|
||||||
# os.makedirs("data/Fox-Cogs/fight")
|
|
||||||
|
|
||||||
|
|
||||||
# def check_files():
|
|
||||||
# if not dataIO.is_valid_json("data/Fox-Cogs/fight/fight.json"):
|
|
||||||
# dataIO.save_json("data/Fox-Cogs/fight/fight.json", {})
|
|
||||||
|
|
||||||
|
|
||||||
def setup(bot):
|
|
||||||
# check_folders()
|
|
||||||
# check_files()
|
|
||||||
n = Fight(bot)
|
|
||||||
bot.add_cog(n)
|
|
||||||
File diff suppressed because it is too large
Load Diff
@ -1,10 +0,0 @@
|
|||||||
{
|
|
||||||
"author" : ["Bobloy"],
|
|
||||||
"bot_version" : [3,0,0],
|
|
||||||
"description" : "[Incomplete] Cog to organize tournaments within Discord",
|
|
||||||
"hidden" : true,
|
|
||||||
"install_msg" : "Thank you for installing Fight. Run with [p]fight or [p]fightset",
|
|
||||||
"requirements" : [],
|
|
||||||
"short" : "[Incomplete] Cog to organize tournaments",
|
|
||||||
"tags" : ["game", "fun", "fight", "tournament", "tourney", "elimination", "bracket", "bobloy"]
|
|
||||||
}
|
|
||||||
@ -1,5 +0,0 @@
|
|||||||
from .howdoi import Howdoi
|
|
||||||
|
|
||||||
|
|
||||||
def setup(bot):
|
|
||||||
bot.add_cog(Howdoi(bot))
|
|
||||||
@ -1,85 +0,0 @@
|
|||||||
import discord
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
from .utils.chat_formatting import pagify
|
|
||||||
from .utils.chat_formatting import box
|
|
||||||
|
|
||||||
from .howdoi_source import howdoi as hdi, Answer
|
|
||||||
|
|
||||||
|
|
||||||
class Howdoi:
|
|
||||||
"""Cog for answering coding questions"""
|
|
||||||
|
|
||||||
def __init__(self, bot):
|
|
||||||
self.bot = bot
|
|
||||||
self.query = ""
|
|
||||||
self.config = Config.get_conf(self, identifier=72111119100111105)
|
|
||||||
default_global = {
|
|
||||||
"query": "",
|
|
||||||
"pos": 1,
|
|
||||||
"all": False,
|
|
||||||
"link": True,
|
|
||||||
"color": False,
|
|
||||||
"num_answers": 1,
|
|
||||||
"clear_cache": False,
|
|
||||||
"version": False
|
|
||||||
}
|
|
||||||
|
|
||||||
self.config.register_global(**default_global)
|
|
||||||
|
|
||||||
@commands.group(pass_context=True)
|
|
||||||
async def howdoiset(self, ctx):
|
|
||||||
"""Adjust howdoi settings
|
|
||||||
Settings are reset on reload"""
|
|
||||||
if ctx.invoked_subcommand is None:
|
|
||||||
pass
|
|
||||||
|
|
||||||
@howdoiset.command(pass_context=True, name="answers")
|
|
||||||
async def howdoiset_answers(self, ctx, num_answers: int=1):
|
|
||||||
"""Adjust number of answers provided.
|
|
||||||
Defaults to 1"""
|
|
||||||
|
|
||||||
await self.config.num_answers.set(num_answers)
|
|
||||||
await ctx.send("Number of answers provided will now be {}".format(num_answers))
|
|
||||||
|
|
||||||
@howdoiset.command(pass_context=True, name="link")
|
|
||||||
async def howdoiset_link(self, ctx):
|
|
||||||
"""Toggles providing in-line answers or a link
|
|
||||||
Default On"""
|
|
||||||
|
|
||||||
await self.config.link.set(not (await self.config.link()))
|
|
||||||
|
|
||||||
if await self.config.link():
|
|
||||||
await ctx.send("Answers will now be provided as a link")
|
|
||||||
else:
|
|
||||||
await ctx.send("Answers will now be provided as the response")
|
|
||||||
|
|
||||||
@howdoiset.command(pass_context=True, name="full")
|
|
||||||
async def howdoiset_full(self, ctx):
|
|
||||||
"""Toggles providing full answers or just first code found
|
|
||||||
Default Off
|
|
||||||
Only works if links are turned off"""
|
|
||||||
|
|
||||||
await self.config.all.set(not (await self.config.all()))
|
|
||||||
|
|
||||||
if await self.config.all():
|
|
||||||
await ctx.send("Answers will now be provided in full context")
|
|
||||||
else:
|
|
||||||
await ctx.send("Answers will now be provided as a code snippet")
|
|
||||||
|
|
||||||
@commands.command(pass_context=True)
|
|
||||||
async def howdoi(self, ctx, *question):
|
|
||||||
"""Ask a coding question"""
|
|
||||||
self.query = " ".join(question)
|
|
||||||
|
|
||||||
await self.config.query.set(self.query)
|
|
||||||
|
|
||||||
out = hdi.howdoi(await self.config.all()) # .encode('utf-8', 'ignore')
|
|
||||||
|
|
||||||
if await self.config.links():
|
|
||||||
await ctx.send(out)
|
|
||||||
else:
|
|
||||||
await ctx.send(box(out,"python"))
|
|
||||||
# for page in pagify(out, shorten_by=24):
|
|
||||||
# await ctx.send(box(page))
|
|
||||||
@ -1,322 +0,0 @@
|
|||||||
#!/usr/bin/env python
|
|
||||||
|
|
||||||
######################################################
|
|
||||||
#
|
|
||||||
# howdoi - instant coding answers via the command line
|
|
||||||
# written by Benjamin Gleitzman (gleitz@mit.edu)
|
|
||||||
# inspired by Rich Jones (rich@anomos.info)
|
|
||||||
#
|
|
||||||
######################################################
|
|
||||||
|
|
||||||
import argparse
|
|
||||||
import glob
|
|
||||||
import os
|
|
||||||
import random
|
|
||||||
import re
|
|
||||||
import requests
|
|
||||||
import requests_cache
|
|
||||||
import sys
|
|
||||||
from . import __version__
|
|
||||||
|
|
||||||
from pygments import highlight
|
|
||||||
from pygments.lexers import guess_lexer, get_lexer_by_name
|
|
||||||
from pygments.formatters.terminal import TerminalFormatter
|
|
||||||
from pygments.util import ClassNotFound
|
|
||||||
|
|
||||||
from pyquery import PyQuery as pq
|
|
||||||
from requests.exceptions import ConnectionError
|
|
||||||
from requests.exceptions import SSLError
|
|
||||||
|
|
||||||
# Handle imports for Python 2 and 3
|
|
||||||
if sys.version < '3':
|
|
||||||
import codecs
|
|
||||||
from urllib import quote as url_quote
|
|
||||||
from urllib import getproxies
|
|
||||||
|
|
||||||
# Handling Unicode: http://stackoverflow.com/a/6633040/305414
|
|
||||||
def u(x):
|
|
||||||
return codecs.unicode_escape_decode(x)[0]
|
|
||||||
else:
|
|
||||||
from urllib.request import getproxies
|
|
||||||
from urllib.parse import quote as url_quote
|
|
||||||
|
|
||||||
def u(x):
|
|
||||||
return x
|
|
||||||
|
|
||||||
|
|
||||||
if os.getenv('HOWDOI_DISABLE_SSL'): # Set http instead of https
|
|
||||||
SCHEME = 'http://'
|
|
||||||
VERIFY_SSL_CERTIFICATE = False
|
|
||||||
else:
|
|
||||||
SCHEME = 'https://'
|
|
||||||
VERIFY_SSL_CERTIFICATE = True
|
|
||||||
|
|
||||||
URL = os.getenv('HOWDOI_URL') or 'stackoverflow.com'
|
|
||||||
|
|
||||||
USER_AGENTS = ('Mozilla/5.0 (Macintosh; Intel Mac OS X 10.7; rv:11.0) Gecko/20100101 Firefox/11.0',
|
|
||||||
'Mozilla/5.0 (X11; Ubuntu; Linux x86_64; rv:22.0) Gecko/20100 101 Firefox/22.0',
|
|
||||||
'Mozilla/5.0 (Windows NT 6.1; rv:11.0) Gecko/20100101 Firefox/11.0',
|
|
||||||
('Mozilla/5.0 (Macintosh; Intel Mac OS X 10_7_4) AppleWebKit/536.5 (KHTML, like Gecko) '
|
|
||||||
'Chrome/19.0.1084.46 Safari/536.5'),
|
|
||||||
('Mozilla/5.0 (Windows; Windows NT 6.1) AppleWebKit/536.5 (KHTML, like Gecko) Chrome/19.0.1084.46'
|
|
||||||
'Safari/536.5'), )
|
|
||||||
SEARCH_URLS = {
|
|
||||||
'bing': SCHEME + 'www.bing.com/search?q=site:{0}%20{1}',
|
|
||||||
'google': SCHEME + 'www.google.com/search?q=site:{0}%20{1}'
|
|
||||||
}
|
|
||||||
STAR_HEADER = u('\u2605')
|
|
||||||
ANSWER_HEADER = u('{2} Answer from {0} {2}\n{1}')
|
|
||||||
NO_ANSWER_MSG = '< no answer given >'
|
|
||||||
XDG_CACHE_DIR = os.environ.get('XDG_CACHE_HOME',
|
|
||||||
os.path.join(os.path.expanduser('~'), '.cache'))
|
|
||||||
CACHE_DIR = os.path.join(XDG_CACHE_DIR, 'howdoi')
|
|
||||||
CACHE_FILE = os.path.join(CACHE_DIR, 'cache{0}'.format(
|
|
||||||
sys.version_info[0] if sys.version_info[0] == 3 else ''))
|
|
||||||
howdoi_session = requests.session()
|
|
||||||
|
|
||||||
|
|
||||||
def get_proxies():
|
|
||||||
proxies = getproxies()
|
|
||||||
filtered_proxies = {}
|
|
||||||
for key, value in proxies.items():
|
|
||||||
if key.startswith('http'):
|
|
||||||
if not value.startswith('http'):
|
|
||||||
filtered_proxies[key] = 'http://%s' % value
|
|
||||||
else:
|
|
||||||
filtered_proxies[key] = value
|
|
||||||
return filtered_proxies
|
|
||||||
|
|
||||||
|
|
||||||
def _get_result(url):
|
|
||||||
try:
|
|
||||||
return howdoi_session.get(url, headers={'User-Agent': random.choice(USER_AGENTS)}, proxies=get_proxies(),
|
|
||||||
verify=VERIFY_SSL_CERTIFICATE).text
|
|
||||||
except requests.exceptions.SSLError as e:
|
|
||||||
print('[ERROR] Encountered an SSL Error. Try using HTTP instead of '
|
|
||||||
'HTTPS by setting the environment variable "HOWDOI_DISABLE_SSL".\n')
|
|
||||||
raise e
|
|
||||||
|
|
||||||
|
|
||||||
def _get_text(element):
|
|
||||||
''' return inner text in pyquery element '''
|
|
||||||
return element.text(squash_space=False)
|
|
||||||
|
|
||||||
|
|
||||||
def _extract_links_from_bing(html):
|
|
||||||
html.remove_namespaces()
|
|
||||||
return [a.attrib['href'] for a in html('.b_algo')('h2')('a')]
|
|
||||||
|
|
||||||
|
|
||||||
def _extract_links_from_google(html):
|
|
||||||
return [a.attrib['href'] for a in html('.l')] or \
|
|
||||||
[a.attrib['href'] for a in html('.r')('a')]
|
|
||||||
|
|
||||||
|
|
||||||
def _extract_links(html, search_engine):
|
|
||||||
if search_engine == 'bing':
|
|
||||||
return _extract_links_from_bing(html)
|
|
||||||
return _extract_links_from_google(html)
|
|
||||||
|
|
||||||
|
|
||||||
def _get_search_url(search_engine):
|
|
||||||
return SEARCH_URLS.get(search_engine, SEARCH_URLS['google'])
|
|
||||||
|
|
||||||
|
|
||||||
def _get_links(query):
|
|
||||||
search_engine = os.getenv('HOWDOI_SEARCH_ENGINE', 'google')
|
|
||||||
search_url = _get_search_url(search_engine)
|
|
||||||
|
|
||||||
result = _get_result(search_url.format(URL, url_quote(query)))
|
|
||||||
html = pq(result)
|
|
||||||
return _extract_links(html, search_engine)
|
|
||||||
|
|
||||||
|
|
||||||
def get_link_at_pos(links, position):
|
|
||||||
if not links:
|
|
||||||
return False
|
|
||||||
|
|
||||||
if len(links) >= position:
|
|
||||||
link = links[position - 1]
|
|
||||||
else:
|
|
||||||
link = links[-1]
|
|
||||||
return link
|
|
||||||
|
|
||||||
|
|
||||||
def _format_output(code, args):
|
|
||||||
if not args['color']:
|
|
||||||
return code
|
|
||||||
lexer = None
|
|
||||||
|
|
||||||
# try to find a lexer using the StackOverflow tags
|
|
||||||
# or the query arguments
|
|
||||||
for keyword in args['query'].split() + args['tags']:
|
|
||||||
try:
|
|
||||||
lexer = get_lexer_by_name(keyword)
|
|
||||||
break
|
|
||||||
except ClassNotFound:
|
|
||||||
pass
|
|
||||||
|
|
||||||
# no lexer found above, use the guesser
|
|
||||||
if not lexer:
|
|
||||||
try:
|
|
||||||
lexer = guess_lexer(code)
|
|
||||||
except ClassNotFound:
|
|
||||||
return code
|
|
||||||
|
|
||||||
return highlight(code,
|
|
||||||
lexer,
|
|
||||||
TerminalFormatter(bg='dark'))
|
|
||||||
|
|
||||||
|
|
||||||
def _is_question(link):
|
|
||||||
return re.search('questions/\d+/', link)
|
|
||||||
|
|
||||||
|
|
||||||
def _get_questions(links):
|
|
||||||
return [link for link in links if _is_question(link)]
|
|
||||||
|
|
||||||
|
|
||||||
def _get_answer(args, links):
|
|
||||||
link = get_link_at_pos(links, args['pos'])
|
|
||||||
if not link:
|
|
||||||
return False
|
|
||||||
if args.get('link'):
|
|
||||||
return link
|
|
||||||
page = _get_result(link + '?answertab=votes')
|
|
||||||
html = pq(page)
|
|
||||||
|
|
||||||
first_answer = html('.answer').eq(0)
|
|
||||||
instructions = first_answer.find('pre') or first_answer.find('code')
|
|
||||||
args['tags'] = [t.text for t in html('.post-tag')]
|
|
||||||
|
|
||||||
if not instructions and not args['all']:
|
|
||||||
text = _get_text(first_answer.find('.post-text').eq(0))
|
|
||||||
elif args['all']:
|
|
||||||
texts = []
|
|
||||||
for html_tag in first_answer.items('.post-text > *'):
|
|
||||||
current_text = _get_text(html_tag)
|
|
||||||
if current_text:
|
|
||||||
if html_tag[0].tag in ['pre', 'code']:
|
|
||||||
texts.append(_format_output(current_text, args))
|
|
||||||
else:
|
|
||||||
texts.append(current_text)
|
|
||||||
text = '\n'.join(texts)
|
|
||||||
else:
|
|
||||||
text = _format_output(_get_text(instructions.eq(0)), args)
|
|
||||||
if text is None:
|
|
||||||
text = NO_ANSWER_MSG
|
|
||||||
text = text.strip()
|
|
||||||
return text
|
|
||||||
|
|
||||||
|
|
||||||
def _get_instructions(args):
|
|
||||||
links = _get_links(args['query'])
|
|
||||||
if not links:
|
|
||||||
return False
|
|
||||||
|
|
||||||
question_links = _get_questions(links)
|
|
||||||
if not question_links:
|
|
||||||
return False
|
|
||||||
|
|
||||||
only_hyperlinks = args.get('link')
|
|
||||||
star_headers = (args['num_answers'] > 1 or args['all'])
|
|
||||||
|
|
||||||
answers = []
|
|
||||||
initial_position = args['pos']
|
|
||||||
spliter_length = 80
|
|
||||||
answer_spliter = '\n' + '=' * spliter_length + '\n\n'
|
|
||||||
|
|
||||||
for answer_number in range(args['num_answers']):
|
|
||||||
current_position = answer_number + initial_position
|
|
||||||
args['pos'] = current_position
|
|
||||||
link = get_link_at_pos(question_links, current_position)
|
|
||||||
answer = _get_answer(args, question_links)
|
|
||||||
if not answer:
|
|
||||||
continue
|
|
||||||
if not only_hyperlinks:
|
|
||||||
answer = format_answer(link, answer, star_headers)
|
|
||||||
answer += '\n'
|
|
||||||
answers.append(answer)
|
|
||||||
return answer_spliter.join(answers)
|
|
||||||
|
|
||||||
|
|
||||||
def format_answer(link, answer, star_headers):
|
|
||||||
if star_headers:
|
|
||||||
return ANSWER_HEADER.format(link, answer, STAR_HEADER)
|
|
||||||
return answer
|
|
||||||
|
|
||||||
|
|
||||||
def _enable_cache():
|
|
||||||
if not os.path.exists(CACHE_DIR):
|
|
||||||
os.makedirs(CACHE_DIR)
|
|
||||||
requests_cache.install_cache(CACHE_FILE)
|
|
||||||
|
|
||||||
|
|
||||||
def _clear_cache():
|
|
||||||
for cache in glob.iglob('{0}*'.format(CACHE_FILE)):
|
|
||||||
os.remove(cache)
|
|
||||||
|
|
||||||
|
|
||||||
def howdoi(args):
|
|
||||||
args['query'] = ' '.join(args['query']).replace('?', '')
|
|
||||||
try:
|
|
||||||
return _get_instructions(args) or 'Sorry, couldn\'t find any help with that topic\n'
|
|
||||||
except (ConnectionError, SSLError):
|
|
||||||
return 'Failed to establish network connection\n'
|
|
||||||
|
|
||||||
|
|
||||||
def get_parser():
|
|
||||||
parser = argparse.ArgumentParser(description='instant coding answers via the command line')
|
|
||||||
parser.add_argument('query', metavar='QUERY', type=str, nargs='*',
|
|
||||||
help='the question to answer')
|
|
||||||
parser.add_argument('-p', '--pos', help='select answer in specified position (default: 1)', default=1, type=int)
|
|
||||||
parser.add_argument('-a', '--all', help='display the full text of the answer',
|
|
||||||
action='store_true')
|
|
||||||
parser.add_argument('-l', '--link', help='display only the answer link',
|
|
||||||
action='store_true')
|
|
||||||
parser.add_argument('-c', '--color', help='enable colorized output',
|
|
||||||
action='store_true')
|
|
||||||
parser.add_argument('-n', '--num-answers', help='number of answers to return', default=1, type=int)
|
|
||||||
parser.add_argument('-C', '--clear-cache', help='clear the cache',
|
|
||||||
action='store_true')
|
|
||||||
parser.add_argument('-v', '--version', help='displays the current version of howdoi',
|
|
||||||
action='store_true')
|
|
||||||
return parser
|
|
||||||
|
|
||||||
|
|
||||||
def command_line_runner():
|
|
||||||
parser = get_parser()
|
|
||||||
args = vars(parser.parse_args())
|
|
||||||
|
|
||||||
if args['version']:
|
|
||||||
print(__version__)
|
|
||||||
return
|
|
||||||
|
|
||||||
if args['clear_cache']:
|
|
||||||
_clear_cache()
|
|
||||||
print('Cache cleared successfully')
|
|
||||||
return
|
|
||||||
|
|
||||||
if not args['query']:
|
|
||||||
parser.print_help()
|
|
||||||
return
|
|
||||||
|
|
||||||
# enable the cache if user doesn't want it to be disabled
|
|
||||||
if not os.getenv('HOWDOI_DISABLE_CACHE'):
|
|
||||||
_enable_cache()
|
|
||||||
|
|
||||||
if os.getenv('HOWDOI_COLORIZE'):
|
|
||||||
args['color'] = True
|
|
||||||
|
|
||||||
utf8_result = howdoi(args).encode('utf-8', 'ignore')
|
|
||||||
if sys.version < '3':
|
|
||||||
print(utf8_result)
|
|
||||||
else:
|
|
||||||
# Write UTF-8 to stdout: https://stackoverflow.com/a/3603160
|
|
||||||
sys.stdout.buffer.write(utf8_result)
|
|
||||||
# close the session to release connection
|
|
||||||
howdoi_session.close()
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
|
||||||
command_line_runner()
|
|
||||||
@ -1,10 +0,0 @@
|
|||||||
{
|
|
||||||
"author" : ["Bobloy"],
|
|
||||||
"bot_version" : [3,0,0],
|
|
||||||
"description" : "Answer coding questions using stackexchange",
|
|
||||||
"hidden" : false,
|
|
||||||
"install_msg" : "Thank you for installing Howdoi.",
|
|
||||||
"requirements" : ["Pygments==2.1.1", "argparse==1.4.0", "cssselect==0.9.1", "pyquery==1.4.0", "requests==2.9.1", "requests-cache==0.4.11"],
|
|
||||||
"short" : "Answer coding questions",
|
|
||||||
"tags" : ["coding", "tools", "utils", "bobloy"]
|
|
||||||
}
|
|
||||||
@ -1,5 +0,0 @@
|
|||||||
from .nudity import Nudity
|
|
||||||
|
|
||||||
|
|
||||||
def setup(bot):
|
|
||||||
bot.add_cog(Nudity(bot))
|
|
||||||
@ -1,20 +0,0 @@
|
|||||||
{
|
|
||||||
"author": [
|
|
||||||
"Bobloy"
|
|
||||||
],
|
|
||||||
"bot_version": [
|
|
||||||
3,
|
|
||||||
0,
|
|
||||||
0
|
|
||||||
],
|
|
||||||
"description": "Keep track of when users were last seen online",
|
|
||||||
"hidden": true,
|
|
||||||
"install_msg": "Thank you for installing LastSeen. Get started with `[p]load nudity`, then `[p]help Nudity`",
|
|
||||||
"requirements": ["nudepy"],
|
|
||||||
"short": "Last seen tracker",
|
|
||||||
"tags": [
|
|
||||||
"bobloy",
|
|
||||||
"utils",
|
|
||||||
"tools"
|
|
||||||
]
|
|
||||||
}
|
|
||||||
@ -1,45 +0,0 @@
|
|||||||
from datetime import datetime
|
|
||||||
|
|
||||||
import discord
|
|
||||||
import nude
|
|
||||||
from nude import Nude
|
|
||||||
from redbot.core import Config
|
|
||||||
from redbot.core import commands
|
|
||||||
from redbot.core.bot import Red
|
|
||||||
|
|
||||||
|
|
||||||
class Nudity:
|
|
||||||
"""
|
|
||||||
V3 Cog Template
|
|
||||||
"""
|
|
||||||
|
|
||||||
online_status = discord.Status.online
|
|
||||||
|
|
||||||
offline_status = discord.Status.offline
|
|
||||||
|
|
||||||
def __init__(self, bot: Red):
|
|
||||||
self.bot = bot
|
|
||||||
self.config = Config.get_conf(self, identifier=9811198108111121, force_registration=True)
|
|
||||||
|
|
||||||
default_guild = {
|
|
||||||
"enabled": False
|
|
||||||
}
|
|
||||||
|
|
||||||
self.config.register_guild(**default_guild)
|
|
||||||
|
|
||||||
@commands.command(aliases=['togglenudity'], name='nudity')
|
|
||||||
async def nudity(self, ctx: commands.Context):
|
|
||||||
"""Toggle nude-checking on or off"""
|
|
||||||
is_on = await self.config.guild(ctx.guild).enabled()
|
|
||||||
await self.config.guild(ctx.guild).enabled.set(not is_on)
|
|
||||||
await ctx.send("Nude checking is now set to {}".format(not is_on))
|
|
||||||
|
|
||||||
async def on_message(self, message: discord.Message):
|
|
||||||
is_on = await self.config.guild(message.guild).enabled()
|
|
||||||
if not is_on:
|
|
||||||
return
|
|
||||||
|
|
||||||
if not message.attachments:
|
|
||||||
return
|
|
||||||
|
|
||||||
|
|
||||||
@ -1,5 +0,0 @@
|
|||||||
from .secrethitler import Werewolf
|
|
||||||
|
|
||||||
|
|
||||||
def setup(bot):
|
|
||||||
bot.add_cog(Werewolf(bot))
|
|
||||||
@ -1,114 +0,0 @@
|
|||||||
import asyncio
|
|
||||||
|
|
||||||
import discord
|
|
||||||
|
|
||||||
|
|
||||||
from redbot.core import Config, commands
|
|
||||||
|
|
||||||
from datetime import datetime, timedelta
|
|
||||||
|
|
||||||
from werewolf.game import Game
|
|
||||||
|
|
||||||
|
|
||||||
class Werewolf:
|
|
||||||
"""
|
|
||||||
Base to host werewolf on a guild
|
|
||||||
"""
|
|
||||||
|
|
||||||
def __init__(self, bot):
|
|
||||||
self.bot = bot
|
|
||||||
self.config = Config.get_conf(self, identifier=87101114101119111108102, force_registration=True)
|
|
||||||
default_global = {}
|
|
||||||
default_guild = {
|
|
||||||
}
|
|
||||||
|
|
||||||
self.config.register_global(**default_global)
|
|
||||||
self.config.register_guild(**default_guild)
|
|
||||||
|
|
||||||
self.games = {} # Active games stored here, id is per guild
|
|
||||||
|
|
||||||
@commands.group()
|
|
||||||
async def ww(self, ctx: commands.Context):
|
|
||||||
"""
|
|
||||||
Base command for this cog. Check help for the commands list.
|
|
||||||
"""
|
|
||||||
if ctx.invoked_subcommand is None:
|
|
||||||
pass
|
|
||||||
|
|
||||||
@ww.command()
|
|
||||||
async def new(self, ctx, game_code):
|
|
||||||
"""
|
|
||||||
Create and join a new game of Werewolf
|
|
||||||
"""
|
|
||||||
|
|
||||||
game = self._get_game(ctx.guild, game_code)
|
|
||||||
|
|
||||||
if not game:
|
|
||||||
await ctx.send("Failed to start a new game")
|
|
||||||
else:
|
|
||||||
await ctx.send("New game has started")
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
@ww.command()
|
|
||||||
async def join(self, ctx):
|
|
||||||
"""
|
|
||||||
Joins a game of Werewolf
|
|
||||||
"""
|
|
||||||
|
|
||||||
game = self._get_game(ctx.guild)
|
|
||||||
|
|
||||||
if not game:
|
|
||||||
await ctx.send("No game to join!\nCreate a new one with `[p]ww new`")
|
|
||||||
return
|
|
||||||
|
|
||||||
await game.join(ctx.author, ctx.channel)
|
|
||||||
|
|
||||||
@ww.command()
|
|
||||||
async def quit(self, ctx):
|
|
||||||
"""
|
|
||||||
Quit a game of Werewolf
|
|
||||||
"""
|
|
||||||
|
|
||||||
game = self._get_game(ctx.guild)
|
|
||||||
|
|
||||||
await game.quit(ctx.author, ctx.channel)
|
|
||||||
|
|
||||||
@ww.command()
|
|
||||||
async def start(self, ctx):
|
|
||||||
"""
|
|
||||||
Checks number of players and attempts to start the game
|
|
||||||
"""
|
|
||||||
game = self._get_game(ctx.guild)
|
|
||||||
if not game:
|
|
||||||
await ctx.send("No game running, cannot start")
|
|
||||||
|
|
||||||
await game.setup(ctx)
|
|
||||||
|
|
||||||
@ww.command()
|
|
||||||
async def vote(self, ctx, id):
|
|
||||||
"""
|
|
||||||
Vote for a player by ID
|
|
||||||
"""
|
|
||||||
game = self._get_game(ctx.guild)
|
|
||||||
if not game:
|
|
||||||
await ctx.send("No game running, cannot vote")
|
|
||||||
|
|
||||||
# Game handles response now
|
|
||||||
channel = ctx.channel
|
|
||||||
if channel is game.village_channel:
|
|
||||||
await game.vote(ctx.author, id, channel)
|
|
||||||
|
|
||||||
if channel in (c["channel"] for c in game.p_channels.values()):
|
|
||||||
await game.vote(ctx.author, id, channel)
|
|
||||||
|
|
||||||
def _get_game(self, guild, game_code=None):
|
|
||||||
if guild.id not in self.games:
|
|
||||||
if not game_code:
|
|
||||||
return None
|
|
||||||
self.games[guild.id] = Game(guild, game_code)
|
|
||||||
|
|
||||||
return self.games[guild.id]
|
|
||||||
|
|
||||||
async def _game_start(self, game):
|
|
||||||
await game.start()
|
|
||||||
Loading…
Reference in new issue