Initial build
This commit is contained in:
commit
21c7d1cefe
4 changed files with 220 additions and 0 deletions
141
.gitignore
vendored
Normal file
141
.gitignore
vendored
Normal file
|
@ -0,0 +1,141 @@
|
|||
# Byte-compiled / optimized / DLL files
|
||||
__pycache__/
|
||||
*.py[cod]
|
||||
*$py.class
|
||||
|
||||
# C extensions
|
||||
*.so
|
||||
|
||||
# Distribution / packaging
|
||||
.Python
|
||||
build/
|
||||
develop-eggs/
|
||||
dist/
|
||||
downloads/
|
||||
eggs/
|
||||
.eggs/
|
||||
lib/
|
||||
lib64/
|
||||
parts/
|
||||
sdist/
|
||||
var/
|
||||
wheels/
|
||||
share/python-wheels/
|
||||
*.egg-info/
|
||||
.installed.cfg
|
||||
*.egg
|
||||
MANIFEST
|
||||
|
||||
# PyInstaller
|
||||
# Usually these files are written by a python script from a template
|
||||
# before PyInstaller builds the exe, so as to inject date/other infos into it.
|
||||
*.manifest
|
||||
*.spec
|
||||
|
||||
# Installer logs
|
||||
pip-log.txt
|
||||
pip-delete-this-directory.txt
|
||||
|
||||
# Unit test / coverage reports
|
||||
htmlcov/
|
||||
.tox/
|
||||
.nox/
|
||||
.coverage
|
||||
.coverage.*
|
||||
.cache
|
||||
nosetests.xml
|
||||
coverage.xml
|
||||
*.cover
|
||||
*.py,cover
|
||||
.hypothesis/
|
||||
.pytest_cache/
|
||||
cover/
|
||||
|
||||
# Translations
|
||||
*.mo
|
||||
*.pot
|
||||
|
||||
# Django stuff:
|
||||
*.log
|
||||
local_settings.py
|
||||
db.sqlite3
|
||||
db.sqlite3-journal
|
||||
|
||||
# Flask stuff:
|
||||
instance/
|
||||
.webassets-cache
|
||||
|
||||
# Scrapy stuff:
|
||||
.scrapy
|
||||
|
||||
# Sphinx documentation
|
||||
docs/_build/
|
||||
|
||||
# PyBuilder
|
||||
.pybuilder/
|
||||
target/
|
||||
|
||||
# Jupyter Notebook
|
||||
.ipynb_checkpoints
|
||||
|
||||
# IPython
|
||||
profile_default/
|
||||
ipython_config.py
|
||||
|
||||
# pyenv
|
||||
# For a library or package, you might want to ignore these files since the code is
|
||||
# intended to run in multiple environments; otherwise, check them in:
|
||||
# .python-version
|
||||
|
||||
# pipenv
|
||||
# According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
|
||||
# However, in case of collaboration, if having platform-specific dependencies or dependencies
|
||||
# having no cross-platform support, pipenv may install dependencies that don't work, or not
|
||||
# install all needed dependencies.
|
||||
#Pipfile.lock
|
||||
|
||||
# PEP 582; used by e.g. github.com/David-OConnor/pyflow
|
||||
__pypackages__/
|
||||
|
||||
# Celery stuff
|
||||
celerybeat-schedule
|
||||
celerybeat.pid
|
||||
|
||||
# SageMath parsed files
|
||||
*.sage.py
|
||||
|
||||
# Environments
|
||||
.env
|
||||
.venv
|
||||
env/
|
||||
venv/
|
||||
ENV/
|
||||
env.bak/
|
||||
venv.bak/
|
||||
|
||||
# Spyder project settings
|
||||
.spyderproject
|
||||
.spyproject
|
||||
|
||||
# Rope project settings
|
||||
.ropeproject
|
||||
|
||||
# mkdocs documentation
|
||||
/site
|
||||
|
||||
# mypy
|
||||
.mypy_cache/
|
||||
.dmypy.json
|
||||
dmypy.json
|
||||
|
||||
# Pyre type checker
|
||||
.pyre/
|
||||
|
||||
# pytype static type analyzer
|
||||
.pytype/
|
||||
|
||||
# Cython debug symbols
|
||||
cython_debug/
|
||||
|
||||
# Config file
|
||||
config.yaml
|
3
config.example.yaml
Normal file
3
config.example.yaml
Normal file
|
@ -0,0 +1,3 @@
|
|||
---
|
||||
token: api key here
|
||||
client_id: 123456789012345678
|
BIN
jarvis.png
Normal file
BIN
jarvis.png
Normal file
Binary file not shown.
After Width: | Height: | Size: 367 KiB |
76
src/jarvis.py
Normal file
76
src/jarvis.py
Normal file
|
@ -0,0 +1,76 @@
|
|||
from discord import Embed, Color, Intents
|
||||
from discord.ext import commands, tasks
|
||||
from dataclasses import dataclass
|
||||
from yaml import load
|
||||
from psutil import Process
|
||||
|
||||
try:
|
||||
from yaml import CLoader as Loader
|
||||
except ImportError:
|
||||
from yaml import Loader
|
||||
|
||||
intents = Intents.default()
|
||||
intents.members = True
|
||||
|
||||
jarvis = commands.Bot(command_prefix=">", intents=intents)
|
||||
jarvis_self = Process()
|
||||
|
||||
|
||||
@dataclass
|
||||
class Config:
|
||||
token: str
|
||||
client_id: str
|
||||
|
||||
@classmethod
|
||||
def from_yaml(cls, y):
|
||||
instance = cls(token=y["token"], client_id=y["client_id"])
|
||||
return instance
|
||||
|
||||
|
||||
def get_config(path: str = "config.yaml") -> Config:
|
||||
with open(path) as f:
|
||||
raw = f.read()
|
||||
y = load(raw, Loader=Loader)
|
||||
return Config.from_yaml(y)
|
||||
|
||||
|
||||
def convert_bytesize(bytes: int) -> str:
|
||||
bytes = float(bytes)
|
||||
sizes = ["B", "KB", "MB", "GB"]
|
||||
size = 0
|
||||
while bytes >= 1024 and size < len(sizes) - 1:
|
||||
bytes = bytes / 1024
|
||||
size += 1
|
||||
return f"{bytes:0.2f}{sizes[size]}"
|
||||
|
||||
|
||||
@jarvis.event
|
||||
async def on_ready():
|
||||
print("Logged in as {0.user}".format(jarvis))
|
||||
print("Connected to {} guild(s)".format(len(jarvis.guilds)))
|
||||
|
||||
|
||||
@jarvis.command(name="status", help="Retrieve J.A.R.V.I.S. status")
|
||||
async def _status(ctx):
|
||||
title = "J.A.R.V.I.S. Status"
|
||||
description = "All systems online"
|
||||
color = Color.from_rgb(152, 204, 218)
|
||||
embed = Embed(title=title, description=description, color=color)
|
||||
with jarvis_self.oneshot():
|
||||
embed.add_field(name="CPU Usage", value=jarvis_self.cpu_percent())
|
||||
embed.add_field(
|
||||
name="RAM Usage",
|
||||
value=convert_bytesize(jarvis_self.memory_info().rss),
|
||||
)
|
||||
embed.add_field(name="PID", value=jarvis_self.pid)
|
||||
await ctx.send(embed=embed)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
config = get_config()
|
||||
print(
|
||||
"https://discord.com/api/oauth2/authorize?client_id={}&permissions=8&scope=bot".format(
|
||||
config.client_id
|
||||
)
|
||||
)
|
||||
jarvis.run(config.token)
|
Loading…
Add table
Reference in a new issue