Refactored bot.py to wrap Geoffrey in his own class. This will be better for the future for expansions and allowed for cleaning up a lot of the config mess. The Config now all features back except for the named tunnel directions which are removed for now...

doc_update
Joey Hines 2018-08-30 11:34:19 -05:00
parent 29942ac2dd
commit 6748f01208
12 changed files with 139 additions and 153 deletions

View File

@ -18,10 +18,6 @@ def create_config(config, path):
}
config['Minecraft'] = {'Dynmap_Url': '',
'World_Name': '',
'North_Tunnel': 'North',
"East_Tunnel": 'South',
"South_Tunnel": 'East',
"West_Tunnel": 'West'
}
config['Logging'] = {'Count': '7',
'Rotation_Duration': '1'
@ -59,10 +55,6 @@ class Config:
self.error_users = self.config['Discord']['Error_Users'].split(',')
self.dynmap_url = self.config['Minecraft']['Dynmap_Url']
self.north_tunnel = self.config['Minecraft']['North_Tunnel']
self.east_tunnel = self.config['Minecraft']['East_Tunnel']
self.south_tunnel = self.config['Minecraft']['South_Tunnel']
self.west_tunnel = self.config['Minecraft']['West_Tunnel']
self.count = int(self.config['Logging']['Count'])
self.rotation_duration = int(self.config['Logging']['Rotation_Duration'])

View File

@ -88,8 +88,7 @@ class Commands:
location_name = location_list[0].name
tunnel = self.interface.add_tunnel(session, player, tunnel_color, tunnel_number, location_name,
self.bot_config)
tunnel = self.interface.add_tunnel(session, player, tunnel_color, tunnel_number, location_name)
tunnel_info = tunnel.__str__()
finally:
@ -233,7 +232,7 @@ class Commands:
location.tunnel.tunnel_direction = TunnelDirection.str_to_tunnel_dir(tunnel_color)
location.tunnel.tunnel_number = tunnel_number
else:
self.interface.add_tunnel(session, player, tunnel_color, tunnel_number, loc_name, self.bot_config)
self.interface.add_tunnel(session, player, tunnel_color, tunnel_number, loc_name)
loc_str = location.__str__()

View File

@ -16,7 +16,7 @@ class DatabaseInterface:
self.database.add_object(session, shop)
return shop
def add_tunnel(self, session, owner, color, number, location_name, config):
def add_tunnel(self, session, owner, color, number, location_name):
tunnels = self.find_tunnel_by_owner(session, owner)
if location_name is None:
if len(tunnels):
@ -33,7 +33,7 @@ class DatabaseInterface:
except IndexError:
raise LocationLookUpError
tunnel = Tunnel(owner, color, number, config, location)
tunnel = Tunnel(owner, color, number, location)
self.database.add_object(session, tunnel)
return tunnel

View File

@ -16,7 +16,7 @@ SQL_Base = declarative_base()
def check_similarity(a, b):
ratio = SequenceMatcher(None, a, b).ratio()
if (ratio > 0.6) or (a[0] == b[0]):
if (ratio > 0.80) or (a[0] == b[0]):
return True
else:
return False
@ -85,21 +85,22 @@ class GeoffreyDatabase:
class TunnelDirection(enum.Enum):
North = "North"
East = "East"
South = "South"
West = "West"
North = "north"
East = "east"
South = "south"
West = "west"
def str_to_tunnel_dir(bot_config, arg):
def str_to_tunnel_dir(arg):
arg = arg.lower()
if check_similarity(bot_config.north_tunnel, arg):
if check_similarity(TunnelDirection.North.value, arg):
return TunnelDirection.North
elif check_similarity(bot_config.east_tunnel, arg):
elif check_similarity(TunnelDirection.East.value, arg):
return TunnelDirection.East
elif check_similarity(bot_config.south_tunnel, arg):
elif check_similarity(TunnelDirection.South.value, arg):
return TunnelDirection.South
elif check_similarity(bot_config.west_tunnel, arg):
elif check_similarity(TunnelDirection.West.value, arg):
return TunnelDirection.West
else:
raise InvalidTunnelError
@ -151,11 +152,11 @@ class Tunnel(SQL_Base):
location_id = Column(Integer, ForeignKey('geoffrey_locations.id', ondelete='CASCADE'))
location = relationship("Location", back_populates="tunnel", lazy="joined")
def __init__(self, owner, tunnel_color, tunnel_number, config, location=None):
def __init__(self, owner, tunnel_color, tunnel_number, location=None):
try:
self.owner = owner
self.location = location
self.tunnel_direction = TunnelDirection.str_to_tunnel_dir(config, tunnel_color)
self.tunnel_direction = TunnelDirection.str_to_tunnel_dir(tunnel_color)
self.tunnel_number = tunnel_number
except (ValueError, IndexError):
raise TunnelInitError

View File

@ -37,93 +37,95 @@ extensions = ['geoffrey.cogs.Add_Commands',
'geoffrey.cogs.Search_Commands',
'geoffrey.cogs.Admin_Commands']
bot_config = None
bot_commands = None
bot = commands.Bot(command_prefix='?', description=description, case_insensitive=True)
class GeoffreyBot(commands.Bot):
def __init__(self, config):
super().__init__(command_prefix=config.prefix, description=description, pm_help=True, case_insensitive=True)
self.error_users = config.error_users
self.admin_users = config.bot_mod
self.bot_commands = Commands(config)
for extension in extensions:
try:
self.load_extension(extension)
except Exception as e:
logger.info('Failed to load extension {}'.format(extension))
raise e
@bot.event
async def on_ready():
logger.info("%s Online, ID: %s", bot.user.name, bot.user.id)
info = await bot.application_info()
url = oauth_url(info.id)
logger.info("Bot url: %s", url)
await bot.change_presence(game=Game(name="Geoffrey"))
async def on_ready(self):
logger.info("%s Online, ID: %s", self.user.name, self.user.id)
info = await self.application_info()
url = oauth_url(info.id)
logger.info("Bot url: %s", url)
await self.change_presence(game=Game(name="Geoffrey"))
async def on_command(self, command, ctx):
if ctx.invoked_subcommand is None:
subcommand = ""
else:
subcommand = ":" + ctx.invoked_subcommand
@bot.event
async def on_command(command, ctx):
if ctx.invoked_subcommand is None:
subcommand = ""
else:
subcommand = ":" + ctx.invoked_subcommand
logger.info("User %s, used command %s%s with context: %s", ctx.message.author, command, subcommand,
ctx.args)
logger.info("User %s, used command %s%s with context: %s", ctx.message.author, command, subcommand,
ctx.args)
@bot.event
async def on_command_error(error, ctx):
error_str = ''
if hasattr(ctx, 'cog'):
if "Admin_Commands" in ctx.cog.__str__():
async def on_command_error(self, error, ctx):
error_str = ''
if hasattr(ctx, 'cog'):
if "Admin_Commands" in ctx.cog.__str__():
return
if hasattr(error, 'original'):
if isinstance(error.original, NoPermissionError):
error_str = 'You don\'t have permission for that cool command.'
elif isinstance(error.original, UsernameLookupFailed):
error_str = 'Your user name was not found, either Mojang is having a fucky wucky ' \
'or your nickname is not set correctly. *stares at the Mods*'
elif isinstance(error.original, PlayerNotFound):
error_str = 'Make sure to use ?register first you ding dong.'
elif isinstance(error.original, EntryNameNotUniqueError):
error_str = 'An entry in the database already has that name you ding dong.'
elif isinstance(error.original, DatabaseValueError):
error_str = 'Use a shorter name or a smaller value, dong ding.'
elif isinstance(error.original, NotOnServerError):
error_str = 'Command needs to be run on 24CC. Run this command there whoever you are.'.format()
elif isinstance(error.original, OperationalError):
await self.send_error_message('Error connecting to the MySQL server, is it offline?')
error_str = 'Database connection issue, looks like some admin has to fix something.'.format()
elif isinstance(error, commands.CommandOnCooldown):
return
if hasattr(error, 'original'):
if isinstance(error.original, NoPermissionError):
error_str = 'You don\'t have permission for that cool command.'
elif isinstance(error.original, UsernameLookupFailed):
error_str = 'Your user name was not found, either Mojang is having a fucky wucky ' \
'or your nickname is not set correctly. *stares at the Mods*'
elif isinstance(error.original, PlayerNotFound):
error_str = 'Make sure to use ?register first you ding dong.'
elif isinstance(error.original, EntryNameNotUniqueError):
error_str = 'An entry in the database already has that name you ding dong.'
elif isinstance(error.original, DatabaseValueError):
error_str = 'Use a shorter name or a smaller value, dong ding.'
elif isinstance(error.original, NotOnServerError):
error_str = 'Command needs to be run on 24CC. Run this command there whoever you are.'.format()
elif isinstance(error.original, OperationalError):
await send_error_message('Error connecting to the MySQL server, is it offline?')
error_str = 'Database connection issue, looks like some admin has to fix something.'.format()
elif isinstance(error, commands.CommandOnCooldown):
return
elif isinstance(error, commands.UserInputError):
error_str = 'Invalid syntax for **{}** you ding dong:' \
.format(ctx.invoked_with, ctx.invoked_with)
elif isinstance(error, commands.UserInputError):
error_str = 'Invalid syntax for **{}** you ding dong:' \
.format(ctx.invoked_with, ctx.invoked_with)
pages = bot.formatter.format_help_for(ctx, ctx.command)
for page in pages:
error_str = error_str + '\n' + page
elif isinstance(error, commands.CommandNotFound):
return
pages = self.formatter.format_help_for(ctx, ctx.command)
for page in pages:
error_str = error_str + '\n' + page
elif isinstance(error, commands.CommandNotFound):
return
if error_str is None:
await send_error_message(
'Geoffrey encountered unhandled exception: {}. Context:'.format(error, ctx.args))
if error_str is '':
await self.send_error_message(
'Geoffrey encountered unhandled exception: {}. Context:'.format(error, ctx.args))
logger.error("Geoffrey encountered unhandled exception: %s", error)
error_str = bad_error_message.format(ctx.invoked_with)
logger.error("Geoffrey encountered unhandled exception: %s", error)
error_str = bad_error_message.format(ctx.invoked_with)
await bot.send_message(ctx.message.channel,
'{} **Error Running Command:** {}'.format(ctx.message.author.mention,
error_str))
await self.send_message(ctx.message.channel,
'{} **Error Running Command:** {}'.format(ctx.message.author.mention,
error_str))
async def send_error_message(self, msg):
for user_id in self.error_users:
user = await self.get_user_info(user_id)
await self.send_message(user, msg)
async def send_error_message(msg):
for user_id in bot_config.error_users:
user = await bot.get_user_info(user_id)
await bot.send_message(user, msg)
async def username_update():
async def username_update(bot):
await bot.wait_until_ready()
while not bot.is_closed:
session = bot_commands.interface.database.Session()
session = bot.bot_commands.interface.database.Session()
try:
logger.info("Updating MC usernames...")
session = bot_commands.interface.database.Session()
session = bot.bot_commands.interface.database.Session()
player_list = session.query(Player).all()
for player in player_list:
player.name = grab_playername(player.mc_uuid)
@ -135,7 +137,7 @@ async def username_update():
logger.info("Username lookup error.")
session.rollback()
except OperationalError:
await send_error_message('Error connecting to the MySQL server, is it offline?')
await bot.send_error_message('Error connecting to the MySQL server, is it offline?')
logger.info("MySQL connection error")
finally:
session.close()
@ -169,20 +171,13 @@ def setup_logging(config):
def start_bot(config_path="{}/GeoffreyConfig.ini".format(path.dirname(path.abspath(__file__)))):
try:
global bot_config, bot_commands
bot_config = get_config(config_path)
bot = GeoffreyBot(bot_config)
setup_logging(bot_config)
bot_commands = Commands(bot_config)
for extension in extensions:
try:
bot.load_extension(extension)
except Exception as e:
logger.info('Failed to load extension {}'.format(extension))
raise e
logger.info('Logging into Discord...')
bot.loop.create_task(username_update(bot))
bot.run(bot_config.token)
except KeyboardInterrupt:

View File

@ -2,7 +2,6 @@ from discord.ext import commands
from geoffrey.BotErrors import *
from geoffrey.DiscordHelperFunctions import *
from geoffrey.bot import bot_commands, bot_config
@commands.cooldown(5, 60, commands.BucketType.user)
@ -25,7 +24,7 @@ class Add_Commands:
try:
player_name = get_nickname(ctx.message.author, bot_config)
bot_commands.register(player_name, ctx.message.author.id)
self.bot.bot_commands.register(player_name, ctx.message.author.id)
await self.bot.say('{}, you have been added to the database.'.format(ctx.message.author.mention))
except AttributeError:
raise NotOnServerError
@ -43,7 +42,7 @@ class Add_Commands:
name = get_name(args)
try:
base = bot_commands.add_base(x_pos, z_pos, base_name=name, discord_uuid=ctx.message.author.id)
base = self.bot.bot_commands.add_base(x_pos, z_pos, base_name=name, discord_uuid=ctx.message.author.id)
await self.bot.say(
'{}, your base has been added to the database: \n\n{}'.format(ctx.message.author.mention, base))
except LocationInitError:
@ -68,7 +67,7 @@ class Add_Commands:
name = get_name(args)
try:
shop = bot_commands.add_shop(x_pos, z_pos, shop_name=name, discord_uuid=ctx.message.author.id)
shop = self.bot.bot_commands.add_shop(x_pos, z_pos, shop_name=name, discord_uuid=ctx.message.author.id)
await self.bot.say(
'{}, your shop has been added to the database: \n\n{}'.format(ctx.message.author.mention, shop))
except LocationInitError:
@ -93,7 +92,7 @@ class Add_Commands:
loc_name = get_name(args)
try:
bot_commands.add_tunnel(tunnel_color, tunnel_number, discord_uuid=ctx.message.author.id,
self.bot.bot_commands.add_tunnel(tunnel_color, tunnel_number, discord_uuid=ctx.message.author.id,
location_name=loc_name)
await self.bot.say('{}, your tunnel has been added to the database'.format(ctx.message.author.mention))
except LocationLookUpError:
@ -120,7 +119,7 @@ class Add_Commands:
"""
shop_name = get_name(args)
try:
bot_commands.add_item(item_name, quantity, diamond_price, shop_name=shop_name,
self.bot.bot_commands.add_item(item_name, quantity, diamond_price, shop_name=shop_name,
discord_uuid=ctx.message.author.id)
await self.bot.say(
'{}, **{}** has been added to the inventory of your shop.'.format(ctx.message.author.mention,

View File

@ -2,13 +2,12 @@ from discord import Game
from discord.ext import commands
from geoffrey.BotErrors import *
from geoffrey.bot import bot_commands, bot_config
def check_mod(user):
def check_mod(user, admin_users):
try:
for role in user.roles:
if role.id in bot_config.bot_mod:
if role.id in admin_users:
return True
except AttributeError:
raise NotOnServerError
@ -39,7 +38,7 @@ class Admin_Commands:
"""
Checks if the bot is alive.
"""
if check_mod(ctx.message.author):
if check_mod(ctx.message.author, self.bot.admin_users):
await self.bot.say('I\'m here you ding dong')
else:
raise NoPermissionError
@ -49,7 +48,7 @@ class Admin_Commands:
"""
Bot moderation tools.
"""
if check_mod(ctx.message.author):
if check_mod(ctx.message.author, self.bot.admin_users):
if ctx.invoked_subcommand is None:
await self.bot.say('{}, invalid sub-command for command **mod**.'.format(ctx.message.author.mention))
else:
@ -60,7 +59,7 @@ class Admin_Commands:
"""
Deletes a location in the database.
"""
bot_commands.delete(location_name, discord_uuid=discord_uuid)
self.bot.bot_commands.delete(location_name, discord_uuid=discord_uuid)
await self.bot.say('{}, **{}** has been deleted.'.format(ctx.message.author.mention, location_name))
@delete.error
@ -72,7 +71,7 @@ class Admin_Commands:
"""
Edits the name of a location in the database.
"""
bot_commands.edit_name(new_name, current_name, discord_uuid=discord_uuid)
self.bot.bot_commands.edit_name(new_name, current_name, discord_uuid=discord_uuid)
await self.bot.say('{}, **{}** has been rename to **{}**.'.format(ctx.message.author.mention, current_name,
new_name))
@ -85,7 +84,7 @@ class Admin_Commands:
"""
Updates a user's MC UUID.
"""
bot_commands.update_mc_uuid(discord_uuid, mc_uuid)
self.bot.bot_commands.update_mc_uuid(discord_uuid, mc_uuid)
await self.bot.say('{}, **{}** has been updated.'.format(ctx.message.author.mention, discord_uuid))
@update_mc_uuid.error
@ -97,7 +96,7 @@ class Admin_Commands:
"""
Updates a user's Discord UUID.
"""
bot_commands.update_mc_uuid(current_discord_uuid, new_discord_uuid)
self.bot.bot_commands.update_mc_uuid(current_discord_uuid, new_discord_uuid)
await self.bot.say('{}, user **{}** has been updated.'.format(ctx.message.author.mention, current_discord_uuid))
@update_discord_uuid.error
@ -109,7 +108,7 @@ class Admin_Commands:
"""
Updates a user's MC name to the current name on the MC UUID.
"""
bot_commands.update_mc_name(discord_uuid)
self.bot.bot_commands.update_mc_name(discord_uuid)
await self.bot.say('{}, user **{}**\'s MC name has update.'.format(ctx.message.author.mention, discord_uuid))
@update_mc_name.error
@ -121,7 +120,7 @@ class Admin_Commands:
"""
Updates "playing [game]" status of the bot.
"""
await self.bot.change_presence(activity=Game(status))
await self.bot.change_presence(game=Game(name=status))
await self.bot.say('{}, status has been changed'.format(ctx.message.author.mention))

View File

@ -2,7 +2,6 @@ from discord.ext import commands
from geoffrey.BotErrors import *
from geoffrey.DiscordHelperFunctions import *
from geoffrey.bot import bot_commands, bot_config
class Delete_Commands:
@ -26,7 +25,7 @@ class Delete_Commands:
if loc is None:
raise commands.UserInputError
bot_commands.delete(loc, discord_uuid=ctx.message.author.id)
self.bot.bot_commands.delete(loc, discord_uuid=ctx.message.author.id)
await self.bot.say(
'{}, your location named **{}** has been deleted.'.format(ctx.message.author.mention, loc))
except (DeleteEntryError, PlayerNotFound):
@ -42,7 +41,7 @@ class Delete_Commands:
shop = get_name(args)
try:
bot_commands.delete_item(item, shop, discord_uuid=ctx.message.author.id)
self.bot.bot_commands.delete_item(item, shop, discord_uuid=ctx.message.author.id)
await self.bot.say('{}, **{}** has been removed from the inventory of **{}**.'.
format(ctx.message.author.mention, item, shop))

View File

@ -2,7 +2,6 @@ from discord.ext import commands
from geoffrey.BotErrors import *
from geoffrey.DiscordHelperFunctions import *
from geoffrey.bot import bot_commands, bot_config
class Edit_Commands:
@ -21,7 +20,7 @@ class Edit_Commands:
"""
loc = get_name(args)
try:
loc_str = bot_commands.edit_pos(x_pos, y_pos, loc, discord_uuid=ctx.message.author.id)
loc_str = self.bot.bot_commands.edit_pos(x_pos, y_pos, loc, discord_uuid=ctx.message.author.id)
await self.bot.say(
'{}, the following location has been updated: \n\n{}'.format(ctx.message.author.mention, loc_str))
@ -38,7 +37,7 @@ class Edit_Commands:
"""
loc = get_name(args)
try:
loc_str = bot_commands.edit_tunnel(tunnel_color, tunnel_number, loc, discord_uuid=ctx.message.author.id)
loc_str = self.bot.bot_commands.edit_tunnel(tunnel_color, tunnel_number, loc, discord_uuid=ctx.message.author.id)
await self.bot.say(
'{}, the following location has been updated: \n\n{}'.format(ctx.message.author.mention, loc_str))
@ -58,7 +57,7 @@ class Edit_Commands:
?edit_name [New Name] [Current Name]
"""
try:
loc_str = bot_commands.edit_name(new_name, current_name, discord_uuid=ctx.message.author.id)
loc_str = self.bot.bot_commands.edit_name(new_name, current_name, discord_uuid=ctx.message.author.id)
await self.bot.say(
'{}, the following location has been updated: \n\n{}'.format(ctx.message.author.mention, loc_str))

View File

@ -2,7 +2,6 @@ from discord.ext import commands
from geoffrey.BotErrors import *
from geoffrey.DiscordHelperFunctions import *
from geoffrey.bot import bot_commands, bot_config
class Search_Commands:
@ -26,7 +25,7 @@ class Search_Commands:
if search is None:
raise commands.UserInputError
result = bot_commands.find(search)
result = self.bot.bot_commands.find(search)
await self.bot.say(
'{}, The following entries match **{}**:\n{}'.format(ctx.message.author.mention, search, result))
@ -42,7 +41,7 @@ class Search_Commands:
?tunnel [Player]
"""
try:
result = bot_commands.tunnel(player)
result = self.bot.bot_commands.tunnel(player)
await self.bot.say(
'{}, **{}** owns the following tunnels: \n{}'.format(ctx.message.author.mention, player, result))
@ -78,7 +77,7 @@ class Search_Commands:
if args[1] == '-d':
dimension = args[2]
base_string = bot_commands.find_around(x_pos, z_pos, radius, dimension)
base_string = self.bot.bot_commands.find_around(x_pos, z_pos, radius, dimension)
if len(base_string) != 0:
await self.bot.say('{}, the following locations(s) within **{}** blocks of that point: \n {}'.format(
@ -101,7 +100,7 @@ class Search_Commands:
?selling [item]
"""
try:
result = bot_commands.selling(item_name)
result = self.bot.bot_commands.selling(item_name)
await self.bot.say(
'{}, the following shops sell **{}**: \n{}'.format(ctx.message.author.mention, item_name, result))
except ItemNotFound:
@ -121,7 +120,7 @@ class Search_Commands:
if loc is None:
raise commands.UserInputError
info_str = bot_commands.info(loc)
info_str = self.bot.bot_commands.info(loc)
await self.bot.say(info_str)
except IndexError:
await self.bot.say('{}, no locations in the database match **{}**.'.format(ctx.message.author.mention, loc))
@ -134,7 +133,7 @@ class Search_Commands:
"""
try:
loc_str = bot_commands.me(discord_uuid=ctx.message.author.id)
loc_str = self.bot.bot_commands.me(discord_uuid=ctx.message.author.id)
await self.bot.say('{}, here are your locations in the database: \n {}'.format(ctx.message.author.mention,
loc_str))
except PlayerNotFound:

View File

@ -1,3 +1,4 @@
import os
from unittest import TestCase
from Commands import *
@ -6,7 +7,8 @@ from BotConfig import get_config
class TestCommands(TestCase):
def setUp(self):
self.bot_config = get_config()
path = os.path.dirname(os.path.abspath(__file__))
self.bot_config = get_config('{}/GeoffreyConfig.ini'.format(path))
self.commands = Commands(self.bot_config, True)
self.session = self.commands.interface.database.Session()
self.commands.interface.database.clear_all(self.session)
@ -50,13 +52,13 @@ class TestCommands(TestCase):
self.commands.register('ZeroHD', '143072699567177728')
self.commands.add_shop(0, 0, shop_name='test shop', discord_uuid='143072699567177728')
tunnel2 = self.commands.add_tunnel(self.bot_config.east_tunnel, 50, location_name='test_shop',
tunnel2 = self.commands.add_tunnel("East", 50, location_name='test_shop',
discord_uuid='143072699567177728')
if self.bot_config.east_tunnel not in tunnel2:
if "East" not in tunnel2:
self.fail()
self.assertRaises(LocationHasTunnelError, self.commands.add_tunnel, self.bot_config.east_tunnel, 50,
self.assertRaises(LocationHasTunnelError, self.commands.add_tunnel, "East", 50,
location_name='test_shop', discord_uuid='143072699567177728')
def test_find(self):
@ -127,11 +129,11 @@ class TestCommands(TestCase):
self.commands.register('ZeroHD', '143072699567177728')
self.commands.add_shop(0, 0, shop_name='frick', discord_uuid='143072699567177728')
self.commands.add_tunnel(self.bot_config.north_tunnel, 50, location_name='frick', discord_uuid='143072699567177728')
self.commands.add_tunnel("West", 50, location_name='frick', discord_uuid='143072699567177728')
result = self.commands.info('frick')
if self.bot_config.north_tunnel in result:
if "West" in result:
pass
else:
self.fail()
@ -140,11 +142,11 @@ class TestCommands(TestCase):
self.commands.register('ZeroHD', '143072699567177728')
self.commands.add_shop(0, 0, shop_name='test shop', discord_uuid='143072699567177728')
self.commands.add_tunnel(self.bot_config.south_tunnel, 50, None, discord_uuid='143072699567177728')
self.commands.add_tunnel("soUTH", 50, None, discord_uuid='143072699567177728')
result = self.commands.tunnel('ZeroHD')
if self.bot_config.south_tunnel in result:
if "South" in result:
pass
else:
self.fail()
@ -191,11 +193,11 @@ class TestCommands(TestCase):
self.commands.register('ZeroHD', '143072699567177728')
self.commands.add_shop(0, 0, shop_name='test shop', discord_uuid='143072699567177728')
self.commands.edit_tunnel(self.bot_config.east_tunnel, 500, 'test shop', discord_uuid='143072699567177728')
self.commands.edit_tunnel("West", 500, 'test shop', discord_uuid='143072699567177728')
result = self.commands.info('test shop')
if self.bot_config.east_tunnel in result:
if "West" in result:
pass
else:
self.fail()

View File

@ -1,3 +1,4 @@
import os
from unittest import TestCase
from DatabaseInterface import *
@ -5,14 +6,15 @@ from BotConfig import *
class TestGeoffreyDatabase(TestCase):
def setUp(self):
self.config = get_config()
self.interface = DatabaseInterface(self.config, debug=True)
path = os.path.dirname(os.path.abspath(__file__))
self.bot_config = get_config('{}/GeoffreyConfig.ini'.format(path))
self.interface = DatabaseInterface(self.bot_config, True)
self.session = self.interface.database.Session()
self.interface.database.clear_all(self.session)
self.owner = Player('ZeroHD', '143072699567177728')
self.loc = Location('test', 1, 3, self.owner, dimension='Nether')
self.tunnel = Tunnel(self.owner, self.config.west_tunnel, 105, self.config, self.loc)
self.tunnel = Tunnel(self.owner, "west", 105, self.loc)
def tearDown(self):
self.session.commit()
@ -91,7 +93,7 @@ class TestGeoffreyDatabase(TestCase):
def test_add_tunnel(self):
player = self.add_player()
tunnel1 = self.interface.add_tunnel(self.session, player, self.config.south_tunnel, 155, None, self.config)
tunnel1 = self.interface.add_tunnel(self.session, player, "South", 155, None)
tunnel2 = self.interface.find_tunnel_by_owner_name(self.session, 'ZeroHD')[0]
self.assertEqual(tunnel1, tunnel2)