Wrapped sqlalchemy calls more into GeoffreyDatabase
parent
c642f535f8
commit
2320acf7bf
|
@ -18,16 +18,59 @@ class GeoffreyDatabase:
|
||||||
self.session = Session()
|
self.session = Session()
|
||||||
SQL_Base.metadata.create_all(self.engine)
|
SQL_Base.metadata.create_all(self.engine)
|
||||||
|
|
||||||
|
def add_base(self, player_name, name, x_pos, y_pos, z_pos, args):
|
||||||
|
owner = self.add_player(player_name)
|
||||||
|
base = Location(name, x_pos, y_pos, z_pos, owner, args)
|
||||||
|
self.add_object(base)
|
||||||
|
|
||||||
|
return base
|
||||||
|
|
||||||
|
def add_player(self, player_name):
|
||||||
|
expr = Player.name == player_name
|
||||||
|
player_list = self.query_by_filter(Player, expr)
|
||||||
|
|
||||||
|
if len(player_list) == 0:
|
||||||
|
uuid = grab_UUID(player_name)
|
||||||
|
expr = Player.id == uuid
|
||||||
|
player_list = self.query_by_filter(Player, expr)
|
||||||
|
|
||||||
|
if len(player_list) == 0:
|
||||||
|
player = Player(player_name)
|
||||||
|
self.add_object(player)
|
||||||
|
else:
|
||||||
|
player_list[0].name = player_name
|
||||||
|
else:
|
||||||
|
player = player_list[0]
|
||||||
|
|
||||||
|
return player
|
||||||
|
|
||||||
def add_object(self, obj):
|
def add_object(self, obj):
|
||||||
ret = not self.session.query(exists().where(type(obj).id == obj.id))
|
ret = not self.session.query(exists().where(type(obj).id == obj.id))
|
||||||
if not ret:
|
if not ret:
|
||||||
self.session.add(obj)
|
self.session.add(obj)
|
||||||
self.session.commit()
|
self.session.commit()
|
||||||
|
|
||||||
|
def find_base_by_owner(self, owner_name):
|
||||||
|
player = self.add_player(owner_name)
|
||||||
|
expr = Location.owner == player
|
||||||
|
return self.query_by_filter(Location, expr)
|
||||||
|
|
||||||
|
def find_base_around(self, x_pos, z_pos, radius):
|
||||||
|
expr = (Location.x < x_pos + radius) & (Location.x > x_pos - radius) & (Location.z < z_pos + radius) & \
|
||||||
|
(Location.z > z_pos - radius)
|
||||||
|
|
||||||
|
return self.query_by_filter(Location, expr)
|
||||||
|
|
||||||
def query_by_filter(self, obj_type, * args):
|
def query_by_filter(self, obj_type, * args):
|
||||||
filter_value = self.combine_filter(args)
|
filter_value = self.combine_filter(args)
|
||||||
return self.session.query(obj_type).filter(filter_value).all()
|
return self.session.query(obj_type).filter(filter_value).all()
|
||||||
|
|
||||||
|
def delete_base(self, player_name, base_name):
|
||||||
|
player = self.add_player(player_name)
|
||||||
|
expr = (Location.owner == player) & (Location.name == base_name)
|
||||||
|
|
||||||
|
self.delete_entry(Location, expr)
|
||||||
|
|
||||||
def delete_entry(self, obj_type, * args):
|
def delete_entry(self, obj_type, * args):
|
||||||
filter_value = self.combine_filter(args)
|
filter_value = self.combine_filter(args)
|
||||||
entry = self.session.query(obj_type).filter(filter_value)
|
entry = self.session.query(obj_type).filter(filter_value)
|
||||||
|
@ -46,6 +89,7 @@ class GeoffreyDatabase:
|
||||||
s = s + '\n' + obj.id
|
s = s + '\n' + obj.id
|
||||||
return s
|
return s
|
||||||
|
|
||||||
|
|
||||||
def combine_filter(self, filter_value):
|
def combine_filter(self, filter_value):
|
||||||
return sqlalchemy.sql.expression.and_(filter_value[0])
|
return sqlalchemy.sql.expression.and_(filter_value[0])
|
||||||
|
|
||||||
|
|
91
Geoffrey.py
91
Geoffrey.py
|
@ -3,7 +3,6 @@ from DatabaseModels import *
|
||||||
from BotErrors import *
|
from BotErrors import *
|
||||||
from MinecraftAccountInfoGrabber import *
|
from MinecraftAccountInfoGrabber import *
|
||||||
|
|
||||||
|
|
||||||
TOKEN = ''
|
TOKEN = ''
|
||||||
command_prefix = '?'
|
command_prefix = '?'
|
||||||
description = '''
|
description = '''
|
||||||
|
@ -21,8 +20,9 @@ bot = commands.Bot(command_prefix=command_prefix, description=description, case_
|
||||||
database = GeoffreyDatabase('sqlite:///:memory:')
|
database = GeoffreyDatabase('sqlite:///:memory:')
|
||||||
|
|
||||||
|
|
||||||
|
# Bot Commands ******************************************************************'
|
||||||
|
|
||||||
|
|
||||||
# Bot Commands ******************************************************************
|
|
||||||
@bot.event
|
@bot.event
|
||||||
async def on_ready():
|
async def on_ready():
|
||||||
print('GeoffreyBot')
|
print('GeoffreyBot')
|
||||||
|
@ -53,6 +53,7 @@ async def test():
|
||||||
'''
|
'''
|
||||||
await bot.say('I\'m here you ding dong')
|
await bot.say('I\'m here you ding dong')
|
||||||
|
|
||||||
|
|
||||||
@bot.command(pass_context=True)
|
@bot.command(pass_context=True)
|
||||||
async def addbase(ctx, name: str, x_pos: int, y_pos: int, z_pos: int, * args):
|
async def addbase(ctx, name: str, x_pos: int, y_pos: int, z_pos: int, * args):
|
||||||
'''
|
'''
|
||||||
|
@ -61,46 +62,13 @@ async def addbase(ctx, name: str, x_pos: int, y_pos: int, z_pos: int, * args):
|
||||||
?addbase [Base Name] [X Coordinate] [Y Coordinate] [Z Coordinate] [Tunnel Color] [Tunnel Position]
|
?addbase [Base Name] [X Coordinate] [Y Coordinate] [Z Coordinate] [Tunnel Color] [Tunnel Position]
|
||||||
'''
|
'''
|
||||||
|
|
||||||
if ctx.message.author.nick is None:
|
player_name = get_nickname(ctx.message.author)
|
||||||
player_name = ctx.message.author.display_name
|
|
||||||
else:
|
|
||||||
player_name = ctx.message.author.nick
|
|
||||||
|
|
||||||
expr = Player.name == player_name
|
|
||||||
player_list = database.query_by_filter(Player, expr)
|
|
||||||
|
|
||||||
if len(player_list) == 0:
|
|
||||||
uuid = grab_UUID(player_name)
|
|
||||||
expr = Player.id == uuid
|
|
||||||
player_list = database.query_by_filter(Player, expr)
|
|
||||||
|
|
||||||
if len(player_list) == 0:
|
|
||||||
owner = Player(player_name)
|
|
||||||
else:
|
|
||||||
player_list[0].name == player_name
|
|
||||||
else:
|
|
||||||
owner = player_list[0]
|
|
||||||
|
|
||||||
try:
|
try:
|
||||||
base = Location(name, x_pos, y_pos, z_pos, owner, args)
|
base = database.add_base(player_name, name, x_pos, y_pos, z_pos, args)
|
||||||
except LocationInitError:
|
except LocationInitError:
|
||||||
raise commands.UserInputError
|
raise commands.UserInputError
|
||||||
|
|
||||||
owner.locations.append(base)
|
|
||||||
|
|
||||||
database.add_object(base)
|
|
||||||
database.add_object(owner)
|
|
||||||
|
|
||||||
#expr = Player.id == owner.id
|
|
||||||
#player_list = database.query_by_filter(Player, expr)
|
|
||||||
'''
|
|
||||||
if len(player_list) == 0:
|
|
||||||
database.add_object(owner)
|
|
||||||
else:
|
|
||||||
if player_list[0].name != owner.name:
|
|
||||||
player_list[0].name = owner.name
|
|
||||||
database.session.commit()
|
|
||||||
'''
|
|
||||||
await bot.say('{}, your base named {} located at {} has been added'
|
await bot.say('{}, your base named {} located at {} has been added'
|
||||||
' to the database.'.format(ctx.message.author.mention, base.name, base.pos_to_str()))
|
' to the database.'.format(ctx.message.author.mention, base.name, base.pos_to_str()))
|
||||||
|
|
||||||
|
@ -112,13 +80,7 @@ async def findbase(ctx, name: str):
|
||||||
?findbase [Player name]
|
?findbase [Player name]
|
||||||
'''
|
'''
|
||||||
|
|
||||||
uuid = grab_UUID(name)
|
base_list = database.find_base_by_owner(name)
|
||||||
expr = Player.id == uuid
|
|
||||||
p = database.query_by_filter(Player, expr)[0]
|
|
||||||
|
|
||||||
expr = Location.owner == p
|
|
||||||
base_list = database.query_by_filter(Location, expr)
|
|
||||||
|
|
||||||
|
|
||||||
if len(base_list) != 0:
|
if len(base_list) != 0:
|
||||||
base_string = base_list_string(base_list, '{} \n{}')
|
base_string = base_list_string(base_list, '{} \n{}')
|
||||||
|
@ -129,15 +91,6 @@ async def findbase(ctx, name: str):
|
||||||
await bot.say('{}, the player {} is not in the database'.format(ctx.message.author.mention, name))
|
await bot.say('{}, the player {} is not in the database'.format(ctx.message.author.mention, name))
|
||||||
|
|
||||||
|
|
||||||
def base_list_string(base_list, str_format):
|
|
||||||
base_string = ''
|
|
||||||
|
|
||||||
for base in base_list:
|
|
||||||
base_string = str_format.format(base_string, base)
|
|
||||||
|
|
||||||
return base_string
|
|
||||||
|
|
||||||
|
|
||||||
@bot.command(pass_context=True)
|
@bot.command(pass_context=True)
|
||||||
async def deletebase(ctx, name: str):
|
async def deletebase(ctx, name: str):
|
||||||
'''
|
'''
|
||||||
|
@ -145,12 +98,9 @@ async def deletebase(ctx, name: str):
|
||||||
?deletebase [Base name]
|
?deletebase [Base name]
|
||||||
'''
|
'''
|
||||||
|
|
||||||
uuid = grab_UUID(str(ctx.message.author.nick))
|
player_name = get_nickname(ctx.message.author)
|
||||||
|
|
||||||
expr = (Location.owner_id == uuid) & (Location.name == name)
|
|
||||||
|
|
||||||
try:
|
try:
|
||||||
database.delete_entry(Location, expr)
|
database.delete_base(player_name, name)
|
||||||
await bot.say('{}, your base named "{}" has been deleted.'.format(ctx.message.author.mention, name))
|
await bot.say('{}, your base named "{}" has been deleted.'.format(ctx.message.author.mention, name))
|
||||||
except DeleteEntryError:
|
except DeleteEntryError:
|
||||||
await bot.say('{}, you do not have a base named "{}".'.format(ctx.message.author.mention, name))
|
await bot.say('{}, you do not have a base named "{}".'.format(ctx.message.author.mention, name))
|
||||||
|
@ -171,10 +121,7 @@ async def findbasearound(ctx, x_pos: int, z_pos: int, * args):
|
||||||
except ValueError:
|
except ValueError:
|
||||||
raise commands.UserInputError
|
raise commands.UserInputError
|
||||||
|
|
||||||
expr = (Location.x < x_pos + radius) & (Location.x > x_pos - radius) & (Location.z < z_pos + radius) & \
|
base_list = database.find_base_around(x_pos, z_pos, radius)
|
||||||
(Location.z > z_pos - radius)
|
|
||||||
|
|
||||||
base_list = database.query_by_filter(Location, expr)
|
|
||||||
|
|
||||||
if len(base_list) != 0:
|
if len(base_list) != 0:
|
||||||
base_string = base_list_string(base_list, '{} \n{}')
|
base_string = base_list_string(base_list, '{} \n{}')
|
||||||
|
@ -185,7 +132,27 @@ async def findbasearound(ctx, x_pos: int, z_pos: int, * args):
|
||||||
await bot.say('{}, there are no bases within {} blocks of that point'
|
await bot.say('{}, there are no bases within {} blocks of that point'
|
||||||
.format(ctx.message.author.mention, radius))
|
.format(ctx.message.author.mention, radius))
|
||||||
|
|
||||||
|
# Helper Functions ************************************************************
|
||||||
|
|
||||||
|
|
||||||
|
def get_nickname(discord_user) :
|
||||||
|
if discord_user.nick is None:
|
||||||
|
return discord_user.display_name
|
||||||
|
else:
|
||||||
|
return discord_user.nick
|
||||||
|
|
||||||
|
|
||||||
|
def base_list_string(base_list, str_format):
|
||||||
|
base_string = ''
|
||||||
|
|
||||||
|
for base in base_list:
|
||||||
|
base_string = str_format.format(base_string, base)
|
||||||
|
|
||||||
|
return base_string
|
||||||
|
|
||||||
# Bot Startup ******************************************************************
|
# Bot Startup ******************************************************************
|
||||||
|
|
||||||
|
|
||||||
try:
|
try:
|
||||||
file = open('token.dat', 'r')
|
file = open('token.dat', 'r')
|
||||||
TOKEN = file.read()
|
TOKEN = file.read()
|
||||||
|
|
Loading…
Reference in New Issue