Added global error handling

doc_update
Joey Hines 2018-05-25 11:32:29 -05:00
parent 60846c81d1
commit ee3bfb0fee
1 changed files with 142 additions and 128 deletions

View File

@ -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()
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'
__tablename__ = 'Players'
id = Column(Integer, primary_key=True)
in_game_name = Column(String)
id = Column(Integer, primary_key=True)
in_game_name = Column(String)
def __init__(self,name) :
self.in_game_name = name
def __init__(self, name):
self.in_game_name = name
class Location(SQL_Base):
__tablename__ = 'Locations'
__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'))
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
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])
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 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 __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')
@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]')
'''Check if the bot is alive.'''
await bot.say('I\'m here you ding dong')
@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) :
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]
'''
baseList = session.query(Location).filter_by(owner=args[0]).all()
owner = Player(str(ctx.message.author.nick))
base = Location(args, owner.in_game_name)
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] ')
session.add(owner)
session.add(base)
#Bot Startup ******************************************************************
try :
file = open('token.dat','r')
TOKEN = file.read()
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]
'''
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)