gnouseu/tuxbot-bot
4
0
Fork 0
Code Issues Pull requests Wiki Activity

fix(linting): set the not to 10/10 on pylint

Browse source
This commit is contained in:
Romain J 2020-10-19 00:20:58 +02:00
parent e0788137ff
commit cebb1b0123
24 changed files with 663 additions and 538 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
.gitignore vendored
View file

@ -2,7 +2,30 @@ __pycache__/
*.py[cod] *.py[cod]
*$py.class *$py.class
.idea/workspace.xml # Covers JetBrains IDEs: IntelliJ, RubyMine, PhpStorm, AppCode, PyCharm, CLion, Android Studio, WebStorm and Rider
# Reference: https://intellij-support.jetbrains.com/hc/en-us/articles/206544839
# User-specific stuff
.idea/**/workspace.xml
.idea/**/tasks.xml
.idea/**/usage.statistics.xml
.idea/**/dictionaries
.idea/**/shelf
# Generated files
.idea/**/contentModel.xml
# Sensitive or high-churn files
.idea/**/dataSources/
.idea/**/dataSources.ids
.idea/**/dataSources.local.xml
.idea/**/sqlDataSources.xml
.idea/**/dynamic.xml
.idea/**/uiDesigner.xml
.idea/**/dbnavigator.xml
.idea/sonarlint/
*.log *.log

5
.idea/dictionaries/romain.xml
View file

@ -1,10 +1,12 @@
<component name="ProjectDictionaryState"> <component name="ProjectDictionaryState">
<dictionary name="romain"> <dictionary name="romain">
<words> <words>
<w>anglais</w>
<w>appdirs</w> <w>appdirs</w>
<w>asctime</w> <w>asctime</w>
<w>commandstats</w> <w>commandstats</w>
<w>francais</w> <w>francais</w>
<w>français</w>
<w>gnous</w> <w>gnous</w>
<w>ipinfo</w> <w>ipinfo</w>
<w>iplocalise</w> <w>iplocalise</w>
@ -13,10 +15,13 @@
<w>localiseip</w> <w>localiseip</w>
<w>postgresql</w> <w>postgresql</w>
<w>pred</w> <w>pred</w>
<w>pylint</w>
<w>releaselevel</w> <w>releaselevel</w>
<w>rprint</w>
<w>socketstats</w> <w>socketstats</w>
<w>splt</w> <w>splt</w>
<w>systemd</w> <w>systemd</w>
<w>tldr</w>
<w>tutux</w> <w>tutux</w>
<w>tuxbot</w> <w>tuxbot</w>
<w>tuxbot's</w> <w>tuxbot's</w>

11
.idea/statistic.xml Normal file
View file

@ -0,0 +1,11 @@
<?xml version="1.0" encoding="UTF-8"?>
<project version="4">
<component name="Statistic">
<option name="excludedDirectories">
<list>
<option value="$PROJECT_DIR$/Tuxbot_bot.egg-info" />
<option value="$PROJECT_DIR$/venv" />
</list>
</option>
</component>
</project>

13
.pylintrc Normal file
View file

@ -0,0 +1,13 @@
[BASIC]
good-names=
e, # (exception) as e
f, # (file) as f
[MASTER]
disable=
C0114, # missing-module-docstring
C0115, # missing-class-docstring
C0116, # missing-function-docstring
W0703, # broad-except
R0902, # too-many-instance-attributes
R0903, # too-few-public-methods

238
LICENSE.md → LICENSE
View file

