Added global error handling
parent
60846c81d1
commit
ee3bfb0fee
270
MCInfoBot.py
270
MCInfoBot.py
|
|
@ -8,158 +8,172 @@ from sqlalchemy.orm import sessionmaker
|
|||
|
||||
TOKEN = ''
|
||||
command_prefix = '?'
|
||||
description = '''Geoffrey is an inside joke none of you will understand, at
|
||||
least he knows where your bases are.
|
||||
description = '''
|
||||
Geoffrey started his life as inside joke none of you will understand.
|
||||
At some point, she was to become an airhorn bot. Now, they know where your bases are.
|
||||
|
||||
Please respect Geoffrey, the bot is very sensitive.
|
||||
'''
|
||||
|
||||
engine = create_engine('sqlite:///:memory:', echo=True)
|
||||
SQL_Base = declarative_base()
|
||||
SQL_Base = declarative_base()
|
||||
|
||||
bot = commands.Bot(command_prefix=command_prefix , description=description)
|
||||
bot = commands.Bot(command_prefix=command_prefix, description=description, case_insensitive=True)
|
||||
|
||||
|
||||
class DataBaseError(Exception):
|
||||
'''Base class for exceptions in this module.'''
|
||||
pass
|
||||
|
||||
|
||||
class LocationInitError(DataBaseError):
|
||||
'''Error in initializing Location'''
|
||||
|
||||
|
||||
class LocationLookUpError(DataBaseError) :
|
||||
'''Error in finding location in database'''
|
||||
|
||||
|
||||
class TunnelDirection(enum.Enum):
|
||||
North = 'green'
|
||||
East = 'blue'
|
||||
South = 'red'
|
||||
West = 'yellow'
|
||||
|
||||
class TunnelDirection(enum.Enum) :
|
||||
North = 'green'
|
||||
East = 'blue'
|
||||
South = 'red'
|
||||
West = 'yellow'
|
||||
|
||||
class Player(SQL_Base):
|
||||
__tablename__ = 'Players'
|
||||
|
||||
id = Column(Integer, primary_key=True)
|
||||
in_game_name = Column(String)
|
||||
|
||||
def __init__(self,name) :
|
||||
self.in_game_name = name
|
||||
|
||||
__tablename__ = 'Players'
|
||||
|
||||
id = Column(Integer, primary_key=True)
|
||||
in_game_name = Column(String)
|
||||
|
||||
def __init__(self, name):
|
||||
self.in_game_name = name
|
||||
|
||||
|
||||
class Location(SQL_Base):
|
||||
__tablename__ = 'Locations'
|
||||
|
||||
id = Column(Integer, primary_key=True)
|
||||
name = Column(String)
|
||||
x = Column(Integer)
|
||||
y = Column(Integer)
|
||||
z = Column(Integer)
|
||||
tunnelNumber = Column(Integer)
|
||||
direction = Column(Enum(TunnelDirection))
|
||||
owner = Column(String, ForeignKey('Players.in_game_name'))
|
||||
|
||||
def __init__(self,args,owner) :
|
||||
self.name = args[0]
|
||||
self.x = int(args[1])
|
||||
self.y = int(args[2])
|
||||
self.z = int(args[3])
|
||||
self.owner = owner
|
||||
|
||||
if (len(args) >= 5) :
|
||||
self.tunnelNumber = int(args[5])
|
||||
self.direction = strToTunnelDirection(args[4])
|
||||
__tablename__ = 'Locations'
|
||||
|
||||
id = Column(Integer, primary_key=True)
|
||||
name = Column(String)
|
||||
x = Column(Integer)
|
||||
y = Column(Integer)
|
||||
z = Column(Integer)
|
||||
tunnelNumber = Column(Integer)
|
||||
direction = Column(Enum(TunnelDirection))
|
||||
owner = Column(String, ForeignKey('Players.in_game_name'))
|
||||
|
||||
def __init__(self, args, owner):
|
||||
try:
|
||||
self.name = args[0]
|
||||
self.x = int(args[1])
|
||||
self.y = int(args[2])
|
||||
self.z = int(args[3])
|
||||
self.owner = owner
|
||||
|
||||
if (len(args) >= 5):
|
||||
self.tunnelNumber = int(args[5])
|
||||
self.direction = strToTunnelDirection(args[4])
|
||||
except (ValueError, IndexError):
|
||||
raise LocationInitError
|
||||
|
||||
def posToStr(self):
|
||||
return '(x= {}, y= {}, z= {})'.format(self.x, self.y, self.z)
|
||||
|
||||
def netherTunnelAddrToStr(self):
|
||||
return '{} {}'.format(self.direction.value.title(), self.tunnelNumber)
|
||||
|
||||
def __str__(self):
|
||||
if (self.direction is not None):
|
||||
return "Name: {}, Position: {}, Tunnel: {}".format(self.name, self.posToStr(), self.netherTunnelAddrToStr())
|
||||
else:
|
||||
return "Name: {}, Position: {}".format(self.name, self.posToStr())
|
||||
|
||||
|
||||
def posToStr(self) :
|
||||
return '(x= {}, y= {}, z= {})'.format(self.x, self.y, self.z)
|
||||
def netherTunnelAddrToStr(self) :
|
||||
return '{} {}'.format(self.direction.value.title(), self.tunnelNumber)
|
||||
|
||||
def __str__(self):
|
||||
if (self.direction is not None) :
|
||||
return "Name: {}, Position: {}, Tunnel: {}".format(self.name,self.posToStr(),self.netherTunnelAddrToStr())
|
||||
else :
|
||||
return "Name: {}, Position: {}".format(self.name,self.posToStr())
|
||||
|
||||
SQL_Base.metadata.create_all(engine)
|
||||
|
||||
Session = sessionmaker(bind=engine)
|
||||
session = Session()
|
||||
|
||||
def strToTunnelDirection(str) :
|
||||
str = str.lower()
|
||||
if (str == TunnelDirection.North.value) :
|
||||
return TunnelDirection.North
|
||||
elif (str == TunnelDirection.East.value) :
|
||||
return TunnelDirection.East
|
||||
elif (str == TunnelDirection.South.value):
|
||||
return TunnelDirection.South
|
||||
elif (str == TunnelDirection.West.value):
|
||||
return TunnelDirection.West
|
||||
else:
|
||||
raise ValueError
|
||||
|
||||
#Bot Commands ******************************************************************
|
||||
|
||||
|
||||
def strToTunnelDirection(str):
|
||||
str = str.lower()
|
||||
if (str == TunnelDirection.North.value):
|
||||
return TunnelDirection.North
|
||||
elif (str == TunnelDirection.East.value):
|
||||
return TunnelDirection.East
|
||||
elif (str == TunnelDirection.South.value):
|
||||
return TunnelDirection.South
|
||||
elif (str == TunnelDirection.West.value):
|
||||
return TunnelDirection.West
|
||||
else:
|
||||
raise ValueError
|
||||
|
||||
|
||||
# Bot Commands ******************************************************************
|
||||
|
||||
|
||||
@bot.event
|
||||
async def on_ready():
|
||||
print('GeoffreyBot')
|
||||
print('Username: ' + bot.user.name)
|
||||
print('ID: ' + bot.user.id)
|
||||
print('GeoffreyBot')
|
||||
print('Username: ' + bot.user.name)
|
||||
print('ID: ' + bot.user.id)
|
||||
|
||||
|
||||
@bot.event
|
||||
async def on_command_error(error, ctx):
|
||||
error_str = 'Error in command {}, please use ?help {}'.format(ctx.invoked_with, ctx.invoked_with)
|
||||
await bot.send_message(ctx.message.channel, error_str)
|
||||
|
||||
|
||||
@bot.command()
|
||||
async def test():
|
||||
'''Check if the bot is alive.'''
|
||||
await bot.say('I\'m here you ding dong')
|
||||
'''Check if the bot is alive.'''
|
||||
await bot.say('I\'m here you ding dong')
|
||||
|
||||
|
||||
@bot.command(pass_context=True)
|
||||
async def addbase(ctx, * args):
|
||||
'''Add your base to the database.
|
||||
?addbase [Base Name] [X Coordinate] [Y Coordinate] [Z Coordinate] [Tunnel Color] [Tunnel Position]
|
||||
'''
|
||||
if (len(args) == 4) :
|
||||
try:
|
||||
owner = Player(str(ctx.message.author.nick))
|
||||
base = Location(args,owner.in_game_name)
|
||||
|
||||
session.add(owner)
|
||||
session.add(base)
|
||||
|
||||
await bot.say('{}, your base named {} located at {} has been added'
|
||||
' to the database.'.format(ctx.message.author.mention, base.name, base.posToStr()))
|
||||
except ValueError:
|
||||
await bot.say('Invalid syntax, try again (?addbase [name] [x coord] [z coord])')
|
||||
elif(len(args) == 6) :
|
||||
try:
|
||||
owner = Player(str(ctx.message.author.nick))
|
||||
base = Location(args,owner.in_game_name)
|
||||
|
||||
session.add(owner)
|
||||
session.add(base)
|
||||
|
||||
await bot.say('{}, your base named {} located at {} has been added'
|
||||
' to the database.'.format(ctx.message.author.mention, base.name, base.posToStr()))
|
||||
except ValueError:
|
||||
await bot.say('Invalid syntax, try again (?addbase [name] [x coord] [z coord] [Tunnel Color] [Tunnel Position])')
|
||||
|
||||
else :
|
||||
await bot.say('Allows you to add your base location to the database. '
|
||||
'Syntax: ?addbase [Base Name] [X Coordinate] [Z Coordinate]')
|
||||
|
||||
async def addbase(ctx, *args, ):
|
||||
'''
|
||||
Add your base to the database.
|
||||
The tunnel address is optional.
|
||||
?addbase [Base Name] [X Coordinate] [Y Coordinate] [Z Coordinate] [Tunnel Color] [Tunnel Position]
|
||||
'''
|
||||
|
||||
owner = Player(str(ctx.message.author.nick))
|
||||
base = Location(args, owner.in_game_name)
|
||||
|
||||
session.add(owner)
|
||||
session.add(base)
|
||||
|
||||
await bot.say('{}, your base named {} located at {} has been added'
|
||||
' to the database.'.format(ctx.message.author.mention, base.name, base.posToStr()))
|
||||
|
||||
|
||||
@bot.command(pass_context=True)
|
||||
async def findbase(ctx, * args):
|
||||
'''Allows you to find a base in the database.
|
||||
?findbase [Player name]
|
||||
'''
|
||||
if (len(args) > 0) :
|
||||
|
||||
baseList = session.query(Location).filter_by(owner=args[0]).all()
|
||||
|
||||
if (baseList is not None) :
|
||||
await bot.say('{}, {} has {} base(s):'.format(ctx.message.author.mention,args[0], len(baseList)))
|
||||
for base in baseList:
|
||||
await bot.say(base)
|
||||
else :
|
||||
await bot.say('{}, {} is not in the database'.format(ctx.message.author.mention, args[0]))
|
||||
else :
|
||||
await bot.say('Allows you to add your base location to the database. '
|
||||
'Syntax: ?findbase [Base Name] ')
|
||||
|
||||
#Bot Startup ******************************************************************
|
||||
try :
|
||||
file = open('token.dat','r')
|
||||
TOKEN = file.read()
|
||||
@bot.command(pass_context=True)
|
||||
async def findbase(ctx, *args):
|
||||
'''Allows you to find a base in the database.
|
||||
?findbase [Player name]
|
||||
'''
|
||||
search_key = args[0]
|
||||
baseList = session.query(Location).filter_by(owner=search_key).all()
|
||||
|
||||
if baseList is not None:
|
||||
await bot.say('{}, {} has {} base(s):'.format(ctx.message.author.mention, args[0], len(baseList)))
|
||||
for base in baseList:
|
||||
await bot.say(base)
|
||||
else:
|
||||
await bot.say('{}, {} is not in the database'.format(ctx.message.author.mention, args[0]))
|
||||
|
||||
|
||||
|
||||
# Bot Startup ******************************************************************
|
||||
try:
|
||||
file = open('token.dat', 'r')
|
||||
TOKEN = file.read()
|
||||
except FileNotFoundError:
|
||||
print('token.dat not found.')
|
||||
print('token.dat not found.')
|
||||
except IOError:
|
||||
print('Error reading token.dat')
|
||||
print('Error reading token.dat')
|
||||
|
||||
bot.run(TOKEN)
|
||||
bot.run(TOKEN)
|
||||
|
|
|
|||
Loading…
Reference in New Issue