def test_no_token(self):
mp = MapBox()
with pytest.raises(BaseMap.AuthMissingError):
_ = mp.get_geocode("The Moon")
class GeoCog(commands.Cog):
def __init__(self, bot):
self.mapbox = MapBox(token=get_token())
self.gbif = GBIF()
self.ebird = eBirdMap()
self.typesense = ebl.TypeSenseSearch(api_key="changeMe!")
self.typesense.connect()
self.meili = ebl.MeilisearchSearch(api_key="changeMe!")
self.meili.connect()
self.map_size = 512
def find_species_from_name(self, arg, backend):
try:
r = backend.name_to_codes(arg, "all")
return r
except Exception:
return None
@commands.command(
brief="Gets a lat/lon from an address, or vice versa.",
help="Gets a lat/lon from an address, or vice versa. Mapbox version",
usage="[query]",
)
async def geocode(self, ctx, *, arg):
res = self.mapbox.get_geocode(arg)
await ctx.send(res)
@commands.command(
brief="Gets a test range map.",
help="Gets a test range map. Just Bushtits from GBIF.",
usage="[query]",
)
async def gbifmap(self, ctx, *, arg):
print("~~~~~~~~~~~~~^^~~~~~~~~~~~~~~~")
print("gbifmap: ", arg)
scientific_name, taxon_id = self.gbif.lookup_species(arg)
if all((scientific_name, taxon_id)):
start = datetime.now()
result = generate_gbif_mapbox_range(taxon_id, self.gbif,
self.mapbox, self.map_size)
dur = datetime.now() - start
if result is None:
embed = discord.Embed(
title="Error:",
description="Range map creation failed.",
color=0xFF0000,
)
await ctx.send(embed=embed)
desc = f"Source: GBIF, Mapbox. GBIF taxon id: {taxon_id}.\nDebug: {dur.seconds}s"
gbif_url = f"https://www.gbif.org/species/{taxon_id}"
embed = discord.Embed(title=scientific_name,
description=desc,
url=gbif_url,
color=0x007F00)
file = discord.File(result.asbytes(), filename=f"{taxon_id}.png")
await ctx.send(file=file, embed=embed)
else:
embed = discord.Embed(title="Error:",
description=f"Lookup of {arg} failed.",
color=0xFF0000)
await ctx.send(embed=embed)
@commands.command(
brief="Test GBIF species lookup.",
help="Test GBIF species lookup.",
usage="[query]",
)
async def glookup(self, ctx, *, arg):
scientific_name, taxon_id = self.gbif.lookup_species(arg)
if all((scientific_name, taxon_id)):
await ctx.send(f"{arg} -> {scientific_name}, id: {taxon_id}.")
else:
await ctx.send("Lookup failed.")
@commands.command(
brief="Gets a lat/lon from an address, or vice versa.",
help="Gets a lat/lon from an address, or vice versa. Mapbox version",
usage="[query]",
)
async def ebirdmap(self, ctx, *, arg):
ab = randint(0, 1)
backend = [self.typesense, self.meili][ab]
backend_name = ["typesense", "meili"][ab]
res = self.find_species_from_name(arg, backend)
res_img = None
if res:
print("res:", res, backend_name)
species_code = res["species_code"]
common_name = res["name"]
scientific_name = res["scientific_name"]
title = f"{common_name} (_{scientific_name}_)."
ebird_url = f"{self.ebird.species_url}{species_code}"
try:
start = datetime.now()
res_img, no_data = self.ebird.make_map(species_code,
self.mapbox, 512)
img = res_img.asbytes()
if not no_data:
res_img.save(f"geocog-ebird-{species_code}.png")
end = datetime.now()
desc = "Source: eBird, Mapbox."
desc += f"\nDebug: generated in: {(end - start).seconds}s. Search: {backend_name}."
embed = discord.Embed(title=title,
url=ebird_url,
description=desc,
color=0x7F007F)
file = discord.File(img, filename=f"{species_code}.png")
else:
desc = "**No data on eBird**.\nSource: eBird, Mapbox."
desc += f"\nDebug: no data, Search: {backend_name}."
embed = discord.Embed(title=title,
url=ebird_url,
description=desc,
color=0x7F0000)
file = discord.File(img, filename=f"{species_code}.png")
except Exception:
traceback.print_exc()
pass
if res_img:
await ctx.send(file=file, embed=embed)
elif not res:
embed = discord.Embed(title="Error:",
description=f"Lookup of {arg} failed.",
color=0xFF0000)
await ctx.send(embed=embed)
else:
embed = discord.Embed(
title="Error:",
description="Range map creation failed.",
color=0xFF0000,
)
await ctx.send(embed=embed)