@ -1,11 +1,11 @@
GNU Affero General Public License GNU AFFERO GENERAL PUBLIC LICENSE
================================= Version 3, 19 November 2007
_Copyright © 2017-2020 GnousEU Developper Team, &lt;<http://gnous.eu/>&gt;_
Copyright (C) 2007 Free Software Foundation, Inc. <https://fsf.org/>
Everyone is permitted to copy and distribute verbatim copies Everyone is permitted to copy and distribute verbatim copies
of this license document, but changing it is not allowed. of this license document, but changing it is not allowed.
## Preamble Preamble
The GNU Affero General Public License is a free, copyleft license for The GNU Affero General Public License is a free, copyleft license for
software and other kinds of works, specifically designed to ensure software and other kinds of works, specifically designed to ensure
@ -25,7 +25,7 @@ want it, that you can change the software or use pieces of it in new
free programs, and that you know you can do these things. free programs, and that you know you can do these things.
Developers that use our General Public Licenses protect your rights Developers that use our General Public Licenses protect your rights
with two steps: **(1)** assert copyright on the software, and **(2)** offer with two steps: (1) assert copyright on the software, and (2) offer
you this License which gives you legal permission to copy, distribute you this License which gives you legal permission to copy, distribute
and/or modify the software. and/or modify the software.
@ -56,70 +56,70 @@ this license.
The precise terms and conditions for copying, distribution and The precise terms and conditions for copying, distribution and
modification follow. modification follow.
## TERMS AND CONDITIONS TERMS AND CONDITIONS
### 0. Definitions 0. Definitions.
“This License” refers to version 3 of the GNU Affero General Public License. "This License" refers to version 3 of the GNU Affero General Public License.
“Copyright” also means copyright-like laws that apply to other kinds of "Copyright" also means copyright-like laws that apply to other kinds of
works, such as semiconductor masks. works, such as semiconductor masks.
“The Program” refers to any copyrightable work licensed under this "The Program" refers to any copyrightable work licensed under this
License. Each licensee is addressed as “you”. “Licensees” and License. Each licensee is addressed as "you". "Licensees" and
“recipients” may be individuals or organizations. "recipients" may be individuals or organizations.
To “modify” a work means to copy from or adapt all or part of the work To "modify" a work means to copy from or adapt all or part of the work
in a fashion requiring copyright permission, other than the making of an in a fashion requiring copyright permission, other than the making of an
exact copy. The resulting work is called a “modified version” of the exact copy. The resulting work is called a "modified version" of the
earlier work or a work “based on” the earlier work. earlier work or a work "based on" the earlier work.
A “covered work” means either the unmodified Program or a work based A "covered work" means either the unmodified Program or a work based
on the Program. on the Program.
To “propagate” a work means to do anything with it that, without To "propagate" a work means to do anything with it that, without
permission, would make you directly or secondarily liable for permission, would make you directly or secondarily liable for
infringement under applicable copyright law, except executing it on a infringement under applicable copyright law, except executing it on a
computer or modifying a private copy. Propagation includes copying, computer or modifying a private copy. Propagation includes copying,
distribution (with or without modification), making available to the distribution (with or without modification), making available to the
public, and in some countries other activities as well. public, and in some countries other activities as well.
To “convey” a work means any kind of propagation that enables other To "convey" a work means any kind of propagation that enables other
parties to make or receive copies. Mere interaction with a user through parties to make or receive copies. Mere interaction with a user through
a computer network, with no transfer of a copy, is not conveying. a computer network, with no transfer of a copy, is not conveying.
An interactive user interface displays “Appropriate Legal Notices” An interactive user interface displays "Appropriate Legal Notices"
to the extent that it includes a convenient and prominently visible to the extent that it includes a convenient and prominently visible
feature that **(1)** displays an appropriate copyright notice, and **(2)** feature that (1) displays an appropriate copyright notice, and (2)
tells the user that there is no warranty for the work (except to the tells the user that there is no warranty for the work (except to the
extent that warranties are provided), that licensees may convey the extent that warranties are provided), that licensees may convey the
work under this License, and how to view a copy of this License. If work under this License, and how to view a copy of this License. If
the interface presents a list of user commands or options, such as a the interface presents a list of user commands or options, such as a
menu, a prominent item in the list meets this criterion. menu, a prominent item in the list meets this criterion.
### 1. Source Code 1. Source Code.
The “source code” for a work means the preferred form of the work The "source code" for a work means the preferred form of the work
for making modifications to it. “Object code” means any non-source for making modifications to it. "Object code" means any non-source
form of a work. form of a work.
A “Standard Interface” means an interface that either is an official A "Standard Interface" means an interface that either is an official
standard defined by a recognized standards body, or, in the case of standard defined by a recognized standards body, or, in the case of
interfaces specified for a particular programming language, one that interfaces specified for a particular programming language, one that
is widely used among developers working in that language. is widely used among developers working in that language.
The “System Libraries” of an executable work include anything, other The "System Libraries" of an executable work include anything, other
than the work as a whole, that **(a)** is included in the normal form of than the work as a whole, that (a) is included in the normal form of
packaging a Major Component, but which is not part of that Major packaging a Major Component, but which is not part of that Major
Component, and **(b)** serves only to enable use of the work with that Component, and (b) serves only to enable use of the work with that
Major Component, or to implement a Standard Interface for which an Major Component, or to implement a Standard Interface for which an
implementation is available to the public in source code form. A implementation is available to the public in source code form. A
“Major Component”, in this context, means a major essential component "Major Component", in this context, means a major essential component
(kernel, window system, and so on) of the specific operating system (kernel, window system, and so on) of the specific operating system
(if any) on which the executable work runs, or a compiler used to (if any) on which the executable work runs, or a compiler used to
produce the work, or an object code interpreter used to run it. produce the work, or an object code interpreter used to run it.
The “Corresponding Source” for a work in object code form means all The "Corresponding Source" for a work in object code form means all
the source code needed to generate, install, and (for an executable the source code needed to generate, install, and (for an executable
work) run the object code and to modify the work, including scripts to work) run the object code and to modify the work, including scripts to
control those activities. However, it does not include the work's control those activities. However, it does not include the work's
@ -139,7 +139,7 @@ Source.
The Corresponding Source for a work in source code form is that The Corresponding Source for a work in source code form is that
same work. same work.
### 2. Basic Permissions 2. Basic Permissions.
All rights granted under this License are granted for the term of All rights granted under this License are granted for the term of
copyright on the Program, and are irrevocable provided the stated copyright on the Program, and are irrevocable provided the stated
@ -164,7 +164,7 @@ Conveying under any other circumstances is permitted solely under
the conditions stated below. Sublicensing is not allowed; section 10 the conditions stated below. Sublicensing is not allowed; section 10
makes it unnecessary. makes it unnecessary.
### 3. Protecting Users' Legal Rights From Anti-Circumvention Law 3. Protecting Users' Legal Rights From Anti-Circumvention Law.
No covered work shall be deemed part of an effective technological No covered work shall be deemed part of an effective technological
measure under any applicable law fulfilling obligations under article measure under any applicable law fulfilling obligations under article
@ -180,7 +180,7 @@ modification of the work as a means of enforcing, against the work's
users, your or third parties' legal rights to forbid circumvention of users, your or third parties' legal rights to forbid circumvention of
technological measures. technological measures.
### 4. Conveying Verbatim Copies 4. Conveying Verbatim Copies.
You may convey verbatim copies of the Program's source code as you You may convey verbatim copies of the Program's source code as you
receive it, in any medium, provided that you conspicuously and receive it, in any medium, provided that you conspicuously and
@ -193,26 +193,29 @@ recipients a copy of this License along with the Program.
You may charge any price or no price for each copy that you convey, You may charge any price or no price for each copy that you convey,
and you may offer support or warranty protection for a fee. and you may offer support or warranty protection for a fee.
### 5. Conveying Modified Source Versions 5. Conveying Modified Source Versions.
You may convey a work based on the Program, or the modifications to You may convey a work based on the Program, or the modifications to
produce it from the Program, in the form of source code under the produce it from the Program, in the form of source code under the
terms of section 4, provided that you also meet all of these conditions: terms of section 4, provided that you also meet all of these conditions:
* **a)** The work must carry prominent notices stating that you modified a) The work must carry prominent notices stating that you modified
it, and giving a relevant date. it, and giving a relevant date.
* **b)** The work must carry prominent notices stating that it is
released under this License and any conditions added under section 7. b) The work must carry prominent notices stating that it is
This requirement modifies the requirement in section 4 to released under this License and any conditions added under section
“keep intact all notices”. 7. This requirement modifies the requirement in section 4 to
* **c)** You must license the entire work, as a whole, under this "keep intact all notices".
c) You must license the entire work, as a whole, under this
License to anyone who comes into possession of a copy. This License to anyone who comes into possession of a copy. This
License will therefore apply, along with any applicable section 7 License will therefore apply, along with any applicable section 7
additional terms, to the whole of the work, and all its parts, additional terms, to the whole of the work, and all its parts,
regardless of how they are packaged. This License gives no regardless of how they are packaged. This License gives no
permission to license the work in any other way, but it does not permission to license the work in any other way, but it does not
invalidate such permission if you have separately received it. invalidate such permission if you have separately received it.
* **d)** If the work has interactive user interfaces, each must display
d) If the work has interactive user interfaces, each must display
Appropriate Legal Notices; however, if the Program has interactive Appropriate Legal Notices; however, if the Program has interactive
interfaces that do not display Appropriate Legal Notices, your interfaces that do not display Appropriate Legal Notices, your
work need not make them do so. work need not make them do so.
@ -221,40 +224,43 @@ A compilation of a covered work with other separate and independent
works, which are not by their nature extensions of the covered work, works, which are not by their nature extensions of the covered work,
and which are not combined with it such as to form a larger program, and which are not combined with it such as to form a larger program,
in or on a volume of a storage or distribution medium, is called an in or on a volume of a storage or distribution medium, is called an
“aggregate” if the compilation and its resulting copyright are not "aggregate" if the compilation and its resulting copyright are not
used to limit the access or legal rights of the compilation's users used to limit the access or legal rights of the compilation's users
beyond what the individual works permit. Inclusion of a covered work beyond what the individual works permit. Inclusion of a covered work
in an aggregate does not cause this License to apply to the other in an aggregate does not cause this License to apply to the other
parts of the aggregate. parts of the aggregate.
### 6. Conveying Non-Source Forms 6. Conveying Non-Source Forms.
You may convey a covered work in object code form under the terms You may convey a covered work in object code form under the terms
of sections 4 and 5, provided that you also convey the of sections 4 and 5, provided that you also convey the
machine-readable Corresponding Source under the terms of this License, machine-readable Corresponding Source under the terms of this License,
in one of these ways: in one of these ways:
* **a)** Convey the object code in, or embodied in, a physical product a) Convey the object code in, or embodied in, a physical product
(including a physical distribution medium), accompanied by the (including a physical distribution medium), accompanied by the
Corresponding Source fixed on a durable physical medium Corresponding Source fixed on a durable physical medium
customarily used for software interchange. customarily used for software interchange.
* **b)** Convey the object code in, or embodied in, a physical product
b) Convey the object code in, or embodied in, a physical product
(including a physical distribution medium), accompanied by a (including a physical distribution medium), accompanied by a
written offer, valid for at least three years and valid for as written offer, valid for at least three years and valid for as
long as you offer spare parts or customer support for that product long as you offer spare parts or customer support for that product
model, to give anyone who possesses the object code either **(1)** a model, to give anyone who possesses the object code either (1) a
copy of the Corresponding Source for all the software in the copy of the Corresponding Source for all the software in the
product that is covered by this License, on a durable physical product that is covered by this License, on a durable physical
medium customarily used for software interchange, for a price no medium customarily used for software interchange, for a price no
more than your reasonable cost of physically performing this more than your reasonable cost of physically performing this
conveying of source, or **(2)** access to copy the conveying of source, or (2) access to copy the
Corresponding Source from a network server at no charge. Corresponding Source from a network server at no charge.
* **c)** Convey individual copies of the object code with a copy of the
c) Convey individual copies of the object code with a copy of the
written offer to provide the Corresponding Source. This written offer to provide the Corresponding Source. This
alternative is allowed only occasionally and noncommercially, and alternative is allowed only occasionally and noncommercially, and
only if you received the object code with such an offer, in accord only if you received the object code with such an offer, in accord
with subsection 6b. with subsection 6b.
* **d)** Convey the object code by offering access from a designated
d) Convey the object code by offering access from a designated
place (gratis or for a charge), and offer equivalent access to the place (gratis or for a charge), and offer equivalent access to the
Corresponding Source in the same way through the same place at no Corresponding Source in the same way through the same place at no
further charge. You need not require recipients to copy the further charge. You need not require recipients to copy the
@ -266,7 +272,8 @@ clear directions next to the object code saying where to find the
Corresponding Source. Regardless of what server hosts the Corresponding Source. Regardless of what server hosts the
Corresponding Source, you remain obligated to ensure that it is Corresponding Source, you remain obligated to ensure that it is
available for as long as needed to satisfy these requirements. available for as long as needed to satisfy these requirements.
* **e)** Convey the object code using peer-to-peer transmission, provided
e) Convey the object code using peer-to-peer transmission, provided
you inform other peers where the object code and Corresponding you inform other peers where the object code and Corresponding
Source of the work are being offered to the general public at no Source of the work are being offered to the general public at no
charge under subsection 6d. charge under subsection 6d.
@ -275,12 +282,12 @@ A separable portion of the object code, whose source code is excluded
from the Corresponding Source as a System Library, need not be from the Corresponding Source as a System Library, need not be
included in conveying the object code work. included in conveying the object code work.
A “User Product” is either **(1)** a “consumer product”, which means any A "User Product" is either (1) a "consumer product", which means any
tangible personal property which is normally used for personal, family, tangible personal property which is normally used for personal, family,
or household purposes, or **(2)** anything designed or sold for incorporation or household purposes, or (2) anything designed or sold for incorporation
into a dwelling. In determining whether a product is a consumer product, into a dwelling. In determining whether a product is a consumer product,
doubtful cases shall be resolved in favor of coverage. For a particular doubtful cases shall be resolved in favor of coverage. For a particular
product received by a particular user, “normally used” refers to a product received by a particular user, "normally used" refers to a
typical or common use of that class of product, regardless of the status typical or common use of that class of product, regardless of the status
of the particular user or of the way in which the particular user of the particular user or of the way in which the particular user
actually uses, or expects or is expected to use, the product. A product actually uses, or expects or is expected to use, the product. A product
@ -288,7 +295,7 @@ is a consumer product regardless of whether the product has substantial
commercial, industrial or non-consumer uses, unless such uses represent commercial, industrial or non-consumer uses, unless such uses represent
the only significant mode of use of the product. the only significant mode of use of the product.
“Installation Information” for a User Product means any methods, "Installation Information" for a User Product means any methods,
procedures, authorization keys, or other information required to install procedures, authorization keys, or other information required to install
and execute modified versions of a covered work in that User Product from and execute modified versions of a covered work in that User Product from
a modified version of its Corresponding Source. The information must a modified version of its Corresponding Source. The information must
@ -321,9 +328,9 @@ documented (and with an implementation available to the public in
source code form), and must require no special password or key for source code form), and must require no special password or key for
unpacking, reading or copying. unpacking, reading or copying.
### 7. Additional Terms 7. Additional Terms.
“Additional permissions” are terms that supplement the terms of this "Additional permissions" are terms that supplement the terms of this
License by making exceptions from one or more of its conditions. License by making exceptions from one or more of its conditions.
Additional permissions that are applicable to the entire Program shall Additional permissions that are applicable to the entire Program shall
be treated as though they were included in this License, to the extent be treated as though they were included in this License, to the extent
@ -343,26 +350,31 @@ Notwithstanding any other provision of this License, for material you
add to a covered work, you may (if authorized by the copyright holders of add to a covered work, you may (if authorized by the copyright holders of
that material) supplement the terms of this License with terms: that material) supplement the terms of this License with terms:
* **a)** Disclaiming warranty or limiting liability differently from the a) Disclaiming warranty or limiting liability differently from the
terms of sections 15 and 16 of this License; or terms of sections 15 and 16 of this License; or
* **b)** Requiring preservation of specified reasonable legal notices or
b) Requiring preservation of specified reasonable legal notices or
author attributions in that material or in the Appropriate Legal author attributions in that material or in the Appropriate Legal
Notices displayed by works containing it; or Notices displayed by works containing it; or
* **c)** Prohibiting misrepresentation of the origin of that material, or
c) Prohibiting misrepresentation of the origin of that material, or
requiring that modified versions of such material be marked in requiring that modified versions of such material be marked in
reasonable ways as different from the original version; or reasonable ways as different from the original version; or
* **d)** Limiting the use for publicity purposes of names of licensors or
d) Limiting the use for publicity purposes of names of licensors or
authors of the material; or authors of the material; or
* **e)** Declining to grant rights under trademark law for use of some
e) Declining to grant rights under trademark law for use of some
trade names, trademarks, or service marks; or trade names, trademarks, or service marks; or
* **f)** Requiring indemnification of licensors and authors of that
f) Requiring indemnification of licensors and authors of that
material by anyone who conveys the material (or modified versions of material by anyone who conveys the material (or modified versions of
it) with contractual assumptions of liability to the recipient, for it) with contractual assumptions of liability to the recipient, for
any liability that these contractual assumptions directly impose on any liability that these contractual assumptions directly impose on
those licensors and authors. those licensors and authors.
All other non-permissive additional terms are considered “further All other non-permissive additional terms are considered "further
restrictions” within the meaning of section 10. If the Program as you restrictions" within the meaning of section 10. If the Program as you
received it, or any part of it, contains a notice stating that it is received it, or any part of it, contains a notice stating that it is
governed by this License along with a term that is a further governed by this License along with a term that is a further
restriction, you may remove that term. If a license document contains restriction, you may remove that term. If a license document contains
@ -380,7 +392,7 @@ Additional terms, permissive or non-permissive, may be stated in the
form of a separately written license, or stated as exceptions; form of a separately written license, or stated as exceptions;
the above requirements apply either way. the above requirements apply either way.
### 8. Termination 8. Termination.
You may not propagate or modify a covered work except as expressly You may not propagate or modify a covered work except as expressly
provided under this License. Any attempt otherwise to propagate or provided under this License. Any attempt otherwise to propagate or
@ -389,9 +401,9 @@ this License (including any patent licenses granted under the third
paragraph of section 11). paragraph of section 11).
However, if you cease all violation of this License, then your However, if you cease all violation of this License, then your
license from a particular copyright holder is reinstated **(a)** license from a particular copyright holder is reinstated (a)
provisionally, unless and until the copyright holder explicitly and provisionally, unless and until the copyright holder explicitly and
finally terminates your license, and **(b)** permanently, if the copyright finally terminates your license, and (b) permanently, if the copyright
holder fails to notify you of the violation by some reasonable means holder fails to notify you of the violation by some reasonable means
prior to 60 days after the cessation. prior to 60 days after the cessation.
@ -408,7 +420,7 @@ this License. If your rights have been terminated and not permanently
reinstated, you do not qualify to receive new licenses for the same reinstated, you do not qualify to receive new licenses for the same
material under section 10. material under section 10.
### 9. Acceptance Not Required for Having Copies 9. Acceptance Not Required for Having Copies.
You are not required to accept this License in order to receive or You are not required to accept this License in order to receive or
run a copy of the Program. Ancillary propagation of a covered work run a copy of the Program. Ancillary propagation of a covered work
@ -419,14 +431,14 @@ modify any covered work. These actions infringe copyright if you do
not accept this License. Therefore, by modifying or propagating a not accept this License. Therefore, by modifying or propagating a
covered work, you indicate your acceptance of this License to do so. covered work, you indicate your acceptance of this License to do so.
### 10. Automatic Licensing of Downstream Recipients 10. Automatic Licensing of Downstream Recipients.
Each time you convey a covered work, the recipient automatically Each time you convey a covered work, the recipient automatically
receives a license from the original licensors, to run, modify and receives a license from the original licensors, to run, modify and
propagate that work, subject to this License. You are not responsible propagate that work, subject to this License. You are not responsible
for enforcing compliance by third parties with this License. for enforcing compliance by third parties with this License.
An “entity transaction” is a transaction transferring control of an An "entity transaction" is a transaction transferring control of an
organization, or substantially all assets of one, or subdividing an organization, or substantially all assets of one, or subdividing an
organization, or merging organizations. If propagation of a covered organization, or merging organizations. If propagation of a covered
work results from an entity transaction, each party to that work results from an entity transaction, each party to that
@ -444,19 +456,19 @@ rights granted under this License, and you may not initiate litigation
any patent claim is infringed by making, using, selling, offering for any patent claim is infringed by making, using, selling, offering for
sale, or importing the Program or any portion of it. sale, or importing the Program or any portion of it.
### 11. Patents 11. Patents.
A “contributor” is a copyright holder who authorizes use under this A "contributor" is a copyright holder who authorizes use under this
License of the Program or a work on which the Program is based. The License of the Program or a work on which the Program is based. The
work thus licensed is called the contributor's “contributor version”. work thus licensed is called the contributor's "contributor version".
A contributor's “essential patent claims” are all patent claims A contributor's "essential patent claims" are all patent claims
owned or controlled by the contributor, whether already acquired or owned or controlled by the contributor, whether already acquired or
hereafter acquired, that would be infringed by some manner, permitted hereafter acquired, that would be infringed by some manner, permitted
by this License, of making, using, or selling its contributor version, by this License, of making, using, or selling its contributor version,
but do not include claims that would be infringed only as a but do not include claims that would be infringed only as a
consequence of further modification of the contributor version. For consequence of further modification of the contributor version. For
purposes of this definition, “control” includes the right to grant purposes of this definition, "control" includes the right to grant
patent sublicenses in a manner consistent with the requirements of patent sublicenses in a manner consistent with the requirements of
this License. this License.
@ -465,10 +477,10 @@ patent license under the contributor's essential patent claims, to
make, use, sell, offer for sale, import and otherwise run, modify and make, use, sell, offer for sale, import and otherwise run, modify and
propagate the contents of its contributor version. propagate the contents of its contributor version.
In the following three paragraphs, a “patent license” is any express In the following three paragraphs, a "patent license" is any express
agreement or commitment, however denominated, not to enforce a patent agreement or commitment, however denominated, not to enforce a patent
(such as an express permission to practice a patent or covenant not to (such as an express permission to practice a patent or covenant not to
sue for patent infringement). To “grant” such a patent license to a sue for patent infringement). To "grant" such a patent license to a
party means to make such an agreement or commitment not to enforce a party means to make such an agreement or commitment not to enforce a
patent against the party. patent against the party.
@ -476,11 +488,11 @@ If you convey a covered work, knowingly relying on a patent license,
and the Corresponding Source of the work is not available for anyone and the Corresponding Source of the work is not available for anyone
to copy, free of charge and under the terms of this License, through a to copy, free of charge and under the terms of this License, through a
publicly available network server or other readily accessible means, publicly available network server or other readily accessible means,
then you must either **(1)** cause the Corresponding Source to be so then you must either (1) cause the Corresponding Source to be so
available, or **(2)** arrange to deprive yourself of the benefit of the available, or (2) arrange to deprive yourself of the benefit of the
patent license for this particular work, or **(3)** arrange, in a manner patent license for this particular work, or (3) arrange, in a manner
consistent with the requirements of this License, to extend the patent consistent with the requirements of this License, to extend the patent
license to downstream recipients. “Knowingly relying” means you have license to downstream recipients. "Knowingly relying" means you have
actual knowledge that, but for the patent license, your conveying the actual knowledge that, but for the patent license, your conveying the
covered work in a country, or your recipient's use of the covered work covered work in a country, or your recipient's use of the covered work
in a country, would infringe one or more identifiable patents in that in a country, would infringe one or more identifiable patents in that
@ -494,7 +506,7 @@ or convey a specific copy of the covered work, then the patent license
you grant is automatically extended to all recipients of the covered you grant is automatically extended to all recipients of the covered
work and works based on it. work and works based on it.
A patent license is “discriminatory” if it does not include within A patent license is "discriminatory" if it does not include within
the scope of its coverage, prohibits the exercise of, or is the scope of its coverage, prohibits the exercise of, or is
conditioned on the non-exercise of one or more of the rights that are conditioned on the non-exercise of one or more of the rights that are
specifically granted under this License. You may not convey a covered specifically granted under this License. You may not convey a covered
@ -503,8 +515,8 @@ in the business of distributing software, under which you make payment
to the third party based on the extent of your activity of conveying to the third party based on the extent of your activity of conveying
the work, and under which the third party grants, to any of the the work, and under which the third party grants, to any of the
parties who would receive the covered work from you, a discriminatory parties who would receive the covered work from you, a discriminatory
patent license **(a)** in connection with copies of the covered work patent license (a) in connection with copies of the covered work
conveyed by you (or copies made from those copies), or **(b)** primarily conveyed by you (or copies made from those copies), or (b) primarily
for and in connection with specific products or compilations that for and in connection with specific products or compilations that
contain the covered work, unless you entered into that arrangement, contain the covered work, unless you entered into that arrangement,
or that patent license was granted, prior to 28 March 2007. or that patent license was granted, prior to 28 March 2007.
@ -513,7 +525,7 @@ Nothing in this License shall be construed as excluding or limiting
any implied license or other defenses to infringement that may any implied license or other defenses to infringement that may
otherwise be available to you under applicable patent law. otherwise be available to you under applicable patent law.
### 12. No Surrender of Others' Freedom 12. No Surrender of Others' Freedom.
If conditions are imposed on you (whether by court order, agreement or If conditions are imposed on you (whether by court order, agreement or
otherwise) that contradict the conditions of this License, they do not otherwise) that contradict the conditions of this License, they do not
@ -525,7 +537,7 @@ to collect a royalty for further conveying from those to whom you convey
the Program, the only way you could satisfy both those terms and this the Program, the only way you could satisfy both those terms and this
License would be to refrain entirely from conveying the Program. License would be to refrain entirely from conveying the Program.
### 13. Remote Network Interaction; Use with the GNU General Public License 13. Remote Network Interaction; Use with the GNU General Public License.
Notwithstanding any other provision of this License, if you modify the Notwithstanding any other provision of this License, if you modify the
Program, your modified version must prominently offer all users Program, your modified version must prominently offer all users
@ -546,7 +558,7 @@ License will continue to apply to the part which is the covered work,
but the work with which it is combined will remain governed by version but the work with which it is combined will remain governed by version
3 of the GNU General Public License. 3 of the GNU General Public License.
### 14. Revised Versions of this License 14. Revised Versions of this License.
The Free Software Foundation may publish revised and/or new versions of The Free Software Foundation may publish revised and/or new versions of
the GNU Affero General Public License from time to time. Such new versions the GNU Affero General Public License from time to time. Such new versions
@ -555,7 +567,7 @@ address new problems or concerns.
Each version is given a distinguishing version number. If the Each version is given a distinguishing version number. If the
Program specifies that a certain numbered version of the GNU Affero General Program specifies that a certain numbered version of the GNU Affero General
Public License “or any later version” applies to it, you have the Public License "or any later version" applies to it, you have the
option of following the terms and conditions either of that numbered option of following the terms and conditions either of that numbered
version or of any later version published by the Free Software version or of any later version published by the Free Software
Foundation. If the Program does not specify a version number of the Foundation. If the Program does not specify a version number of the
@ -572,18 +584,18 @@ permissions. However, no additional obligations are imposed on any
author or copyright holder as a result of your choosing to follow a author or copyright holder as a result of your choosing to follow a
later version. later version.
### 15. Disclaimer of Warranty 15. Disclaimer of Warranty.
THERE IS NO WARRANTY FOR THE PROGRAM, TO THE EXTENT PERMITTED BY THERE IS NO WARRANTY FOR THE PROGRAM, TO THE EXTENT PERMITTED BY
APPLICABLE LAW. EXCEPT WHEN OTHERWISE STATED IN WRITING THE COPYRIGHT APPLICABLE LAW. EXCEPT WHEN OTHERWISE STATED IN WRITING THE COPYRIGHT
HOLDERS AND/OR OTHER PARTIES PROVIDE THE PROGRAM “AS IS” WITHOUT WARRANTY HOLDERS AND/OR OTHER PARTIES PROVIDE THE PROGRAM "AS IS" WITHOUT WARRANTY
OF ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING, BUT NOT LIMITED TO, OF ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING, BUT NOT LIMITED TO,
THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
PURPOSE. THE ENTIRE RISK AS TO THE QUALITY AND PERFORMANCE OF THE PROGRAM PURPOSE. THE ENTIRE RISK AS TO THE QUALITY AND PERFORMANCE OF THE PROGRAM
IS WITH YOU. SHOULD THE PROGRAM PROVE DEFECTIVE, YOU ASSUME THE COST OF IS WITH YOU. SHOULD THE PROGRAM PROVE DEFECTIVE, YOU ASSUME THE COST OF
ALL NECESSARY SERVICING, REPAIR OR CORRECTION. ALL NECESSARY SERVICING, REPAIR OR CORRECTION.
### 16. Limitation of Liability 16. Limitation of Liability.
IN NO EVENT UNLESS REQUIRED BY APPLICABLE LAW OR AGREED TO IN WRITING IN NO EVENT UNLESS REQUIRED BY APPLICABLE LAW OR AGREED TO IN WRITING
WILL ANY COPYRIGHT HOLDER, OR ANY OTHER PARTY WHO MODIFIES AND/OR CONVEYS WILL ANY COPYRIGHT HOLDER, OR ANY OTHER PARTY WHO MODIFIES AND/OR CONVEYS
@ -595,7 +607,7 @@ PARTIES OR A FAILURE OF THE PROGRAM TO OPERATE WITH ANY OTHER PROGRAMS),
EVEN IF SUCH HOLDER OR OTHER PARTY HAS BEEN ADVISED OF THE POSSIBILITY OF EVEN IF SUCH HOLDER OR OTHER PARTY HAS BEEN ADVISED OF THE POSSIBILITY OF
SUCH DAMAGES. SUCH DAMAGES.
### 17. Interpretation of Sections 15 and 16 17. Interpretation of Sections 15 and 16.
If the disclaimer of warranty and limitation of liability provided If the disclaimer of warranty and limitation of liability provided
above cannot be given local legal effect according to their terms, above cannot be given local legal effect according to their terms,
@ -603,3 +615,47 @@ reviewing courts shall apply local law that most closely approximates
an absolute waiver of all civil liability in connection with the an absolute waiver of all civil liability in connection with the
Program, unless a warranty or assumption of liability accompanies a Program, unless a warranty or assumption of liability accompanies a
copy of the Program in return for a fee. copy of the Program in return for a fee.
END OF TERMS AND CONDITIONS
How to Apply These Terms to Your New Programs
If you develop a new program, and you want it to be of the greatest
possible use to the public, the best way to achieve this is to make it
free software which everyone can redistribute and change under these terms.
To do so, attach the following notices to the program. It is safest
to attach them to the start of each source file to most effectively
state the exclusion of warranty; and each file should have at least
the "copyright" line and a pointer to where the full notice is found.
<one line to give the program's name and a brief idea of what it does.>
Copyright (C) <year> <name of author>
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU Affero General Public License as published
by the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU Affero General Public License for more details.
You should have received a copy of the GNU Affero General Public License
along with this program. If not, see <https://www.gnu.org/licenses/>.
Also add information on how to contact you by electronic and paper mail.
If your software can interact with users remotely through a computer
network, you should also make sure that it provides a way for users to
get its source. For example, if your program is a web application, its
interface could display a "Source" link that leads users to an archive
of the code. There are many ways you could offer source, and different
solutions will be better for different programs; see section 13 for the
specific requirements.
You should also get your employer (if you work as a programmer) or school,
if any, to sign a "copyright disclaimer" for the program, if necessary.
For more information on this, and how to apply and follow the GNU AGPL, see
<https://www.gnu.org/licenses/>.

