[V3 Timezone] Initial commit
This commit is contained in:
aikaterna
@@ -23,6 +23,8 @@ retrosign - A v3 port of Anismash's retrosign cog: https://github.com/Anismash/A
|
||||
|
||||
rndstatus - A v3 port of Twentysix's rndstatus cog with a couple extra settings.
|
||||
|
||||
timezone - A v3 port of Fishyfing's timezone cog with a few improvements.
|
||||
|
||||
tools - A collection of mod and admin tools, ported from my v2 version. Sitryk is responsible for a lot of the code in tools... thanks for the help with this cog.
|
||||
|
||||
wolfram - A v3 port of Paddo's abandoned Wolfram Alpha cog.
|
||||
|
||||
5
timezone/__init__.py
Normal file
5
timezone/__init__.py
Normal file
@@ -0,0 +1,5 @@
|
||||
from .timezone import Timezone
|
||||
|
||||
|
||||
def setup(bot):
|
||||
bot.add_cog(Timezone(bot))
|
||||
16
timezone/info.json
Normal file
16
timezone/info.json
Normal file
@@ -0,0 +1,16 @@
|
||||
{
|
||||
"author": [
|
||||
"aikaterna",
|
||||
"fishyfing"
|
||||
],
|
||||
"description": "Check timezones, user times, or times in specific places. Originally made by Fishyfing.",
|
||||
"short": "Check times for users and places.",
|
||||
"tags": [
|
||||
"time",
|
||||
"timezone"
|
||||
],
|
||||
"requirements": [
|
||||
"pytz"
|
||||
],
|
||||
"type": "COG"
|
||||
}
|
||||
183
timezone/timezone.py
Normal file
183
timezone/timezone.py
Normal file
@@ -0,0 +1,183 @@
|
||||
import aiohttp
|
||||
import discord
|
||||
import os
|
||||
import pytz
|
||||
from datetime import datetime
|
||||
from pytz import all_timezones
|
||||
from pytz import country_timezones
|
||||
from redbot.core import Config, commands, checks
|
||||
|
||||
|
||||
class Timezone:
|
||||
"""Gets times across the world..."""
|
||||
|
||||
def __init__(self, bot):
|
||||
self.bot = bot
|
||||
self.config = Config.get_conf(self, 278049241001, force_registration=True)
|
||||
|
||||
default_member = {"usertime": None}
|
||||
|
||||
self.config.register_member(**default_member)
|
||||
self.session = aiohttp.ClientSession()
|
||||
|
||||
def __unload(self):
|
||||
self.session.detach()
|
||||
|
||||
@commands.guild_only()
|
||||
@commands.group()
|
||||
async def time(self, ctx):
|
||||
"""Checks the time.
|
||||
|
||||
For the list of supported timezones, see here: https://en.wikipedia.org/wiki/List_of_tz_database_time_zones"""
|
||||
pass
|
||||
|
||||
@time.command()
|
||||
async def tz(self, ctx, *, tz):
|
||||
"""Gets the time in any timezone."""
|
||||
try:
|
||||
if tz == "":
|
||||
time = datetime.now()
|
||||
fmt = "**%H:%M** %d-%B-%Y"
|
||||
await ctx.send(f"Current system time: {time.strftime(fmt)}")
|
||||
else:
|
||||
fmt = "**%H:%M** %d-%B-%Y **%Z (UTC %z)**"
|
||||
if "'" in tz:
|
||||
tz = tz.replace("'", "")
|
||||
if len(tz) > 4 and "/" not in tz:
|
||||
await ctx.send(
|
||||
"Error: Incorrect format. Use:\n **Continent/City** with correct capitals. e.g. `America/New_York`\n See the full list of supported timezones here:\n <https://en.wikipedia.org/wiki/List_of_tz_database_time_zones>"
|
||||
)
|
||||
else:
|
||||
time = datetime.now(pytz.timezone(tz))
|
||||
await ctx.send(time.strftime(fmt))
|
||||
except Exception as e:
|
||||
await ctx.send(f"**Error:** {str(e)} is an unsupported timezone.")
|
||||
|
||||
@time.command()
|
||||
async def iso(self, ctx, *, code):
|
||||
"""Looks up ISO3166 country codes and gives you a supported timezone."""
|
||||
if code == "":
|
||||
await ctx.send("That doesn't look like a country code!")
|
||||
else:
|
||||
if code in country_timezones:
|
||||
exist = True
|
||||
else:
|
||||
exist = False
|
||||
if exist == True:
|
||||
msg = f"Supported timezones for **{code}:**\n"
|
||||
tz = str(country_timezones(code))
|
||||
tz = tz[:-1]
|
||||
tz = tz[1:]
|
||||
msg += tz
|
||||
msg += f"\n**Use** `[p]time tz Continent/City` **to display the current time in that timezone.**"
|
||||
await ctx.send(msg)
|
||||
else:
|
||||
await ctx.send(
|
||||
"That code isn't supported. For a full list, see here: <https://en.wikipedia.org/wiki/List_of_tz_database_time_zones>"
|
||||
)
|
||||
|
||||
@time.command()
|
||||
async def me(self, ctx, *tz):
|
||||
"""Sets your timezone.
|
||||
Usage: [p]time me Continent/City"""
|
||||
tz = " ".join(tz)
|
||||
if tz in all_timezones:
|
||||
exist = True
|
||||
else:
|
||||
exist = False
|
||||
usertime = await self.config.member(ctx.message.author).usertime()
|
||||
if tz == "":
|
||||
if not usertime:
|
||||
await ctx.send(
|
||||
"You haven't set your timezone. Do `[p]time me Continent/City`: see <https://en.wikipedia.org/wiki/List_of_tz_database_time_zones>"
|
||||
)
|
||||
else:
|
||||
msg = f"Your current timezone is **{usertime}.**\n"
|
||||
time = datetime.now(pytz.timezone(usertime))
|
||||
time = time.strftime("**%H:%M** %d-%B-%Y **%Z (UTC %z)**")
|
||||
msg += f"The current time is: {time}"
|
||||
await ctx.send(msg)
|
||||
elif exist == True:
|
||||
if "'" in tz:
|
||||
tz = tz.replace("'", "")
|
||||
await self.config.member(ctx.message.author).usertime.set(tz)
|
||||
await ctx.send(f"Successfully set your timezone to **{tz}**.")
|
||||
else:
|
||||
await ctx.send(
|
||||
"**Error:** Unrecognized timezone. Try `[p]time me Continent/City`: see <https://en.wikipedia.org/wiki/List_of_tz_database_time_zones>"
|
||||
)
|
||||
|
||||
@time.command()
|
||||
async def place(self, ctx, *, city):
|
||||
"""Shows what time it is in other places."""
|
||||
city = city.replace(" ", "%20")
|
||||
|
||||
async with self.session.request("GET", f"https://timezoneapi.io/api/address/?{city}") as r:
|
||||
place_json = await r.json()
|
||||
|
||||
if place_json["data"]["addresses_found"] != "0":
|
||||
execution_time = place_json["meta"]["execution_time"]
|
||||
city_state_country = place_json["data"]["addresses"][0]["formatted_address"]
|
||||
twelve_hour_first = place_json["data"]["addresses"][0]["datetime"]["hour_12_wolz"]
|
||||
twelve_hour_second = place_json["data"]["addresses"][0]["datetime"]["minutes"]
|
||||
date_month_name = place_json["data"]["addresses"][0]["datetime"]["month_full"]
|
||||
date_day_number = place_json["data"]["addresses"][0]["datetime"]["day"]
|
||||
am_pm = place_json["data"]["addresses"][0]["datetime"]["hour_am_pm"]
|
||||
day_name = place_json["data"]["addresses"][0]["datetime"]["day_full"]
|
||||
part_of_day = place_json["data"]["addresses"][0]["datetime"]["timeday_spe"]
|
||||
timezone = place_json["data"]["addresses"][0]["datetime"]["offset_tzid"]
|
||||
timezone_short = place_json["data"]["addresses"][0]["datetime"]["offset_tzab"]
|
||||
gmt = place_json["data"]["addresses"][0]["datetime"]["offset_gmt"]
|
||||
|
||||
part_of_day = part_of_day.replace("_", " ")
|
||||
part_of_day = part_of_day.capitalize()
|
||||
timezone = timezone.replace("_", " ").replace("/", " - ")
|
||||
|
||||
embed = discord.Embed()
|
||||
embed = discord.Embed(
|
||||
colour=await ctx.embed_colour(), title=f"{city_state_country} - {part_of_day}"
|
||||
)
|
||||
embed.description = f"{day_name}, {date_month_name} {date_day_number}, {twelve_hour_first}:{twelve_hour_second} {am_pm}\n{timezone} ({timezone_short}) {gmt} UTC"
|
||||
await ctx.send(embed=embed)
|
||||
|
||||
if place_json["data"]["addresses_found"] == "0":
|
||||
ctx.send("No result")
|
||||
|
||||
@time.command()
|
||||
@checks.admin_or_permissions(manage_server=True)
|
||||
async def set(self, ctx, user: discord.Member, *, tz):
|
||||
"""Allows the mods to edit timezones."""
|
||||
author = ctx.message.author
|
||||
if not user:
|
||||
user = author
|
||||
|
||||
if tz is None:
|
||||
await ctx.send("That timezone is invalid.")
|
||||
return
|
||||
else:
|
||||
space = " "
|
||||
timezone = tz.split(space, 1)[0]
|
||||
if timezone in all_timezones:
|
||||
if "'" in tz:
|
||||
timezone = timezone.replace("'", "")
|
||||
await self.config.member(user).usertime.set(timezone)
|
||||
await ctx.send(f"Successfully set {user.name}'s timezone.")
|
||||
else:
|
||||
await ctx.send(
|
||||
"**Error:** Unrecognized timezone. Try `[p]time set @user Continent/City`: see <https://en.wikipedia.org/wiki/List_of_tz_database_time_zones>"
|
||||
)
|
||||
|
||||
@time.command()
|
||||
async def user(self, ctx, user: discord.Member = None):
|
||||
"""Shows the current time for user."""
|
||||
if not user:
|
||||
await ctx.send("That isn't a user!")
|
||||
else:
|
||||
usertime = await self.config.member(user).usertime()
|
||||
if usertime:
|
||||
time = datetime.now(pytz.timezone(usertime))
|
||||
fmt = "**%H:%M** %d-%B-%Y **%Z (UTC %z)**"
|
||||
time = time.strftime(fmt)
|
||||
await ctx.send(f"{user.name}'s current time is: {str(time)}")
|
||||
else:
|
||||
await ctx.send("That user hasn't set their timezone.")
|
||||
Reference in New Issue
Block a user