bobloy
4 years ago
commit
36d6e5c9bf
@ -0,0 +1,5 @@
|
|||||||
|
from .launchlib import LaunchLib
|
||||||
|
|
||||||
|
|
||||||
|
def setup(bot):
|
||||||
|
bot.add_cog(LaunchLib(bot))
|
||||||
@ -0,0 +1,254 @@
|
|||||||
|
countryISOMapping = {
|
||||||
|
"AFG": "AF",
|
||||||
|
"ALA": "AX",
|
||||||
|
"ALB": "AL",
|
||||||
|
"DZA": "DZ",
|
||||||
|
"ASM": "AS",
|
||||||
|
"AND": "AD",
|
||||||
|
"AGO": "AO",
|
||||||
|
"AIA": "AI",
|
||||||
|
"ATA": "AQ",
|
||||||
|
"ATG": "AG",
|
||||||
|
"ARG": "AR",
|
||||||
|
"ARM": "AM",
|
||||||
|
"ABW": "AW",
|
||||||
|
"AUS": "AU",
|
||||||
|
"AUT": "AT",
|
||||||
|
"AZE": "AZ",
|
||||||
|
"BHS": "BS",
|
||||||
|
"BHR": "BH",
|
||||||
|
"BGD": "BD",
|
||||||
|
"BRB": "BB",
|
||||||
|
"BLR": "BY",
|
||||||
|
"BEL": "BE",
|
||||||
|
"BLZ": "BZ",
|
||||||
|
"BEN": "BJ",
|
||||||
|
"BMU": "BM",
|
||||||
|
"BTN": "BT",
|
||||||
|
"BOL": "BO",
|
||||||
|
"BIH": "BA",
|
||||||
|
"BWA": "BW",
|
||||||
|
"BVT": "BV",
|
||||||
|
"BRA": "BR",
|
||||||
|
"VGB": "VG",
|
||||||
|
"IOT": "IO",
|
||||||
|
"BRN": "BN",
|
||||||
|
"BGR": "BG",
|
||||||
|
"BFA": "BF",
|
||||||
|
"BDI": "BI",
|
||||||
|
"KHM": "KH",
|
||||||
|
"CMR": "CM",
|
||||||
|
"CAN": "CA",
|
||||||
|
"CPV": "CV",
|
||||||
|
"CYM": "KY",
|
||||||
|
"CAF": "CF",
|
||||||
|
"TCD": "TD",
|
||||||
|
"CHL": "CL",
|
||||||
|
"CHN": "CN",
|
||||||
|
"HKG": "HK",
|
||||||
|
"MAC": "MO",
|
||||||
|
"CXR": "CX",
|
||||||
|
"CCK": "CC",
|
||||||
|
"COL": "CO",
|
||||||
|
"COM": "KM",
|
||||||
|
"COG": "CG",
|
||||||
|
"COD": "CD",
|
||||||
|
"COK": "CK",
|
||||||
|
"CRI": "CR",
|
||||||
|
"CIV": "CI",
|
||||||
|
"HRV": "HR",
|
||||||
|
"CUB": "CU",
|
||||||
|
"CYP": "CY",
|
||||||
|
"CZE": "CZ",
|
||||||
|
"DNK": "DK",
|
||||||
|
"DJI": "DJ",
|
||||||
|
"DMA": "DM",
|
||||||
|
"DOM": "DO",
|
||||||
|
"ECU": "EC",
|
||||||
|
"EGY": "EG",
|
||||||
|
"SLV": "SV",
|
||||||
|
"GNQ": "GQ",
|
||||||
|
"ERI": "ER",
|
||||||
|
"EST": "EE",
|
||||||
|
"ETH": "ET",
|
||||||
|
"FLK": "FK",
|
||||||
|
"FRO": "FO",
|
||||||
|
"FJI": "FJ",
|
||||||
|
"FIN": "FI",
|
||||||
|
"FRA": "FR",
|
||||||
|
"GUF": "GF",
|
||||||
|
"PYF": "PF",
|
||||||
|
"ATF": "TF",
|
||||||
|
"GAB": "GA",
|
||||||
|
"GMB": "GM",
|
||||||
|
"GEO": "GE",
|
||||||
|
"DEU": "DE",
|
||||||
|
"GHA": "GH",
|
||||||
|
"GIB": "GI",
|
||||||
|
"GRC": "GR",
|
||||||
|
"GRL": "GL",
|
||||||
|
"GRD": "GD",
|
||||||
|
"GLP": "GP",
|
||||||
|
"GUM": "GU",
|
||||||
|
"GTM": "GT",
|
||||||
|
"GGY": "GG",
|
||||||
|
"GIN": "GN",
|
||||||
|
"GNB": "GW",
|
||||||
|
"GUY": "GY",
|
||||||
|
"HTI": "HT",
|
||||||
|
"HMD": "HM",
|
||||||
|
"VAT": "VA",
|
||||||
|
"HND": "HN",
|
||||||
|
"HUN": "HU",
|
||||||
|
"ISL": "IS",
|
||||||
|
"IND": "IN",
|
||||||
|
"IDN": "ID",
|
||||||
|
"IRN": "IR",
|
||||||
|
"IRQ": "IQ",
|
||||||
|
"IRL": "IE",
|
||||||
|
"IMN": "IM",
|
||||||
|
"ISR": "IL",
|
||||||
|
"ITA": "IT",
|
||||||
|
"JAM": "JM",
|
||||||
|
"JPN": "JP",
|
||||||
|
"JEY": "JE",
|
||||||
|
"JOR": "JO",
|
||||||
|
"KAZ": "KZ",
|
||||||
|
"KEN": "KE",
|
||||||
|
"KIR": "KI",
|
||||||
|
"PRK": "KP",
|
||||||
|
"KOR": "KR",
|
||||||
|
"KWT": "KW",
|
||||||
|
"KGZ": "KG",
|
||||||
|
"LAO": "LA",
|
||||||
|
"LVA": "LV",
|
||||||
|
"LBN": "LB",
|
||||||
|
"LSO": "LS",
|
||||||
|
"LBR": "LR",
|
||||||
|
"LBY": "LY",
|
||||||
|
"LIE": "LI",
|
||||||
|
"LTU": "LT",
|
||||||
|
"LUX": "LU",
|
||||||
|
"MKD": "MK",
|
||||||
|
"MDG": "MG",
|
||||||
|
"MWI": "MW",
|
||||||
|
"MYS": "MY",
|
||||||
|
"MDV": "MV",
|
||||||
|
"MLI": "ML",
|
||||||
|
"MLT": "MT",
|
||||||
|
"MHL": "MH",
|
||||||
|
"MTQ": "MQ",
|
||||||
|
"MRT": "MR",
|
||||||
|
"MUS": "MU",
|
||||||
|
"MYT": "YT",
|
||||||
|
"MEX": "MX",
|
||||||
|
"FSM": "FM",
|
||||||
|
"MDA": "MD",
|
||||||
|
"MCO": "MC",
|
||||||
|
"MNG": "MN",
|
||||||
|
"MNE": "ME",
|
||||||
|
"MSR": "MS",
|
||||||
|
"MAR": "MA",
|
||||||
|
"MOZ": "MZ",
|
||||||
|
"MMR": "MM",
|
||||||
|
"NAM": "NA",
|
||||||
|
"NRU": "NR",
|
||||||
|
"NPL": "NP",
|
||||||
|
"NLD": "NL",
|
||||||
|
"ANT": "AN",
|
||||||
|
"NCL": "NC",
|
||||||
|
"NZL": "NZ",
|
||||||
|
"NIC": "NI",
|
||||||
|
"NER": "NE",
|
||||||
|
"NGA": "NG",
|
||||||
|
"NIU": "NU",
|
||||||
|
"NFK": "NF",
|
||||||
|
"MNP": "MP",
|
||||||
|
"NOR": "NO",
|
||||||
|
"OMN": "OM",
|
||||||
|
"PAK": "PK",
|
||||||
|
"PLW": "PW",
|
||||||
|
"PSE": "PS",
|
||||||
|
"PAN": "PA",
|
||||||
|
"PNG": "PG",
|
||||||
|
"PRY": "PY",
|
||||||
|
"PER": "PE",
|
||||||
|
"PHL": "PH",
|
||||||
|
"PCN": "PN",
|
||||||
|
"POL": "PL",
|
||||||
|
"PRT": "PT",
|
||||||
|
"PRI": "PR",
|
||||||
|
"QAT": "QA",
|
||||||
|
"REU": "RE",
|
||||||
|
"ROU": "RO",
|
||||||
|
"RUS": "RU",
|
||||||
|
"RWA": "RW",
|
||||||
|
"BLM": "BL",
|
||||||
|
"SHN": "SH",
|
||||||
|
"KNA": "KN",
|
||||||
|
"LCA": "LC",
|
||||||
|
"MAF": "MF",
|
||||||
|
"SPM": "PM",
|
||||||
|
"VCT": "VC",
|
||||||
|
"WSM": "WS",
|
||||||
|
"SMR": "SM",
|
||||||
|
"STP": "ST",
|
||||||
|
"SAU": "SA",
|
||||||
|
"SEN": "SN",
|
||||||
|
"SRB": "RS",
|
||||||
|
"SYC": "SC",
|
||||||
|
"SLE": "SL",
|
||||||
|
"SGP": "SG",
|
||||||
|
"SVK": "SK",
|
||||||
|
"SVN": "SI",
|
||||||
|
"SLB": "SB",
|
||||||
|
"SOM": "SO",
|
||||||
|
"ZAF": "ZA",
|
||||||
|
"SGS": "GS",
|
||||||
|
"SSD": "SS",
|
||||||
|
"ESP": "ES",
|
||||||
|
"LKA": "LK",
|
||||||
|
"SDN": "SD",
|
||||||
|
"SUR": "SR",
|
||||||
|
"SJM": "SJ",
|
||||||
|
"SWZ": "SZ",
|
||||||
|
"SWE": "SE",
|
||||||
|
"CHE": "CH",
|
||||||
|
"SYR": "SY",
|
||||||
|
"TWN": "TW",
|
||||||
|
"TJK": "TJ",
|
||||||
|
"TZA": "TZ",
|
||||||
|
"THA": "TH",
|
||||||
|
"TLS": "TL",
|
||||||
|
"TGO": "TG",
|
||||||
|
"TKL": "TK",
|
||||||
|
"TON": "TO",
|
||||||
|
"TTO": "TT",
|
||||||
|
"TUN": "TN",
|
||||||
|
"TUR": "TR",
|
||||||
|
"TKM": "TM",
|
||||||
|
"TCA": "TC",
|
||||||
|
"TUV": "TV",
|
||||||
|
"UGA": "UG",
|
||||||
|
"UKR": "UA",
|
||||||
|
"ARE": "AE",
|
||||||
|
"GBR": "GB",
|
||||||
|
"USA": "US",
|
||||||
|
"UMI": "UM",
|
||||||
|
"URY": "UY",
|
||||||
|
"UZB": "UZ",
|
||||||
|
"VUT": "VU",
|
||||||
|
"VEN": "VE",
|
||||||
|
"VNM": "VN",
|
||||||
|
"VIR": "VI",
|
||||||
|
"WLF": "WF",
|
||||||
|
"ESH": "EH",
|
||||||
|
"YEM": "YE",
|
||||||
|
"ZMB": "ZM",
|
||||||
|
"ZWE": "ZW",
|
||||||
|
"XKX": "XK",
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
def country_mapping(countrycode):
|
||||||
|
return countryISOMapping[countrycode]
|
||||||
@ -0,0 +1,20 @@
|
|||||||
|
{
|
||||||
|
"author": [
|
||||||
|
"Bobloy"
|
||||||
|
],
|
||||||
|
"min_bot_version": "3.4.0",
|
||||||
|
"description": "Pull information from the Launch Library API for space flights",
|
||||||
|
"hidden": false,
|
||||||
|
"install_msg": "Thank you for installing LaunchLib. Get started with `[p]load launchlib`, then `[p]help LaunchLib`",
|
||||||
|
"short": "Access launch data for space flights",
|
||||||
|
"end_user_data_statement": "This cog does not store any End User Data",
|
||||||
|
"requirements": ["python-launch-library"],
|
||||||
|
"tags": [
|
||||||
|
"bobloy",
|
||||||
|
"utils",
|
||||||
|
"launch",
|
||||||
|
"space",
|
||||||
|
"api",
|
||||||
|
"library"
|
||||||
|
]
|
||||||
|
}
|
||||||
@ -0,0 +1,130 @@
|
|||||||
|
import asyncio
|
||||||
|
import functools
|
||||||
|
import logging
|
||||||
|
|
||||||
|
import discord
|
||||||
|
import launchlibrary as ll
|
||||||
|
from redbot.core import Config, commands
|
||||||
|
from redbot.core.bot import Red
|
||||||
|
|
||||||
|
from launchlib.countrymapper import country_mapping
|
||||||
|
|
||||||
|
log = logging.getLogger("red.fox_v3.launchlib")
|
||||||
|
|
||||||
|
|
||||||
|
class LaunchLib(commands.Cog):
|
||||||
|
"""
|
||||||
|
Cog Description
|
||||||
|
|
||||||
|
Less important information about the cog
|
||||||
|
"""
|
||||||
|
|
||||||
|
def __init__(self, bot: Red):
|
||||||
|
super().__init__()
|
||||||
|
self.bot = bot
|
||||||
|
self.config = Config.get_conf(self, identifier=0, force_registration=True)
|
||||||
|
|
||||||
|
default_guild = {}
|
||||||
|
|
||||||
|
self.config.register_guild(**default_guild)
|
||||||
|
|
||||||
|
self.api = ll.Api()
|
||||||
|
|
||||||
|
async def red_delete_data_for_user(self, **kwargs):
|
||||||
|
"""Nothing to delete"""
|
||||||
|
return
|
||||||
|
|
||||||
|
async def _embed_launch_data(self, launch: ll.Launch):
|
||||||
|
status: ll.LaunchStatus = launch.get_status()
|
||||||
|
|
||||||
|
rocket: ll.Rocket = launch.rocket
|
||||||
|
|
||||||
|
title = launch.name
|
||||||
|
description = status.description
|
||||||
|
|
||||||
|
urls = launch.vid_urls + launch.info_urls
|
||||||
|
if not urls and rocket:
|
||||||
|
urls = rocket.info_urls + [rocket.wiki_url]
|
||||||
|
if urls:
|
||||||
|
url = urls[0]
|
||||||
|
else:
|
||||||
|
url = None
|
||||||
|
|
||||||
|
color = discord.Color.green() if status.id in [1, 3] else discord.Color.red()
|
||||||
|
|
||||||
|
em = discord.Embed(title=title, description=description, url=url, color=color)
|
||||||
|
|
||||||
|
if rocket and rocket.image_url and rocket.image_url != "Array":
|
||||||
|
em.set_image(url=rocket.image_url)
|
||||||
|
|
||||||
|
agency = getattr(launch, "agency", None)
|
||||||
|
if agency is not None:
|
||||||
|
em.set_author(
|
||||||
|
name=agency.name,
|
||||||
|
url=agency.wiki_url,
|
||||||
|
icon_url=f"https://www.countryflags.io/{country_mapping(agency.country_code)}/flat/64.png",
|
||||||
|
)
|
||||||
|
|
||||||
|
field_options = [
|
||||||
|
("failreason", "Fail Reason"),
|
||||||
|
("holdreason", "Hold Reason"),
|
||||||
|
("id", "ID"),
|
||||||
|
("hashtag", "Hashtag"),
|
||||||
|
]
|
||||||
|
for f in field_options:
|
||||||
|
data = getattr(launch, f[0], None)
|
||||||
|
if data is not None and data:
|
||||||
|
em.add_field(name=f[1], value=data)
|
||||||
|
|
||||||
|
if launch.missions:
|
||||||
|
field_options = [
|
||||||
|
("description", "Mission Description"),
|
||||||
|
("typeName", "Mission Type"),
|
||||||
|
("name", "Mission Name"),
|
||||||
|
]
|
||||||
|
for mission in launch.missions:
|
||||||
|
for f in field_options:
|
||||||
|
data = mission.get(f[0], None)
|
||||||
|
if data is not None and data:
|
||||||
|
em.add_field(name=f[1], value=data)
|
||||||
|
|
||||||
|
if rocket and rocket.family:
|
||||||
|
em.add_field(name="Rocket Family", value=rocket.family)
|
||||||
|
|
||||||
|
em.timestamp = launch.windowstart
|
||||||
|
|
||||||
|
em.set_footer()
|
||||||
|
|
||||||
|
return em
|
||||||
|
|
||||||
|
@commands.group()
|
||||||
|
async def launchlib(self, ctx: commands.Context):
|
||||||
|
if ctx.invoked_subcommand is None:
|
||||||
|
pass
|
||||||
|
|
||||||
|
@launchlib.command()
|
||||||
|
async def next(self, ctx: commands.Context, num_launches: int = 1):
|
||||||
|
# launches = await api.async_next_launches(num_launches)
|
||||||
|
loop = asyncio.get_running_loop()
|
||||||
|
|
||||||
|
launches = await loop.run_in_executor(
|
||||||
|
None, functools.partial(self.api.fetch_launch, num=num_launches)
|
||||||
|
)
|
||||||
|
|
||||||
|
# launches = self.api.fetch_launch(num=num_launches)
|
||||||
|
|
||||||
|
print(len(launches))
|
||||||
|
|
||||||
|
async with ctx.typing():
|
||||||
|
for x, launch in enumerate(launches):
|
||||||
|
if x >= num_launches:
|
||||||
|
return
|
||||||
|
|
||||||
|
em = await self._embed_launch_data(launch)
|
||||||
|
if em is not None:
|
||||||
|
try:
|
||||||
|
await ctx.send(embed=em)
|
||||||
|
except discord.HTTPException:
|
||||||
|
await ctx.send(str(launch))
|
||||||
|
log.exception("Failed to send embed")
|
||||||
|
await asyncio.sleep(2)
|
||||||
Loading…
Reference in new issue