2
Makefile
View file

@ -14,7 +14,7 @@ update:
# Blackify code # Blackify code
reformat: reformat:
$(PYTHON) -m black `git ls-files "*.py"` $(PYTHON) -m black `git ls-files "*.py"` --line-length=79 && pylint tuxbot
# Translations # Translations
xgettext: xgettext:

76
README.md
View file

@ -1,76 +0,0 @@
![](https://img.shields.io/badge/python-3.7%20%7C%203.8%20%7C%203.9%20%7C%203.10-%23007ec6)
![](https://img.shields.io/badge/dynamic/json?color=%23dfb317&label=issues&query=%24.open_issues_count&suffix=%20open&url=https%3A%2F%2Fgit.gnous.eu%2Fapi%2Fv1%2Frepos%2FGnousEU%2Ftuxbot-bot%2F)
# Installing Tuxbot
It is preferable to install the bot on a dedicated user. If you don't know how to do it, please refer to [this guide](https://www.digitalocean.com/community/tutorials/how-to-create-a-sudo-user-on-ubuntu-quickstart)
## Installing the pre-requirements
- The pre-requirements are:
- Python 3.7 or greater
- Pip
- Git
### Operating systems
-----
### Arch Linux
```shell script
sudo pacman -Syu python python-pip python-virtualenv git
```
Continue to [create the venv](#creating-the-virtual-environnement).
-----
#### Debian
```shell script
sudo apt update
sudo apt -y install python3 python3-dev python3-pip python3-venv git
```
Continue to [create the venv](#creating-the-virtual-environnement).
-----
#### Windows
*not for now and not for the future*
## Creating the Virtual Environment
To set up the virtual environment and install the bot, simply run this two commands:
```shell script
make
make install
```
Now, switch your environment to the virtual one by run this single command:
```shell script
source ~/tuxvenv/bin/activate
```
## Configuration
It's time to set up your first instance, to do this, you can simply execute this command:
```shell script
tuxbot-setup [your instance name]
```
After following the instructions, you can run your instance by executing this command:
```shell script
tuxbot [your instance name]
```
## Update
*todo*

89
README.rst Normal file
View file

@ -0,0 +1,89 @@
|image0| |image1|
.. role:: bash(code)
:language: bash
Installing Tuxbot
=================
It is preferable to install the bot on a dedicated user. If you don't
know how to do it, please refer to `this guide <https://www.digitalocean.com/community/tutorials/how-to-create-a-sudo-user-on-ubuntu-quickstart>`__
Installing the pre-requirements
-------------------------------
- The pre-requirements are:
- Python 3.7 or greater
- Pip
- Git
Operating systems
~~~~~~~~~~~~~~~~~
Arch Linux
^^^^^^^^^^
.. code-block:: bash
$ sudo pacman -Syu python python-pip python-virtualenv git
Continue to `create the venv <#creating-the-virtual-environnement>`__.
--------------
Debian
^^^^^^
.. code-block:: bash
$ sudo apt update
$ sudo apt -y install python3 python3-dev python3-pip python3-venv git
Continue to `create the venv <#creating-the-virtual-environnement>`__.
--------------
Windows
^^^^^^^
*not for now and not for the future*
Creating the Virtual Environment
--------------------------------
To set up the virtual environment and install the bot, simply run this
two commands:
.. code-block:: bash
$ make
$ make install
Now, switch your environment to the virtual one by run this single
command: :bash:`source ~/tuxvenv/bin/activate`
Configuration
-------------
It's time to set up your first instance, to do this, you can simply
execute this command:
:bash:`tuxbot-setup [your instance name]`
After following the instructions, you can run your instance by executing
this command:
:bash:`tuxbot [your instance name]`
Update
------
To update the whole bot after a :bash:`git pull`, just execute
.. code-block:: bash
$ make update
.. |image0| image:: https://img.shields.io/badge/python-3.7%20%7C%203.8%20%7C%203.9%20%7C%203.10-%23007ec6
.. |image1| image:: https://img.shields.io/badge/dynamic/json?color=%23dfb317&label=issues&query=%24.open_issues_count&suffix=%20open&url=https%3A%2F%2Fgit.gnous.eu%2Fapi%2Fv1%2Frepos%2FGnousEU%2Ftuxbot-bot%2F

17
setup.cfg
View file

@ -1,24 +1,29 @@
[metadata] [metadata]
name = Tuxbot-bot name = Tuxbot-bot
version = attr: tuxbot.__version__ version = attr: tuxbot.__version__
description = A bot made for GnousEU and OpenSource url = https://git.gnous.eu/gnouseu/tuxbot-bot/
author = Romain J. author = Romain J.
author_email = romain@gnous.eu author_email = romain@gnous.eu
url = https://git.gnous.eu/gnouseu/tuxbot-bot/ maintainer = Romain J.
maintainer_email = romain@gnous.eu
description = A discord bot made for GnousEU's guild and OpenSource
long_description = file: README.rst
license = agplv3
platforms = linux
[options] [options]
packages = find_namespace: packages = find_namespace:
python_requires = >=3.7 python_requires = >=3.7
;todo: remove flatten_dict (core/config.py)
install_requires = install_requires =
appdirs>=1.4.4 appdirs>=1.4.4
Babel>=2.8.0 Babel>=2.8.0
discord.py==1.4.1 black==1.4.4
discord.py==1.5.0
discord_flags==2.1.1 discord_flags==2.1.1
flatten_dict>=0.3.0 humanize==2.6.0
jishaku>=1.19.1.200 jishaku>=1.19.1.200
psutil>=5.7.2 psutil>=5.7.2
PyYAML>=5.3.1 pyyaml>=5.3.1
rich>=6.0.0 rich>=6.0.0
structured_config>=4.12 structured_config>=4.12

5
tuxbot/__init__.py
View file

@ -1,15 +1,14 @@
import os import os
from collections import namedtuple from collections import namedtuple
build = os.popen('git rev-parse --short HEAD').read().strip() build = os.popen("git rev-parse --short HEAD").read().strip()
info = os.popen('git log -n 1 -s --format="%s"').read().strip() info = os.popen('git log -n 1 -s --format="%s"').read().strip()
VersionInfo = namedtuple( VersionInfo = namedtuple(
"VersionInfo", "major minor micro releaselevel build, info" "VersionInfo", "major minor micro releaselevel build, info"
) )
version_info = VersionInfo( version_info = VersionInfo(
major=3, minor=0, micro=0, major=3, minor=0, micro=0, releaselevel="alpha", build=build, info=info
releaselevel="alpha", build=build, info=info
) )
__version__ = "v{}.{}.{}-{}.{}".format( __version__ = "v{}.{}.{}-{}.{}".format(

6
tuxbot/__main__.py
View file

@ -10,12 +10,14 @@ install(console=console)
def main() -> NoReturn: def main() -> NoReturn:
try: try:
from .__run__ import run from .__run__ import run # pylint: disable=import-outside-toplevel
run() run()
except SystemExit as exc: except SystemExit as exc:
if exc.code == ExitCodes.RESTART: if exc.code == ExitCodes.RESTART:
from .__run__ import run # reimport to load changes # reimport to load changes
from .__run__ import run # pylint: disable=import-outside-toplevel
run() run()
else: else:
raise exc raise exc

102
tuxbot/__run__.py
View file

@ -1,27 +1,29 @@
import argparse import argparse
import asyncio import asyncio
import json
import logging import logging
import signal import signal
import sys import sys
import os import os
import tracemalloc
from argparse import Namespace from argparse import Namespace
from typing import NoReturn from typing import NoReturn
from datetime import datetime
import discord import discord
import humanize
import pip import pip
import tracemalloc
from rich.columns import Columns from rich.columns import Columns
from rich.console import Console from rich.console import Console
from rich.panel import Panel from rich.panel import Panel
from rich.traceback import install from rich.traceback import install
from rich.table import Table, box from rich.table import Table, box
from rich.text import Text from rich.text import Text
from rich import print from rich import print as rprint
import tuxbot.logging import tuxbot.logging
from tuxbot.core import data_manager
from tuxbot.core.bot import Tux from tuxbot.core.bot import Tux
from tuxbot.core import data_manager
from tuxbot.core import config
from . import __version__, version_info, ExitCodes from . import __version__, version_info, ExitCodes
log = logging.getLogger("tuxbot.main") log = logging.getLogger("tuxbot.main")
@ -30,35 +32,37 @@ console = Console()
install(console=console) install(console=console)
tracemalloc.start() tracemalloc.start()
BORDER_STYLE = "not dim"
def list_instances() -> NoReturn: def list_instances() -> NoReturn:
"""List all available instances """List all available instances"""
app_config = config.ConfigFile(
""" data_manager.config_dir / "config.yaml", config.AppConfig
with data_manager.config_file.open() as fs: ).config
data = json.load(fs)
console.print( console.print(
Panel("[bold green]Instances", style="green"), Panel("[bold green]Instances", style="green"), justify="center"
justify="center"
) )
console.print() console.print()
columns = Columns(expand=True, padding=2, align="center") columns = Columns(expand=True, padding=2, align="center")
for instance, details in data.items(): for instance, details in app_config.instances.items():
is_running = details.get('IS_RUNNING') active = details["active"]
last_run = (
humanize.naturaltime(
datetime.now() - datetime.fromtimestamp(details["last_run"])
)
or "[i]unknown"
)
table = Table( table = Table(
style="dim", border_style="not dim", style="dim", border_style=BORDER_STYLE, box=box.HEAVY_HEAD
box=box.HEAVY_HEAD
) )
table.add_column("Name") table.add_column("Name")
table.add_column(("Running" if is_running else "Down") + " since") table.add_column(("Running since" if active else "Last run"))
table.add_row(instance, "42") table.add_row(instance, last_run)
table.title = Text( table.title = Text(instance, style="green" if active else "red")
instance,
style="green" if is_running else "red"
)
columns.add_renderable(table) columns.add_renderable(table)
console.print(columns) console.print(columns)
console.print() console.print()
@ -67,28 +71,22 @@ def list_instances() -> NoReturn:
def debug_info() -> NoReturn: def debug_info() -> NoReturn:
"""Show debug info relatives to the bot """Show debug info relatives to the bot"""
"""
python_version = sys.version.replace("\n", "") python_version = sys.version.replace("\n", "")
pip_version = pip.__version__ pip_version = pip.__version__
tuxbot_version = __version__ tuxbot_version = __version__
dpy_version = discord.__version__ dpy_version = discord.__version__
uptime = os.popen('uptime').read().strip().split() uptime = os.popen("uptime").read().strip().split()
console.print( console.print(
Panel("[bold blue]Debug Info", style="blue"), Panel("[bold blue]Debug Info", style="blue"), justify="center"
justify="center"
) )
console.print() console.print()
columns = Columns(expand=True, padding=2, align="center") columns = Columns(expand=True, padding=2, align="center")
table = Table( table = Table(style="dim", border_style=BORDER_STYLE, box=box.HEAVY_HEAD)
style="dim", border_style="not dim",
box=box.HEAVY_HEAD
)
table.add_column( table.add_column(
"Bot Info", "Bot Info",
) )
@ -100,10 +98,7 @@ def debug_info() -> NoReturn:
table.add_row(f"[u]Last change:[/u] {version_info.info}") table.add_row(f"[u]Last change:[/u] {version_info.info}")
columns.add_renderable(table) columns.add_renderable(table)
table = Table( table = Table(style="dim", border_style=BORDER_STYLE, box=box.HEAVY_HEAD)
style="dim", border_style="not dim",
box=box.HEAVY_HEAD
)
table.add_column( table.add_column(
"Python Info", "Python Info",
) )
@ -113,10 +108,7 @@ def debug_info() -> NoReturn:
table.add_row(f"[u]Discord.py version:[/u] {dpy_version}") table.add_row(f"[u]Discord.py version:[/u] {dpy_version}")
columns.add_renderable(table) columns.add_renderable(table)
table = Table( table = Table(style="dim", border_style=BORDER_STYLE, box=box.HEAVY_HEAD)
style="dim", border_style="not dim",
box=box.HEAVY_HEAD
)
table.add_column( table.add_column(
"Server Info", "Server Info",
) )
@ -152,19 +144,23 @@ def parse_cli_flags(args: list) -> Namespace:
usage="tuxbot <instance_name> [arguments]", usage="tuxbot <instance_name> [arguments]",
) )
parser.add_argument( parser.add_argument(
"--version", "-V", action="store_true", "--version",
help="Show tuxbot's used version" "-V",
action="store_true",
help="Show tuxbot's used version",
) )
parser.add_argument( parser.add_argument(
"--debug", action="store_true", "--debug", action="store_true", help="Show debug information."
help="Show debug information."
) )
parser.add_argument( parser.add_argument(
"--list-instances", "-L", action="store_true", "--list-instances",
help="List all instance names" "-L",
action="store_true",
help="List all instance names",
)
parser.add_argument(
"--token", "-T", type=str, help="Run Tuxbot with passed token"
) )
parser.add_argument("--token", "-T", type=str,
help="Run Tuxbot with passed token")
parser.add_argument( parser.add_argument(
"instance_name", "instance_name",
nargs="?", nargs="?",
@ -227,7 +223,7 @@ async def run_bot(tux: Tux, cli_flags: Namespace) -> None:
if cli_flags.token: if cli_flags.token:
token = cli_flags.token token = cli_flags.token
else: else:
token = tux.config("core").get("token") token = tux.config.Core.token
if not token: if not token:
log.critical("Token must be set if you want to login.") log.critical("Token must be set if you want to login.")
@ -245,15 +241,14 @@ async def run_bot(tux: Tux, cli_flags: Namespace) -> None:
) )
sys.exit(ExitCodes.CRITICAL) sys.exit(ExitCodes.CRITICAL)
except Exception as e: except Exception as e:
log.critical(e)
raise e raise e
return None return None
def run() -> NoReturn: def run() -> NoReturn:
"""Main function """Main function"""
"""
tux = None tux = None
cli_flags = parse_cli_flags(sys.argv[1:]) cli_flags = parse_cli_flags(sys.argv[1:])
@ -262,8 +257,8 @@ def run() -> NoReturn:
elif cli_flags.debug: elif cli_flags.debug:
debug_info() debug_info()
elif cli_flags.version: elif cli_flags.version:
print(f"Tuxbot V{version_info.major}") rprint(f"Tuxbot V{version_info.major}")
print(f"Complete Version: {__version__}") rprint(f"Complete Version: {__version__}")
sys.exit(os.EX_OK) sys.exit(os.EX_OK)
@ -298,6 +293,7 @@ def run() -> NoReturn:
log.info("Shutting down with exit code: %s", exc.code) log.info("Shutting down with exit code: %s", exc.code)
if tux is not None: if tux is not None:
loop.run_until_complete(shutdown_handler(tux, None, exc.code)) loop.run_until_complete(shutdown_handler(tux, None, exc.code))
raise
except Exception as exc: except Exception as exc:
log.error("Unexpected exception (%s): ", type(exc)) log.error("Unexpected exception (%s): ", type(exc))
console.print_exception() console.print_exception()

27
tuxbot/cogs/admin/admin.py
View file

@ -5,10 +5,16 @@ from discord.ext import commands
from tuxbot.core import checks from tuxbot.core import checks
from tuxbot.core.bot import Tux from tuxbot.core.bot import Tux
from tuxbot.core.i18n import Translator, find_locale, get_locale_name, available_locales from tuxbot.core.i18n import (
Translator,
find_locale,
get_locale_name,
available_locales,
)
from tuxbot.core.utils.functions.extra import ( from tuxbot.core.utils.functions.extra import (
group_extra, command_extra, group_extra,
ContextPlus command_extra,
ContextPlus,
) )
log = logging.getLogger("tuxbot.cogs.admin") log = logging.getLogger("tuxbot.cogs.admin")
@ -20,30 +26,33 @@ class Admin(commands.Cog, name="Admin"):
self.bot = bot self.bot = bot
async def _save_lang(self, ctx: ContextPlus, lang: str): async def _save_lang(self, ctx: ContextPlus, lang: str):
await self.bot.config.update("core", f"guild.{ctx.guild.id}.locale", lang) await self.bot.config.update(
"core", f"guild.{ctx.guild.id}.locale", lang
)
@group_extra(name="lang", aliases=["locale", "langue"], deletable=True) @group_extra(name="lang", aliases=["locale", "langue"], deletable=True)
@commands.guild_only() @commands.guild_only()
@checks.is_admin() @checks.is_admin()
async def _lang(self, ctx: ContextPlus): async def _lang(self, ctx: ContextPlus):
"""Manage lang settings.""" """Manage lang settings."""
pass
@_lang.command(name="set", aliases=["define", "choice"]) @_lang.command(name="set", aliases=["define", "choice"])
async def _lang_set(self, ctx: ContextPlus, lang: str): async def _lang_set(self, ctx: ContextPlus, lang: str):
try: try:
await self._save_lang(ctx, find_locale(lang.lower())) await self._save_lang(ctx, find_locale(lang.lower()))
await ctx.send( await ctx.send(
_("Locale changed to {lang} successfully", ctx, self.bot.config).format( _(
lang=f"`{get_locale_name(lang).lower()}`" "Locale changed to {lang} successfully",
) ctx,
self.bot.config,
).format(lang=f"`{get_locale_name(lang).lower()}`")
) )
except NotImplementedError: except NotImplementedError:
await self._lang_list(ctx) await self._lang_list(ctx)
@_lang.command(name="list", aliases=["liste", "all", "view"]) @_lang.command(name="list", aliases=["liste", "all", "view"])
async def _lang_list(self, ctx: ContextPlus): async def _lang_list(self, ctx: ContextPlus):
description = '' description = ""
for key, value in available_locales.items(): for key, value in available_locales.items():
description += f":flag_{key[-2:].lower()}: {value[0]}\n" description += f":flag_{key[-2:].lower()}: {value[0]}\n"

10
tuxbot/cogs/admin/config.py
View file

@ -10,9 +10,9 @@ class AdminConfig(Structure):
extra = { extra = {
'dm': str, "dm": str,
'mentions': str, "mentions": str,
'guilds': str, "guilds": str,
'errors': str, "errors": str,
'gateway': str, "gateway": str,
} }

3
tuxbot/core/__init__.py
View file

@ -1,8 +1,9 @@
from .. import __version__, version_info, VersionInfo, ExitCodes from .. import __version__, version_info, VersionInfo, ExitCodes
from .config import Config from .config import Config, ConfigFile
__all__ = [ __all__ = [
"Config", "Config",
"ConfigFile",
"__version__", "__version__",
"version_info", "version_info",
"VersionInfo", "VersionInfo",

100
tuxbot/core/bot.py
View file

@ -1,7 +1,6 @@
import asyncio import asyncio
import datetime import datetime
import logging import logging
import sys
from typing import List, Union from typing import List, Union
import discord import discord
@ -13,32 +12,32 @@ from rich.panel import Panel
from rich.progress import Progress, TextColumn, BarColumn from rich.progress import Progress, TextColumn, BarColumn
from rich.table import Table from rich.table import Table
from rich.traceback import install from rich.traceback import install
from tuxbot import version_info from tuxbot import version_info
from . import Config from . import Config, ConfigFile
from .data_manager import logs_data_path from .data_manager import logs_data_path, data_path
from . import __version__, ExitCodes from . import __version__, ExitCodes
from . import exceptions
from .utils.functions.extra import ContextPlus from .utils.functions.extra import ContextPlus
from .utils.functions.prefix import get_prefixes
log = logging.getLogger("tuxbot") log = logging.getLogger("tuxbot")
console = Console() console = Console()
install(console=console) install(console=console)
packages: List[str] = [ packages: List[str] = ["jishaku", "tuxbot.cogs.admin"]
"jishaku",
"tuxbot.cogs.admin"
]
class Tux(commands.AutoShardedBot): class Tux(commands.AutoShardedBot):
_loading: asyncio.Task _loading: asyncio.Task
_progress = { _progress = {
'main': Progress( "main": Progress(
TextColumn("[bold blue]{task.fields[task_name]}", justify="right"), TextColumn("[bold blue]{task.fields[task_name]}", justify="right"),
BarColumn() BarColumn(),
), ),
'tasks': {} "tasks": {},
} }
def __init__(self, *args, cli_flags=None, **kwargs): def __init__(self, *args, cli_flags=None, **kwargs):
@ -50,14 +49,16 @@ class Tux(commands.AutoShardedBot):
self.last_exception = None self.last_exception = None
self.logs = logs_data_path(self.instance_name) self.logs = logs_data_path(self.instance_name)
self.config = Config(self.instance_name) self.config: Config = ConfigFile(
str(data_path(self.instance_name) / "config.yaml"), Config
).config
async def _prefixes(bot, message) -> List[str]: async def _prefixes(bot, message) -> List[str]:
prefixes = self.config("core").get("prefixes") prefixes = self.config.Core.prefixes
prefixes.extend(self.config.get_prefixes(message.guild)) prefixes.extend(get_prefixes(self, message.guild))
if self.config("core").get("mentionable"): if self.config.Core.mentionable:
return commands.when_mentioned_or(*prefixes)(bot, message) return commands.when_mentioned_or(*prefixes)(bot, message)
return prefixes return prefixes
@ -67,7 +68,7 @@ class Tux(commands.AutoShardedBot):
if "owner_ids" in kwargs: if "owner_ids" in kwargs:
kwargs["owner_ids"] = set(kwargs["owner_ids"]) kwargs["owner_ids"] = set(kwargs["owner_ids"])
else: else:
kwargs["owner_ids"] = self.config.owners_id() kwargs["owner_ids"] = self.config.Core.owners_id
message_cache_size = 100_000 message_cache_size = 100_000
kwargs["max_messages"] = message_cache_size kwargs["max_messages"] = message_cache_size
@ -82,8 +83,7 @@ class Tux(commands.AutoShardedBot):
if packages: if packages:
with Progress() as progress: with Progress() as progress:
task = progress.add_task( task = progress.add_task(
"Loading packages...", "Loading packages...", total=len(packages)
total=len(packages)
) )
for package in packages: for package in packages:
@ -92,8 +92,7 @@ class Tux(commands.AutoShardedBot):
progress.console.print(f"{package} loaded") progress.console.print(f"{package} loaded")
except Exception as e: except Exception as e:
log.exception( log.exception(
f"Failed to load package {package}", "Failed to load package %s", package, exc_info=e
exc_info=e
) )
progress.console.print( progress.console.print(
f"[red]Failed to load package {package} " f"[red]Failed to load package {package} "
@ -117,22 +116,19 @@ class Tux(commands.AutoShardedBot):
console.print( console.print(
Panel(f"[bold blue]Tuxbot V{version_info.major}", style="blue"), Panel(f"[bold blue]Tuxbot V{version_info.major}", style="blue"),
justify="center" justify="center",
) )
console.print() console.print()
columns = Columns(expand=True, padding=2, align="center") columns = Columns(expand=True, padding=2, align="center")
table = Table( table = Table(style="dim", border_style="not dim", box=box.HEAVY_HEAD)
style="dim", border_style="not dim",
box=box.HEAVY_HEAD
)
table.add_column( table.add_column(
"INFO", "INFO",
) )
table.add_row(str(self.user)) table.add_row(str(self.user))
table.add_row(f"Prefixes: {', '.join(self.config('core').get('prefixes'))}") table.add_row(f"Prefixes: {', '.join(self.config.Core.prefixes)}")
table.add_row(f"Language: {self.config('core').get('locale')}") table.add_row(f"Language: {self.config.Core.locale}")
table.add_row(f"Tuxbot Version: {__version__}") table.add_row(f"Tuxbot Version: {__version__}")
table.add_row(f"Discord.py Version: {discord.__version__}") table.add_row(f"Discord.py Version: {discord.__version__}")
table.add_row(f"Shards: {self.shard_count}") table.add_row(f"Shards: {self.shard_count}")
@ -140,10 +136,7 @@ class Tux(commands.AutoShardedBot):
table.add_row(f"Users: {len(self.users)}") table.add_row(f"Users: {len(self.users)}")
columns.add_renderable(table) columns.add_renderable(table)
table = Table( table = Table(style="dim", border_style="not dim", box=box.HEAVY_HEAD)
style="dim", border_style="not dim",
box=box.HEAVY_HEAD
)
table.add_column( table.add_column(
"COGS", "COGS",
) )
@ -159,8 +152,9 @@ class Tux(commands.AutoShardedBot):
console.print(columns) console.print(columns)
console.print() console.print()
async def is_owner(self, user: Union[discord.User, discord.Member])\ async def is_owner(
-> bool: self, user: Union[discord.User, discord.Member]
) -> bool:
"""Determines if the user is a bot owner. """Determines if the user is a bot owner.
Parameters Parameters
@ -171,7 +165,7 @@ class Tux(commands.AutoShardedBot):
------- -------
bool bool
""" """
if user.id in self.config.owners_id(): if user.id in self.config.Core.owners_id:
return True return True
owner = False owner = False
@ -179,50 +173,59 @@ class Tux(commands.AutoShardedBot):
app = await self.application_info() app = await self.application_info()
if app.team: if app.team:
ids = [m.id for m in app.team.members] ids = [m.id for m in app.team.members]
await self.config.update("core", "owners_id", ids) self.config.Core.owners_id = ids
owner = user.id in ids owner = user.id in ids
self._app_owners_fetched = True self._app_owners_fetched = True
return owner return owner
# pylint: disable=unused-argument
async def get_context(self, message: discord.Message, *, cls=None): async def get_context(self, message: discord.Message, *, cls=None):
return await super().get_context(message, cls=ContextPlus) return await super().get_context(message, cls=ContextPlus)
async def process_commands(self, message: discord.Message): async def process_commands(self, message: discord.Message):
"""Check for blacklists. """Check for blacklists."""
"""
if message.author.bot: if message.author.bot:
return return
if ( if (
message.guild.id in self.config.get_blacklist("guild") self.config.Servers.all[message.guild.id].blacklisted
or message.channel.id in self.config.get_blacklist("channel") or self.config.Channels.all[message.channel.id].blacklisted
or message.author.id in self.config.get_blacklist("user") or self.config.Users.all[message.author.id].blacklisted
): ):
return return
ctx = await self.get_context(message) ctx: ContextPlus = await self.get_context(message)
if ctx is None or ctx.valid is False: if ctx is None or ctx.valid is False:
self.dispatch("message_without_command", message) self.dispatch("message_without_command", message)
else: else:
if (
ctx.command
in self.config.Servers.all[message.guild.id].disabled_command
):
raise exceptions.DisabledCommandByServerOwner
if ctx.command in self.config.Core.disabled_command:
raise exceptions.DisabledCommandByBotOwner
await self.invoke(ctx) await self.invoke(ctx)
async def on_message(self, message: discord.Message): async def on_message(self, message: discord.Message):
await self.process_commands(message) await self.process_commands(message)
async def start(self, token, bot): async def start(self, token, bot): # pylint: disable=arguments-differ
"""Connect to Discord and start all connections. """Connect to Discord and start all connections.
Todo: add postgresql connect here Todo: add postgresql connect here
""" """
with self._progress.get("main") as pg: with self._progress.get("main") as progress:
task_id = self._progress.get("tasks")["connecting"] = pg.add_task( task_id = self._progress.get("tasks")[
"connecting", "connecting"
task_name="Connecting to Discord...", start=False ] = progress.add_task(
"connecting", task_name="Connecting to Discord...", start=False
) )
pg.update(task_id) progress.update(task_id)
await super().start(token, bot=bot) await super().start(token, bot=bot)
async def logout(self): async def logout(self):
@ -242,8 +245,7 @@ class Tux(commands.AutoShardedBot):
self._progress.get("main").stop() self._progress.get("main").stop()
pending = [ pending = [
t for t in asyncio.all_tasks() if t for t in asyncio.all_tasks() if t is not asyncio.current_task()
t is not asyncio.current_task()
] ]
for task in pending: for task in pending:

13
tuxbot/core/checks.py
View file

@ -22,9 +22,7 @@ __all__ = [
def is_mod(): def is_mod():
"""Is the user a moderator ? """Is the user a moderator ?"""
"""
async def pred(ctx): async def pred(ctx):
if await ctx.bot.is_owner(ctx.author): if await ctx.bot.is_owner(ctx.author):
@ -38,9 +36,7 @@ def is_mod():
def is_admin(): def is_admin():
"""Is the user admin ? """Is the user admin ?"""
"""
async def pred(ctx): async def pred(ctx):
if await ctx.bot.is_owner(ctx.author): if await ctx.bot.is_owner(ctx.author):
@ -53,7 +49,7 @@ def is_admin():
return commands.check(pred) return commands.check(pred)
async def check_permissions(ctx: "ContextPlus", **perms: Dict[str, bool]): async def check_permissions(ctx: ContextPlus, **perms: Dict[str, bool]):
"""Does a user have any perms ? """Does a user have any perms ?
Parameters Parameters
@ -66,8 +62,9 @@ async def check_permissions(ctx: "ContextPlus", **perms: Dict[str, bool]):
if await ctx.bot.is_owner(ctx.author): if await ctx.bot.is_owner(ctx.author):
return True return True
elif not perms: if not perms:
return False return False
resolved = ctx.channel.permissions_for(ctx.author) resolved = ctx.channel.permissions_for(ctx.author)
return all( return all(

33
tuxbot/core/config.py
View file

@ -1,16 +1,15 @@
import asyncio
import logging import logging
from typing import List, Dict from typing import List, Dict
from structured_config import ( from structured_config import (
Structure,
IntField,
StrField,
BoolField,
ConfigFile, ConfigFile,
Structure, IntField, StrField, BoolField
) )
import discord
from tuxbot.core.data_manager import data_path __all__ = ["Config", "ConfigFile"]
__all__ = ["Config"]
log = logging.getLogger("tuxbot.core.config") log = logging.getLogger("tuxbot.core.config")
@ -19,20 +18,30 @@ class Server(Structure):
prefixes: List[str] = [] prefixes: List[str] = []
disabled_command: List[str] = [] disabled_command: List[str] = []
locale: str = StrField("") locale: str = StrField("")
blacklisted: bool = BoolField(False)
class Channel(Structure):
disabled_command: List[str] = []
locale: str = StrField("")
blacklisted: bool = BoolField(False)
class User(Structure): class User(Structure):
aliases: List[dict] = [] aliases: List[dict] = []
locale: str = StrField("") locale: str = StrField("")
blacklisted: bool = BoolField(False)
class Config(Structure): class Config(Structure):
class Servers(Structure): class Servers(Structure):
count: int = IntField(0) all: Dict[int, Server] = {}
all: List[Server] = []
class Channels(Structure):
all: Dict[int, Channel] = {}
class Users(Structure): class Users(Structure):
all: List[User] = [] all: Dict[int, User] = {}
class Core(Structure): class Core(Structure):
owners_id: List[int] = [] owners_id: List[int] = []
@ -40,6 +49,7 @@ class Config(Structure):
token: str = StrField("") token: str = StrField("")
mentionable: bool = BoolField("") mentionable: bool = BoolField("")
locale: str = StrField("") locale: str = StrField("")
disabled_command: List[str] = []
class Cogs(Structure): class Cogs(Structure):
pass pass
@ -49,10 +59,11 @@ class Config(Structure):
# Configuration of Tuxbot Application (not the bot) # Configuration of Tuxbot Application (not the bot)
# ============================================================================= # =============================================================================
class AppConfig(Structure):
class Instance(Structure): class Instance(Structure):
path: str = StrField("") path: str = StrField("")
active: bool = BoolField(False) active: bool = BoolField(False)
last_run: int = IntField(0)
class AppConfig(Structure):
instances: Dict[str, Instance] = {} instances: Dict[str, Instance] = {}

46
tuxbot/core/data_manager.py
View file

@ -25,52 +25,6 @@ def data_path(instance_name: str) -> Path:
return Path(app_dir.user_data_dir) / "data" / instance_name return Path(app_dir.user_data_dir) / "data" / instance_name
def core_path(instance_name: str) -> Path:
"""Return Path for core configs.
Parameters
----------
instance_name:str
Returns
-------
Path
Generated path for core configs.
"""
return data_path(instance_name) / "data" / instance_name / "core"
def cogs_data_path(instance_name: str) -> Path:
"""Return Path for cogs configs.
Parameters
----------
instance_name:str
Returns
-------
Path
Generated path for cogs configs.
"""
return data_path(instance_name) / "data" / instance_name / "cogs"
def cog_data_path(instance_name: str, cog_name: str) -> Path:
"""Return Path for chosen configs for cog.
Parameters
----------
instance_name:str
cog_name:str
Returns
-------
Path
Generated path for cog's configs.
"""
return data_path(instance_name) / "data" / instance_name / "cogs" / cog_name
def logs_data_path(instance_name: str) -> Path: def logs_data_path(instance_name: str) -> Path:
"""Return Path for logs. """Return Path for logs.

18
tuxbot/core/i18n.py
View file

@ -29,9 +29,7 @@ def find_locale(locale: str) -> str:
def get_locale_name(locale: str) -> str: def get_locale_name(locale: str) -> str:
"""Return the name of this `locale` """Return the name of this `locale`"""
"""
return available_locales.get(find_locale(locale))[0] return available_locales.get(find_locale(locale))[0]
@ -57,7 +55,9 @@ class Translator(Callable[[str], str]):
self.load_translations() self.load_translations()
def __call__(self, untranslated: str, ctx: ContextPlus, config: Config) -> str: def __call__(
self, untranslated: str, ctx: ContextPlus, config: Config
) -> str:
try: try:
locale = config.get_value( locale = config.get_value(
"core", "core",
@ -75,10 +75,8 @@ class Translator(Callable[[str], str]):
) )
def load_translations(self): def load_translations(self):
"""Loads the current translations. """Loads the current translations."""
for locale in available_locales:
"""
for locale in available_locales.keys():
locale_path = self.cog_folder / "locales" / f"{locale}.po" locale_path = self.cog_folder / "locales" / f"{locale}.po"
with locale_path.open("r") as f: with locale_path.open("r") as f:
@ -88,7 +86,9 @@ class Translator(Callable[[str], str]):
if message.id: if message.id:
self._add_translation(locale, message.id, message.string) self._add_translation(locale, message.id, message.string)
def _add_translation(self, locale: str, untranslated: str, translated: str): def _add_translation(
self, locale: str, untranslated: str, translated: str
):
if translated: if translated:
if not self.translations.get(locale, False): if not self.translations.get(locale, False):
self.translations[locale] = {} self.translations[locale] = {}

14
tuxbot/core/utils/functions/extra.py
View file

@ -6,6 +6,7 @@ from discord import Embed
from discord.ext import commands, flags from discord.ext import commands, flags
from rich.console import Console from rich.console import Console
console = Console() console = Console()
console.clear() console.clear()
@ -15,13 +16,13 @@ class ContextPlus(commands.Context):
async def send(self, content=None, *args, **kwargs): async def send(self, content=None, *args, **kwargs):
if content is not None: if content is not None:
content = content.replace( content = content.replace(
self.bot.config('core').get('token'), '<token>' self.bot.config("core").get("token"), "<token>"
) )
if kwargs.get('embed'): if kwargs.get("embed"):
e = str(kwargs.get('embed').to_dict()) e = str(kwargs.get("embed").to_dict())
e = e.replace(self.bot.config('core').get('token'), '<token>') e = e.replace(self.bot.config("core").get("token"), "<token>")
e = yaml.load(e, Loader=yaml.FullLoader) e = yaml.load(e, Loader=yaml.FullLoader)
kwargs['embed'] = Embed.from_dict(e) kwargs["embed"] = Embed.from_dict(e)
if ( if (
hasattr(self.command, "deletable") and self.command.deletable hasattr(self.command, "deletable") and self.command.deletable
@ -38,8 +39,7 @@ class ContextPlus(commands.Context):
try: try:
await self.bot.wait_for( await self.bot.wait_for(
"reaction_add", "reaction_add", timeout=42.0, check=check
timeout=42.0, check=check
) )
except asyncio.TimeoutError: except asyncio.TimeoutError:
await message.remove_reaction("🗑", self.bot.user) await message.remove_reaction("🗑", self.bot.user)

22
tuxbot/core/utils/functions/prefix.py Normal file
View file

@ -0,0 +1,22 @@
from typing import List
import discord
from tuxbot.core.bot import Tux
def get_prefixes(tux: Tux, guild: discord.Guild) -> List[str]:
"""Get custom prefixes for one guild.
Parameters
----------
tux:Tux
The bot instance.
guild:discord.Guild
The required guild prefixes.
Returns
-------
List[str]
List of all prefixes.
"""
return tux.config.Servers.all[guild.id].prefixes or []

18
tuxbot/logging.py
View file

@ -6,6 +6,12 @@ import sys
MAX_OLD_LOGS = 8 MAX_OLD_LOGS = 8
MAX_BYTES = 5_000_000 MAX_BYTES = 5_000_000
formatter = logging.Formatter(
"[{asctime}] [{levelname}] {name}: {message}",
datefmt="%Y-%m-%d %H:%M:%S",
style="{",
)
def init_logging(level: int, location: pathlib.Path) -> None: def init_logging(level: int, location: pathlib.Path) -> None:
"""Initialize loggers. """Initialize loggers.
@ -26,19 +32,15 @@ def init_logging(level: int, location: pathlib.Path) -> None:
base_logger.setLevel(level) base_logger.setLevel(level)
base_logger_file = location / "tuxbot.log" base_logger_file = location / "tuxbot.log"
formatter = logging.Formatter(
"[{asctime}] [{levelname}] {name}: {message}",
datefmt="%Y-%m-%d %H:%M:%S",
style="{",
)
dpy_handler = logging.handlers.RotatingFileHandler( dpy_handler = logging.handlers.RotatingFileHandler(
str(dpy_logger_file.resolve()), str(dpy_logger_file.resolve()),
maxBytes=MAX_BYTES, backupCount=MAX_OLD_LOGS maxBytes=MAX_BYTES,
backupCount=MAX_OLD_LOGS,
) )
base_handler = logging.handlers.RotatingFileHandler( base_handler = logging.handlers.RotatingFileHandler(
str(base_logger_file.resolve()), str(base_logger_file.resolve()),
maxBytes=MAX_BYTES, backupCount=MAX_OLD_LOGS maxBytes=MAX_BYTES,
backupCount=MAX_OLD_LOGS,
) )
stdout_handler = logging.StreamHandler(sys.stdout) stdout_handler = logging.StreamHandler(sys.stdout)

94
tuxbot/setup.py
View file

@ -10,6 +10,7 @@ from rich.console import Console
from rich.rule import Rule from rich.rule import Rule
from rich.traceback import install from rich.traceback import install
from tuxbot.logging import formatter
from tuxbot.core.data_manager import config_dir, app_dir from tuxbot.core.data_manager import config_dir, app_dir
from tuxbot.core import config from tuxbot.core import config
@ -20,7 +21,9 @@ install(console=console)
try: try:
config_dir.mkdir(parents=True, exist_ok=True) config_dir.mkdir(parents=True, exist_ok=True)
except PermissionError: except PermissionError:
console.print(f"mkdir: cannot create directory '{config_dir}': Permission denied") console.print(
f"mkdir: cannot create directory '{config_dir}': Permission denied"
)
sys.exit(1) sys.exit(1)
app_config = config.ConfigFile(config_dir / "config.yaml", config.AppConfig) app_config = config.ConfigFile(config_dir / "config.yaml", config.AppConfig)
@ -45,7 +48,7 @@ def get_name() -> str:
"What name do you want to give this instance?\n" "What name do you want to give this instance?\n"
"[i](valid characters: A-Z, a-z, 0-9, _, -)[/i]\n", "[i](valid characters: A-Z, a-z, 0-9, _, -)[/i]\n",
default="prod", default="prod",
console=console console=console,
) )
if re.fullmatch(r"[a-zA-Z0-9_\-]*", name) is None: if re.fullmatch(r"[a-zA-Z0-9_\-]*", name) is None:
console.print() console.print()
@ -89,7 +92,7 @@ def get_data_dir(instance_name: str) -> Path:
Prompt.ask( Prompt.ask(
"where do you want to save the configurations?", "where do you want to save the configurations?",
default=str(data_path), default=str(data_path),
console=console console=console,
) )
) )
@ -114,16 +117,15 @@ def get_data_dir(instance_name: str) -> Path:
f"`{instance_name}` instance" f"`{instance_name}` instance"
) )
if Prompt.ask( if (
"Please confirm", Prompt.ask(
choices=["y", "n"], default="y", "Please confirm", choices=["y", "n"], default="y", console=console
console=console )
) != "y": != "y"
):
console.print("Rerun the process to redo this configuration.") console.print("Rerun the process to redo this configuration.")
sys.exit(0) sys.exit(0)
(data_path_input / "core").mkdir(parents=True, exist_ok=True)
(data_path_input / "cogs").mkdir(parents=True, exist_ok=True)
(data_path_input / "logs").mkdir(parents=True, exist_ok=True) (data_path_input / "logs").mkdir(parents=True, exist_ok=True)
return data_path_input return data_path_input
@ -143,13 +145,16 @@ def get_token() -> str:
token = Prompt.ask( token = Prompt.ask(
"Please enter the bot token " "Please enter the bot token "
"(you can find it at https://discord.com/developers/applications)", "(you can find it at https://discord.com/developers/applications)",
console=console console=console,
) )
if re.fullmatch( if (
re.fullmatch(
r"([a-zA-Z0-9]{24}\.[a-zA-Z0-9_]{6}\.[a-zA-Z0-9_\-]{27}" r"([a-zA-Z0-9]{24}\.[a-zA-Z0-9_]{6}\.[a-zA-Z0-9_\-]{27}"
r"|mfa\.[a-zA-Z0-9_\-]{84})", r"|mfa\.[a-zA-Z0-9_\-]{84})",
token) \ token,
is None: )
is None
):
console.print("[prompt.invalid]ERROR: Invalid token provided") console.print("[prompt.invalid]ERROR: Invalid token provided")
token = "" token = ""
return token return token
@ -183,11 +188,12 @@ def get_multiple(
values = [user_input] values = [user_input]
while Prompt.ask( while (
confirmation, Prompt.ask(
choices=["y", "n"], default="y", confirmation, choices=["y", "n"], default="n", console=console
console=console )
) != "n": != "n"
):
new = prompt.ask("Other") new = prompt.ask("Other")
if new not in values: if new not in values:
@ -209,10 +215,10 @@ def additional_config() -> dict:
dict: dict:
Dict with cog name as key and configs as value. Dict with cog name as key and configs as value.
""" """
p = Path("tuxbot/cogs").glob("**/config.py") paths = Path("tuxbot/cogs").glob("**/config.py")
data = {} data = {}
for file in p: for file in paths:
console.print("\n" * 4) console.print("\n" * 4)
cog_name = str(file.parent).split("/")[-1] cog_name = str(file.parent).split("/")[-1]
data[cog_name] = {} data[cog_name] = {}
@ -238,9 +244,7 @@ def finish_setup(data_dir: Path) -> NoReturn:
Where to save configs. Where to save configs.
""" """
console.print( console.print(
Rule( Rule("Now, it's time to finish this setup by giving bot information")
"Now, it's time to finish this setup by giving bot information"
)
) )
console.print() console.print()
@ -248,24 +252,26 @@ def finish_setup(data_dir: Path) -> NoReturn:
console.print() console.print()
prefixes = get_multiple( prefixes = get_multiple(
"Choice a (or multiple) prefix for the bot", "Add another prefix ?", "Choice a (or multiple) prefix for the bot",
str "Add another prefix ?",
str,
) )
console.print() console.print()
mentionable = Prompt.ask( mentionable = (
Prompt.ask(
"Does the bot answer if it's mentioned?", "Does the bot answer if it's mentioned?",
choices=["y", "n"], choices=["y", "n"],
default="y" default="y",
) == "y" )
== "y"
)
console.print() console.print()
owners_id = get_multiple( owners_id = get_multiple(
"Give the owner id of this bot", "Add another owner ?", int "Give the owner id of this bot", "Add another owner ?", int
) )
# cogs_config = additional_config()
instance_config = config.ConfigFile( instance_config = config.ConfigFile(
str(data_dir / "config.yaml"), config.Config str(data_dir / "config.yaml"), config.Config
) )
@ -278,9 +284,7 @@ def finish_setup(data_dir: Path) -> NoReturn:
def basic_setup() -> NoReturn: def basic_setup() -> NoReturn:
"""Configs who refer to instances. """Configs who refer to instances."""
"""
console.print( console.print(
Rule( Rule(
"Hi ! it's time for you to give me information about you instance" "Hi ! it's time for you to give me information about you instance"
@ -295,16 +299,21 @@ def basic_setup() -> NoReturn:
console.print() console.print()
console.print( console.print(
f"WARNING: An instance named `{name}` already exists " f"WARNING: An instance named `{name}` already exists "
f"Continuing will overwrite this instance configs.", style="red" f"Continuing will overwrite this instance configs.",
style="red",
) )
if Prompt.ask( if (
Prompt.ask(
"Are you sure you want to continue?", "Are you sure you want to continue?",
choices=["y", "n"], default="n" choices=["y", "n"],
) == "n": default="n",
)
== "n"
):
console.print("Abandon...") console.print("Abandon...")
sys.exit(0) sys.exit(0)
instance = config.Instance() instance = config.AppConfig.Instance()
instance.path = str(data_dir.resolve()) instance.path = str(data_dir.resolve())
instance.active = False instance.active = False
@ -323,15 +332,10 @@ def basic_setup() -> NoReturn:
def setup() -> NoReturn: def setup() -> NoReturn:
try: try:
"""Create a new instance.""" # Create a new instance.
level = logging.DEBUG level = logging.DEBUG
base_logger = logging.getLogger("tuxbot") base_logger = logging.getLogger("tuxbot")
base_logger.setLevel(level) base_logger.setLevel(level)
formatter = logging.Formatter(
"[{asctime}] [{levelname}] {name}: {message}",
datefmt="%Y-%m-%d %H:%M:%S",
style="{",
)
stdout_handler = logging.StreamHandler(sys.stdout) stdout_handler = logging.StreamHandler(sys.stdout)
stdout_handler.setFormatter(formatter) stdout_handler.setFormatter(formatter)
base_logger.addHandler(stdout_handler) base_logger.addHandler(stdout_handler)