├── .gitignore ├── BotFather_settings ├── about.txt ├── description.txt └── set_commands.txt ├── LICENSE ├── Procfile ├── README.md ├── requirements.txt └── src ├── .gitignore ├── bot.py ├── categories.py ├── dbwrapper.py ├── ranker.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- 1 | # Our stuff 2 | database_versioning/ 3 | tracking 4 | 5 | # Byte-compiled / optimized / DLL files 6 | __pycache__/ 7 | *.py[cod] 8 | *$py.class 9 | 10 | # Heroku 11 | venv/ 12 | 13 | # VS Code stuff 14 | .vscode/ 15 | 16 | # environment 17 | .env 18 | .env.local 19 | 20 | # C extensions 21 | *.so 22 | 23 | # Distribution / packaging 24 | .Python 25 | build/ 26 | develop-eggs/ 27 | dist/ 28 | downloads/ 29 | eggs/ 30 | .eggs/ 31 | lib/ 32 | lib64/ 33 | parts/ 34 | sdist/ 35 | var/ 36 | wheels/ 37 | pip-wheel-metadata/ 38 | share/python-wheels/ 39 | *.egg-info/ 40 | .installed.cfg 41 | *.egg 42 | MANIFEST 43 | 44 | # PyInstaller 45 | # Usually these files are written by a python script from a template 46 | # before PyInstaller builds the exe, so as to inject date/other infos into it. 47 | *.manifest 48 | *.spec 49 | 50 | # Installer logs 51 | pip-log.txt 52 | pip-delete-this-directory.txt 53 | 54 | # Unit test / coverage reports 55 | htmlcov/ 56 | .tox/ 57 | .nox/ 58 | .coverage 59 | .coverage.* 60 | .cache 61 | nosetests.xml 62 | coverage.xml 63 | *.cover 64 | *.py,cover 65 | .hypothesis/ 66 | .pytest_cache/ 67 | 68 | # Translations 69 | *.mo 70 | *.pot 71 | 72 | # Django stuff: 73 | *.log 74 | local_settings.py 75 | db.sqlite3 76 | db.sqlite3-journal 77 | 78 | # Flask stuff: 79 | instance/ 80 | .webassets-cache 81 | 82 | # Scrapy stuff: 83 | .scrapy 84 | 85 | # Sphinx documentation 86 | docs/_build/ 87 | 88 | # PyBuilder 89 | target/ 90 | 91 | # Jupyter Notebook 92 | .ipynb_checkpoints 93 | 94 | # IPython 95 | profile_default/ 96 | ipython_config.py 97 | 98 | # pyenv 99 | .python-version 100 | 101 | # pipenv 102 | # According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control. 103 | # However, in case of collaboration, if having platform-specific dependencies or dependencies 104 | # having no cross-platform support, pipenv may install dependencies that don't work, or not 105 | # install all needed dependencies. 106 | #Pipfile.lock 107 | 108 | # PEP 582; used by e.g. github.com/David-OConnor/pyflow 109 | __pypackages__/ 110 | 111 | # Celery stuff 112 | celerybeat-schedule 113 | celerybeat.pid 114 | 115 | # SageMath parsed files 116 | *.sage.py 117 | 118 | # Environments 119 | .env 120 | .venv 121 | env/ 122 | venv/ 123 | ENV/ 124 | env.bak/ 125 | venv.bak/ 126 | 127 | # Spyder project settings 128 | .spyderproject 129 | .spyproject 130 | 131 | # Rope project settings 132 | .ropeproject 133 | 134 | # mkdocs documentation 135 | /site 136 | 137 | # mypy 138 | .mypy_cache/ 139 | .dmypy.json 140 | dmypy.json 141 | 142 | # Pyre type checker 143 | .pyre/ 144 | -------------------------------------------------------------------------------- /BotFather_settings/about.txt: -------------------------------------------------------------------------------- 1 | Meu objetivo é conectar estudantes universitários por meio de interesses em comum. Bora lá? -------------------------------------------------------------------------------- /BotFather_settings/description.txt: -------------------------------------------------------------------------------- 1 | Essa mensagem irá aparecem antes da pessoa dar o /start, embaixo de "What can this bot do?". 2 | 3 | Sou o bot do Approxima e meu objetivo é conectar estudantes universitários por meio de interesses em comum. Estou animado pra ver quem você irá conhecer por minha causa! :) 4 | Se precisar de uma lista de comandos, digite "/" que eu te mostrarei os disponíveis (ou use o comando /help). -------------------------------------------------------------------------------- /BotFather_settings/set_commands.txt: -------------------------------------------------------------------------------- 1 | prefs - Selecione suas categorias de interesse. 2 | show - Mostra uma pessoa que tem interesses em comum. 3 | random - Mostra uma pessoa aleatória. 4 | clear - Limpa a lista de "rejeitados". 5 | pending - Mostra uma solicitação que você possui e ainda não respondeu. 6 | friends - Mostra todas as suas conexões. 7 | name - Troca o seu nome. 8 | desc - Troca a sua descrição. 9 | help - Mostra os comandos disponíveis. 10 | start - Reinicia o bot. -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- 1 | GNU GENERAL PUBLIC LICENSE 2 | Version 3, 29 June 2007 3 | 4 | Copyright (C) 2007 Free Software Foundation, Inc. 5 | Everyone is permitted to copy and distribute verbatim copies 6 | of this license document, but changing it is not allowed. 7 | 8 | Preamble 9 | 10 | The GNU General Public License is a free, copyleft license for 11 | software and other kinds of works. 12 | 13 | The licenses for most software and other practical works are designed 14 | to take away your freedom to share and change the works. By contrast, 15 | the GNU General Public License is intended to guarantee your freedom to 16 | share and change all versions of a program--to make sure it remains free 17 | software for all its users. We, the Free Software Foundation, use the 18 | GNU General Public License for most of our software; it applies also to 19 | any other work released this way by its authors. You can apply it to 20 | your programs, too. 21 | 22 | When we speak of free software, we are referring to freedom, not 23 | price. Our General Public Licenses are designed to make sure that you 24 | have the freedom to distribute copies of free software (and charge for 25 | them if you wish), that you receive source code or can get it if you 26 | want it, that you can change the software or use pieces of it in new 27 | free programs, and that you know you can do these things. 28 | 29 | To protect your rights, we need to prevent others from denying you 30 | these rights or asking you to surrender the rights. Therefore, you have 31 | certain responsibilities if you distribute copies of the software, or if 32 | you modify it: responsibilities to respect the freedom of others. 33 | 34 | For example, if you distribute copies of such a program, whether 35 | gratis or for a fee, you must pass on to the recipients the same 36 | freedoms that you received. You must make sure that they, too, receive 37 | or can get the source code. And you must show them these terms so they 38 | know their rights. 39 | 40 | Developers that use the GNU GPL protect your rights with two steps: 41 | (1) assert copyright on the software, and (2) offer you this License 42 | giving you legal permission to copy, distribute and/or modify it. 43 | 44 | For the developers' and authors' protection, the GPL clearly explains 45 | that there is no warranty for this free software. For both users' and 46 | authors' sake, the GPL requires that modified versions be marked as 47 | changed, so that their problems will not be attributed erroneously to 48 | authors of previous versions. 49 | 50 | Some devices are designed to deny users access to install or run 51 | modified versions of the software inside them, although the manufacturer 52 | can do so. This is fundamentally incompatible with the aim of 53 | protecting users' freedom to change the software. The systematic 54 | pattern of such abuse occurs in the area of products for individuals to 55 | use, which is precisely where it is most unacceptable. Therefore, we 56 | have designed this version of the GPL to prohibit the practice for those 57 | products. If such problems arise substantially in other domains, we 58 | stand ready to extend this provision to those domains in future versions 59 | of the GPL, as needed to protect the freedom of users. 60 | 61 | Finally, every program is threatened constantly by software patents. 62 | States should not allow patents to restrict development and use of 63 | software on general-purpose computers, but in those that do, we wish to 64 | avoid the special danger that patents applied to a free program could 65 | make it effectively proprietary. To prevent this, the GPL assures that 66 | patents cannot be used to render the program non-free. 67 | 68 | The precise terms and conditions for copying, distribution and 69 | modification follow. 70 | 71 | TERMS AND CONDITIONS 72 | 73 | 0. Definitions. 74 | 75 | "This License" refers to version 3 of the GNU General Public License. 76 | 77 | "Copyright" also means copyright-like laws that apply to other kinds of 78 | works, such as semiconductor masks. 79 | 80 | "The Program" refers to any copyrightable work licensed under this 81 | License. Each licensee is addressed as "you". "Licensees" and 82 | "recipients" may be individuals or organizations. 83 | 84 | To "modify" a work means to copy from or adapt all or part of the work 85 | in a fashion requiring copyright permission, other than the making of an 86 | exact copy. The resulting work is called a "modified version" of the 87 | earlier work or a work "based on" the earlier work. 88 | 89 | A "covered work" means either the unmodified Program or a work based 90 | on the Program. 91 | 92 | To "propagate" a work means to do anything with it that, without 93 | permission, would make you directly or secondarily liable for 94 | infringement under applicable copyright law, except executing it on a 95 | computer or modifying a private copy. Propagation includes copying, 96 | distribution (with or without modification), making available to the 97 | public, and in some countries other activities as well. 98 | 99 | To "convey" a work means any kind of propagation that enables other 100 | parties to make or receive copies. Mere interaction with a user through 101 | a computer network, with no transfer of a copy, is not conveying. 102 | 103 | An interactive user interface displays "Appropriate Legal Notices" 104 | to the extent that it includes a convenient and prominently visible 105 | feature that (1) displays an appropriate copyright notice, and (2) 106 | tells the user that there is no warranty for the work (except to the 107 | extent that warranties are provided), that licensees may convey the 108 | work under this License, and how to view a copy of this License. If 109 | the interface presents a list of user commands or options, such as a 110 | menu, a prominent item in the list meets this criterion. 111 | 112 | 1. Source Code. 113 | 114 | The "source code" for a work means the preferred form of the work 115 | for making modifications to it. "Object code" means any non-source 116 | form of a work. 117 | 118 | A "Standard Interface" means an interface that either is an official 119 | standard defined by a recognized standards body, or, in the case of 120 | interfaces specified for a particular programming language, one that 121 | is widely used among developers working in that language. 122 | 123 | The "System Libraries" of an executable work include anything, other 124 | than the work as a whole, that (a) is included in the normal form of 125 | packaging a Major Component, but which is not part of that Major 126 | Component, and (b) serves only to enable use of the work with that 127 | Major Component, or to implement a Standard Interface for which an 128 | implementation is available to the public in source code form. A 129 | "Major Component", in this context, means a major essential component 130 | (kernel, window system, and so on) of the specific operating system 131 | (if any) on which the executable work runs, or a compiler used to 132 | produce the work, or an object code interpreter used to run it. 133 | 134 | The "Corresponding Source" for a work in object code form means all 135 | the source code needed to generate, install, and (for an executable 136 | work) run the object code and to modify the work, including scripts to 137 | control those activities. However, it does not include the work's 138 | System Libraries, or general-purpose tools or generally available free 139 | programs which are used unmodified in performing those activities but 140 | which are not part of the work. For example, Corresponding Source 141 | includes interface definition files associated with source files for 142 | the work, and the source code for shared libraries and dynamically 143 | linked subprograms that the work is specifically designed to require, 144 | such as by intimate data communication or control flow between those 145 | subprograms and other parts of the work. 146 | 147 | The Corresponding Source need not include anything that users 148 | can regenerate automatically from other parts of the Corresponding 149 | Source. 150 | 151 | The Corresponding Source for a work in source code form is that 152 | same work. 153 | 154 | 2. Basic Permissions. 155 | 156 | All rights granted under this License are granted for the term of 157 | copyright on the Program, and are irrevocable provided the stated 158 | conditions are met. This License explicitly affirms your unlimited 159 | permission to run the unmodified Program. The output from running a 160 | covered work is covered by this License only if the output, given its 161 | content, constitutes a covered work. This License acknowledges your 162 | rights of fair use or other equivalent, as provided by copyright law. 163 | 164 | You may make, run and propagate covered works that you do not 165 | convey, without conditions so long as your license otherwise remains 166 | in force. You may convey covered works to others for the sole purpose 167 | of having them make modifications exclusively for you, or provide you 168 | with facilities for running those works, provided that you comply with 169 | the terms of this License in conveying all material for which you do 170 | not control copyright. Those thus making or running the covered works 171 | for you must do so exclusively on your behalf, under your direction 172 | and control, on terms that prohibit them from making any copies of 173 | your copyrighted material outside their relationship with you. 174 | 175 | Conveying under any other circumstances is permitted solely under 176 | the conditions stated below. Sublicensing is not allowed; section 10 177 | makes it unnecessary. 178 | 179 | 3. Protecting Users' Legal Rights From Anti-Circumvention Law. 180 | 181 | No covered work shall be deemed part of an effective technological 182 | measure under any applicable law fulfilling obligations under article 183 | 11 of the WIPO copyright treaty adopted on 20 December 1996, or 184 | similar laws prohibiting or restricting circumvention of such 185 | measures. 186 | 187 | When you convey a covered work, you waive any legal power to forbid 188 | circumvention of technological measures to the extent such circumvention 189 | is effected by exercising rights under this License with respect to 190 | the covered work, and you disclaim any intention to limit operation or 191 | modification of the work as a means of enforcing, against the work's 192 | users, your or third parties' legal rights to forbid circumvention of 193 | technological measures. 194 | 195 | 4. Conveying Verbatim Copies. 196 | 197 | You may convey verbatim copies of the Program's source code as you 198 | receive it, in any medium, provided that you conspicuously and 199 | appropriately publish on each copy an appropriate copyright notice; 200 | keep intact all notices stating that this License and any 201 | non-permissive terms added in accord with section 7 apply to the code; 202 | keep intact all notices of the absence of any warranty; and give all 203 | recipients a copy of this License along with the Program. 204 | 205 | You may charge any price or no price for each copy that you convey, 206 | and you may offer support or warranty protection for a fee. 207 | 208 | 5. Conveying Modified Source Versions. 209 | 210 | You may convey a work based on the Program, or the modifications to 211 | produce it from the Program, in the form of source code under the 212 | terms of section 4, provided that you also meet all of these conditions: 213 | 214 | a) The work must carry prominent notices stating that you modified 215 | it, and giving a relevant date. 216 | 217 | b) The work must carry prominent notices stating that it is 218 | released under this License and any conditions added under section 219 | 7. This requirement modifies the requirement in section 4 to 220 | "keep intact all notices". 221 | 222 | c) You must license the entire work, as a whole, under this 223 | License to anyone who comes into possession of a copy. This 224 | License will therefore apply, along with any applicable section 7 225 | additional terms, to the whole of the work, and all its parts, 226 | regardless of how they are packaged. This License gives no 227 | permission to license the work in any other way, but it does not 228 | invalidate such permission if you have separately received it. 229 | 230 | d) If the work has interactive user interfaces, each must display 231 | Appropriate Legal Notices; however, if the Program has interactive 232 | interfaces that do not display Appropriate Legal Notices, your 233 | work need not make them do so. 234 | 235 | A compilation of a covered work with other separate and independent 236 | works, which are not by their nature extensions of the covered work, 237 | and which are not combined with it such as to form a larger program, 238 | in or on a volume of a storage or distribution medium, is called an 239 | "aggregate" if the compilation and its resulting copyright are not 240 | used to limit the access or legal rights of the compilation's users 241 | beyond what the individual works permit. Inclusion of a covered work 242 | in an aggregate does not cause this License to apply to the other 243 | parts of the aggregate. 244 | 245 | 6. Conveying Non-Source Forms. 246 | 247 | You may convey a covered work in object code form under the terms 248 | of sections 4 and 5, provided that you also convey the 249 | machine-readable Corresponding Source under the terms of this License, 250 | in one of these ways: 251 | 252 | a) Convey the object code in, or embodied in, a physical product 253 | (including a physical distribution medium), accompanied by the 254 | Corresponding Source fixed on a durable physical medium 255 | customarily used for software interchange. 256 | 257 | b) Convey the object code in, or embodied in, a physical product 258 | (including a physical distribution medium), accompanied by a 259 | written offer, valid for at least three years and valid for as 260 | long as you offer spare parts or customer support for that product 261 | model, to give anyone who possesses the object code either (1) a 262 | copy of the Corresponding Source for all the software in the 263 | product that is covered by this License, on a durable physical 264 | medium customarily used for software interchange, for a price no 265 | more than your reasonable cost of physically performing this 266 | conveying of source, or (2) access to copy the 267 | Corresponding Source from a network server at no charge. 268 | 269 | c) Convey individual copies of the object code with a copy of the 270 | written offer to provide the Corresponding Source. This 271 | alternative is allowed only occasionally and noncommercially, and 272 | only if you received the object code with such an offer, in accord 273 | with subsection 6b. 274 | 275 | d) Convey the object code by offering access from a designated 276 | place (gratis or for a charge), and offer equivalent access to the 277 | Corresponding Source in the same way through the same place at no 278 | further charge. You need not require recipients to copy the 279 | Corresponding Source along with the object code. If the place to 280 | copy the object code is a network server, the Corresponding Source 281 | may be on a different server (operated by you or a third party) 282 | that supports equivalent copying facilities, provided you maintain 283 | clear directions next to the object code saying where to find the 284 | Corresponding Source. Regardless of what server hosts the 285 | Corresponding Source, you remain obligated to ensure that it is 286 | available for as long as needed to satisfy these requirements. 287 | 288 | e) Convey the object code using peer-to-peer transmission, provided 289 | you inform other peers where the object code and Corresponding 290 | Source of the work are being offered to the general public at no 291 | charge under subsection 6d. 292 | 293 | A separable portion of the object code, whose source code is excluded 294 | from the Corresponding Source as a System Library, need not be 295 | included in conveying the object code work. 296 | 297 | A "User Product" is either (1) a "consumer product", which means any 298 | tangible personal property which is normally used for personal, family, 299 | or household purposes, or (2) anything designed or sold for incorporation 300 | into a dwelling. In determining whether a product is a consumer product, 301 | doubtful cases shall be resolved in favor of coverage. For a particular 302 | product received by a particular user, "normally used" refers to a 303 | typical or common use of that class of product, regardless of the status 304 | of the particular user or of the way in which the particular user 305 | actually uses, or expects or is expected to use, the product. A product 306 | is a consumer product regardless of whether the product has substantial 307 | commercial, industrial or non-consumer uses, unless such uses represent 308 | the only significant mode of use of the product. 309 | 310 | "Installation Information" for a User Product means any methods, 311 | procedures, authorization keys, or other information required to install 312 | and execute modified versions of a covered work in that User Product from 313 | a modified version of its Corresponding Source. The information must 314 | suffice to ensure that the continued functioning of the modified object 315 | code is in no case prevented or interfered with solely because 316 | modification has been made. 317 | 318 | If you convey an object code work under this section in, or with, or 319 | specifically for use in, a User Product, and the conveying occurs as 320 | part of a transaction in which the right of possession and use of the 321 | User Product is transferred to the recipient in perpetuity or for a 322 | fixed term (regardless of how the transaction is characterized), the 323 | Corresponding Source conveyed under this section must be accompanied 324 | by the Installation Information. But this requirement does not apply 325 | if neither you nor any third party retains the ability to install 326 | modified object code on the User Product (for example, the work has 327 | been installed in ROM). 328 | 329 | The requirement to provide Installation Information does not include a 330 | requirement to continue to provide support service, warranty, or updates 331 | for a work that has been modified or installed by the recipient, or for 332 | the User Product in which it has been modified or installed. Access to a 333 | network may be denied when the modification itself materially and 334 | adversely affects the operation of the network or violates the rules and 335 | protocols for communication across the network. 336 | 337 | Corresponding Source conveyed, and Installation Information provided, 338 | in accord with this section must be in a format that is publicly 339 | documented (and with an implementation available to the public in 340 | source code form), and must require no special password or key for 341 | unpacking, reading or copying. 342 | 343 | 7. Additional Terms. 344 | 345 | "Additional permissions" are terms that supplement the terms of this 346 | License by making exceptions from one or more of its conditions. 347 | Additional permissions that are applicable to the entire Program shall 348 | be treated as though they were included in this License, to the extent 349 | that they are valid under applicable law. If additional permissions 350 | apply only to part of the Program, that part may be used separately 351 | under those permissions, but the entire Program remains governed by 352 | this License without regard to the additional permissions. 353 | 354 | When you convey a copy of a covered work, you may at your option 355 | remove any additional permissions from that copy, or from any part of 356 | it. (Additional permissions may be written to require their own 357 | removal in certain cases when you modify the work.) You may place 358 | additional permissions on material, added by you to a covered work, 359 | for which you have or can give appropriate copyright permission. 360 | 361 | Notwithstanding any other provision of this License, for material you 362 | add to a covered work, you may (if authorized by the copyright holders of 363 | that material) supplement the terms of this License with terms: 364 | 365 | a) Disclaiming warranty or limiting liability differently from the 366 | terms of sections 15 and 16 of this License; or 367 | 368 | b) Requiring preservation of specified reasonable legal notices or 369 | author attributions in that material or in the Appropriate Legal 370 | Notices displayed by works containing it; or 371 | 372 | c) Prohibiting misrepresentation of the origin of that material, or 373 | requiring that modified versions of such material be marked in 374 | reasonable ways as different from the original version; or 375 | 376 | d) Limiting the use for publicity purposes of names of licensors or 377 | authors of the material; or 378 | 379 | e) Declining to grant rights under trademark law for use of some 380 | trade names, trademarks, or service marks; or 381 | 382 | f) Requiring indemnification of licensors and authors of that 383 | material by anyone who conveys the material (or modified versions of 384 | it) with contractual assumptions of liability to the recipient, for 385 | any liability that these contractual assumptions directly impose on 386 | those licensors and authors. 387 | 388 | All other non-permissive additional terms are considered "further 389 | restrictions" within the meaning of section 10. If the Program as you 390 | received it, or any part of it, contains a notice stating that it is 391 | governed by this License along with a term that is a further 392 | restriction, you may remove that term. If a license document contains 393 | a further restriction but permits relicensing or conveying under this 394 | License, you may add to a covered work material governed by the terms 395 | of that license document, provided that the further restriction does 396 | not survive such relicensing or conveying. 397 | 398 | If you add terms to a covered work in accord with this section, you 399 | must place, in the relevant source files, a statement of the 400 | additional terms that apply to those files, or a notice indicating 401 | where to find the applicable terms. 402 | 403 | Additional terms, permissive or non-permissive, may be stated in the 404 | form of a separately written license, or stated as exceptions; 405 | the above requirements apply either way. 406 | 407 | 8. Termination. 408 | 409 | You may not propagate or modify a covered work except as expressly 410 | provided under this License. Any attempt otherwise to propagate or 411 | modify it is void, and will automatically terminate your rights under 412 | this License (including any patent licenses granted under the third 413 | paragraph of section 11). 414 | 415 | However, if you cease all violation of this License, then your 416 | license from a particular copyright holder is reinstated (a) 417 | provisionally, unless and until the copyright holder explicitly and 418 | finally terminates your license, and (b) permanently, if the copyright 419 | holder fails to notify you of the violation by some reasonable means 420 | prior to 60 days after the cessation. 421 | 422 | Moreover, your license from a particular copyright holder is 423 | reinstated permanently if the copyright holder notifies you of the 424 | violation by some reasonable means, this is the first time you have 425 | received notice of violation of this License (for any work) from that 426 | copyright holder, and you cure the violation prior to 30 days after 427 | your receipt of the notice. 428 | 429 | Termination of your rights under this section does not terminate the 430 | licenses of parties who have received copies or rights from you under 431 | this License. If your rights have been terminated and not permanently 432 | reinstated, you do not qualify to receive new licenses for the same 433 | material under section 10. 434 | 435 | 9. Acceptance Not Required for Having Copies. 436 | 437 | You are not required to accept this License in order to receive or 438 | run a copy of the Program. Ancillary propagation of a covered work 439 | occurring solely as a consequence of using peer-to-peer transmission 440 | to receive a copy likewise does not require acceptance. However, 441 | nothing other than this License grants you permission to propagate or 442 | modify any covered work. These actions infringe copyright if you do 443 | not accept this License. Therefore, by modifying or propagating a 444 | covered work, you indicate your acceptance of this License to do so. 445 | 446 | 10. Automatic Licensing of Downstream Recipients. 447 | 448 | Each time you convey a covered work, the recipient automatically 449 | receives a license from the original licensors, to run, modify and 450 | propagate that work, subject to this License. You are not responsible 451 | for enforcing compliance by third parties with this License. 452 | 453 | An "entity transaction" is a transaction transferring control of an 454 | organization, or substantially all assets of one, or subdividing an 455 | organization, or merging organizations. If propagation of a covered 456 | work results from an entity transaction, each party to that 457 | transaction who receives a copy of the work also receives whatever 458 | licenses to the work the party's predecessor in interest had or could 459 | give under the previous paragraph, plus a right to possession of the 460 | Corresponding Source of the work from the predecessor in interest, if 461 | the predecessor has it or can get it with reasonable efforts. 462 | 463 | You may not impose any further restrictions on the exercise of the 464 | rights granted or affirmed under this License. For example, you may 465 | not impose a license fee, royalty, or other charge for exercise of 466 | rights granted under this License, and you may not initiate litigation 467 | (including a cross-claim or counterclaim in a lawsuit) alleging that 468 | any patent claim is infringed by making, using, selling, offering for 469 | sale, or importing the Program or any portion of it. 470 | 471 | 11. Patents. 472 | 473 | A "contributor" is a copyright holder who authorizes use under this 474 | License of the Program or a work on which the Program is based. The 475 | work thus licensed is called the contributor's "contributor version". 476 | 477 | A contributor's "essential patent claims" are all patent claims 478 | owned or controlled by the contributor, whether already acquired or 479 | hereafter acquired, that would be infringed by some manner, permitted 480 | by this License, of making, using, or selling its contributor version, 481 | but do not include claims that would be infringed only as a 482 | consequence of further modification of the contributor version. For 483 | purposes of this definition, "control" includes the right to grant 484 | patent sublicenses in a manner consistent with the requirements of 485 | this License. 486 | 487 | Each contributor grants you a non-exclusive, worldwide, royalty-free 488 | patent license under the contributor's essential patent claims, to 489 | make, use, sell, offer for sale, import and otherwise run, modify and 490 | propagate the contents of its contributor version. 491 | 492 | In the following three paragraphs, a "patent license" is any express 493 | agreement or commitment, however denominated, not to enforce a patent 494 | (such as an express permission to practice a patent or covenant not to 495 | sue for patent infringement). To "grant" such a patent license to a 496 | party means to make such an agreement or commitment not to enforce a 497 | patent against the party. 498 | 499 | If you convey a covered work, knowingly relying on a patent license, 500 | and the Corresponding Source of the work is not available for anyone 501 | to copy, free of charge and under the terms of this License, through a 502 | publicly available network server or other readily accessible means, 503 | then you must either (1) cause the Corresponding Source to be so 504 | available, or (2) arrange to deprive yourself of the benefit of the 505 | patent license for this particular work, or (3) arrange, in a manner 506 | consistent with the requirements of this License, to extend the patent 507 | license to downstream recipients. "Knowingly relying" means you have 508 | actual knowledge that, but for the patent license, your conveying the 509 | covered work in a country, or your recipient's use of the covered work 510 | in a country, would infringe one or more identifiable patents in that 511 | country that you have reason to believe are valid. 512 | 513 | If, pursuant to or in connection with a single transaction or 514 | arrangement, you convey, or propagate by procuring conveyance of, a 515 | covered work, and grant a patent license to some of the parties 516 | receiving the covered work authorizing them to use, propagate, modify 517 | or convey a specific copy of the covered work, then the patent license 518 | you grant is automatically extended to all recipients of the covered 519 | work and works based on it. 520 | 521 | A patent license is "discriminatory" if it does not include within 522 | the scope of its coverage, prohibits the exercise of, or is 523 | conditioned on the non-exercise of one or more of the rights that are 524 | specifically granted under this License. You may not convey a covered 525 | work if you are a party to an arrangement with a third party that is 526 | in the business of distributing software, under which you make payment 527 | to the third party based on the extent of your activity of conveying 528 | the work, and under which the third party grants, to any of the 529 | parties who would receive the covered work from you, a discriminatory 530 | patent license (a) in connection with copies of the covered work 531 | conveyed by you (or copies made from those copies), or (b) primarily 532 | for and in connection with specific products or compilations that 533 | contain the covered work, unless you entered into that arrangement, 534 | or that patent license was granted, prior to 28 March 2007. 535 | 536 | Nothing in this License shall be construed as excluding or limiting 537 | any implied license or other defenses to infringement that may 538 | otherwise be available to you under applicable patent law. 539 | 540 | 12. No Surrender of Others' Freedom. 541 | 542 | If conditions are imposed on you (whether by court order, agreement or 543 | otherwise) that contradict the conditions of this License, they do not 544 | excuse you from the conditions of this License. If you cannot convey a 545 | covered work so as to satisfy simultaneously your obligations under this 546 | License and any other pertinent obligations, then as a consequence you may 547 | not convey it at all. For example, if you agree to terms that obligate you 548 | to collect a royalty for further conveying from those to whom you convey 549 | the Program, the only way you could satisfy both those terms and this 550 | License would be to refrain entirely from conveying the Program. 551 | 552 | 13. Use with the GNU Affero General Public License. 553 | 554 | Notwithstanding any other provision of this License, you have 555 | permission to link or combine any covered work with a work licensed 556 | under version 3 of the GNU Affero General Public License into a single 557 | combined work, and to convey the resulting work. The terms of this 558 | License will continue to apply to the part which is the covered work, 559 | but the special requirements of the GNU Affero General Public License, 560 | section 13, concerning interaction through a network will apply to the 561 | combination as such. 562 | 563 | 14. Revised Versions of this License. 564 | 565 | The Free Software Foundation may publish revised and/or new versions of 566 | the GNU General Public License from time to time. Such new versions will 567 | be similar in spirit to the present version, but may differ in detail to 568 | address new problems or concerns. 569 | 570 | Each version is given a distinguishing version number. If the 571 | Program specifies that a certain numbered version of the GNU General 572 | Public License "or any later version" applies to it, you have the 573 | option of following the terms and conditions either of that numbered 574 | version or of any later version published by the Free Software 575 | Foundation. If the Program does not specify a version number of the 576 | GNU General Public License, you may choose any version ever published 577 | by the Free Software Foundation. 578 | 579 | If the Program specifies that a proxy can decide which future 580 | versions of the GNU General Public License can be used, that proxy's 581 | public statement of acceptance of a version permanently authorizes you 582 | to choose that version for the Program. 583 | 584 | Later license versions may give you additional or different 585 | permissions. However, no additional obligations are imposed on any 586 | author or copyright holder as a result of your choosing to follow a 587 | later version. 588 | 589 | 15. Disclaimer of Warranty. 590 | 591 | THERE IS NO WARRANTY FOR THE PROGRAM, TO THE EXTENT PERMITTED BY 592 | APPLICABLE LAW. EXCEPT WHEN OTHERWISE STATED IN WRITING THE COPYRIGHT 593 | HOLDERS AND/OR OTHER PARTIES PROVIDE THE PROGRAM "AS IS" WITHOUT WARRANTY 594 | OF ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING, BUT NOT LIMITED TO, 595 | THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 596 | PURPOSE. THE ENTIRE RISK AS TO THE QUALITY AND PERFORMANCE OF THE PROGRAM 597 | IS WITH YOU. SHOULD THE PROGRAM PROVE DEFECTIVE, YOU ASSUME THE COST OF 598 | ALL NECESSARY SERVICING, REPAIR OR CORRECTION. 599 | 600 | 16. Limitation of Liability. 601 | 602 | IN NO EVENT UNLESS REQUIRED BY APPLICABLE LAW OR AGREED TO IN WRITING 603 | WILL ANY COPYRIGHT HOLDER, OR ANY OTHER PARTY WHO MODIFIES AND/OR CONVEYS 604 | THE PROGRAM AS PERMITTED ABOVE, BE LIABLE TO YOU FOR DAMAGES, INCLUDING ANY 605 | GENERAL, SPECIAL, INCIDENTAL OR CONSEQUENTIAL DAMAGES ARISING OUT OF THE 606 | USE OR INABILITY TO USE THE PROGRAM (INCLUDING BUT NOT LIMITED TO LOSS OF 607 | DATA OR DATA BEING RENDERED INACCURATE OR LOSSES SUSTAINED BY YOU OR THIRD 608 | PARTIES OR A FAILURE OF THE PROGRAM TO OPERATE WITH ANY OTHER PROGRAMS), 609 | EVEN IF SUCH HOLDER OR OTHER PARTY HAS BEEN ADVISED OF THE POSSIBILITY OF 610 | SUCH DAMAGES. 611 | 612 | 17. Interpretation of Sections 15 and 16. 613 | 614 | If the disclaimer of warranty and limitation of liability provided 615 | above cannot be given local legal effect according to their terms, 616 | reviewing courts shall apply local law that most closely approximates 617 | an absolute waiver of all civil liability in connection with the 618 | Program, unless a warranty or assumption of liability accompanies a 619 | copy of the Program in return for a fee. 620 | 621 | END OF TERMS AND CONDITIONS 622 | 623 | How to Apply These Terms to Your New Programs 624 | 625 | If you develop a new program, and you want it to be of the greatest 626 | possible use to the public, the best way to achieve this is to make it 627 | free software which everyone can redistribute and change under these terms. 628 | 629 | To do so, attach the following notices to the program. It is safest 630 | to attach them to the start of each source file to most effectively 631 | state the exclusion of warranty; and each file should have at least 632 | the "copyright" line and a pointer to where the full notice is found. 633 | 634 | 635 | Copyright (C) 636 | 637 | This program is free software: you can redistribute it and/or modify 638 | it under the terms of the GNU General Public License as published by 639 | the Free Software Foundation, either version 3 of the License, or 640 | (at your option) any later version. 641 | 642 | This program is distributed in the hope that it will be useful, 643 | but WITHOUT ANY WARRANTY; without even the implied warranty of 644 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 645 | GNU General Public License for more details. 646 | 647 | You should have received a copy of the GNU General Public License 648 | along with this program. If not, see . 649 | 650 | Also add information on how to contact you by electronic and paper mail. 651 | 652 | If the program does terminal interaction, make it output a short 653 | notice like this when it starts in an interactive mode: 654 | 655 | Copyright (C) 656 | This program comes with ABSOLUTELY NO WARRANTY; for details type `show w'. 657 | This is free software, and you are welcome to redistribute it 658 | under certain conditions; type `show c' for details. 659 | 660 | The hypothetical commands `show w' and `show c' should show the appropriate 661 | parts of the General Public License. Of course, your program's commands 662 | might be different; for a GUI interface, you would use an "about box". 663 | 664 | You should also get your employer (if you work as a programmer) or school, 665 | if any, to sign a "copyright disclaimer" for the program, if necessary. 666 | For more information on this, and how to apply and follow the GNU GPL, see 667 | . 668 | 669 | The GNU General Public License does not permit incorporating your program 670 | into proprietary programs. If your program is a subroutine library, you 671 | may consider it more useful to permit linking proprietary applications with 672 | the library. If this is what you want to do, use the GNU Lesser General 673 | Public License instead of this License. But first, please read 674 | . 675 | -------------------------------------------------------------------------------- /Procfile: -------------------------------------------------------------------------------- 1 | worker: python src/bot.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- 1 | # Approxima Chatbot (Telegram) 2 | 3 | Chatbot de Telegram que será desenvolvido como MVP (Minimum Valueable Product) da iniciativa Approxima 4 | 5 | ## Acabou de clonar o repositório? 6 | 7 | Rode o comando `pip install -r requirements.txt` para instalar as libs necessárias. 8 | 9 | ## Lista de comandos (@approxima_bot) 10 | 11 | - /start => (Re)inicia o bot. Se a pessoa não estiver cadastrada na base de dados, pede para ela fornecer um nome, uma pequena descrição pessoal e sugere a ela escolher seus primeiros interesses. 12 | 13 | - /prefs => Retorna lista de interesses (caixa de seleção). A pessoa pode marcar ou desmarcar o que ela quiser. O que ela marcar aqui será utilizado pelo algoritmo de rankeamento para encontrar as pessoas mais similares à ela. Até o presente momento, não há a intenção de mostrar os interesses marcados por uma pessoa às outras. 14 | 15 | - /show => Mostra a descrição da pessoa mais similar ao usuário, com base nos interesses, e duas opções: "conectar" e "agora não". 16 | 17 | - /random => Mostra a descrição de uma pessoa aleatória e duas opções: "conectar" e "agora não". 18 | 19 | - [POSSIVEL FEATURE] /opposite => Mostra a descrição de uma pessoa que tem interesses opostos (vai com base no ranking reverso) e duas opções: "conectar" e "agora não". 20 | 21 | - /clear => Permite que as pessoas que o usuário respondeu com "agora não" apareçam de novo nas sugestões dele (quando ele responde com "agora não", aquele usuário vai para o campo de "rejeitados", então não irá aparecer como sugestão novamente AO MENOS que ele dê esse comando). 22 | 23 | - /pending => Pega a primeira solicitação de conexão da fila de não-respondidas, mostrando a descrição da pessoa e dois botões: "aceitar" ou "rejeitar". 24 | 25 | - /friends => Mostra o nome, a descrição e o contato (@ do Telegram) de todas as pessoas com que o usuário já se conectou. 26 | 27 | - /name => Troca o nome do usuário. 28 | 29 | - /desc => Troca a descrição do usuário. 30 | 31 | - /help => Mostra os comandos disponíveis. 32 | -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | certifi==2020.6.20 2 | cffi==1.14.1 3 | click==7.1.2 4 | cryptography==3.0 5 | decorator==4.4.2 6 | dnspython==1.16.0 7 | future==0.18.2 8 | idna==2.10 9 | numpy==1.19.1 10 | pycparser==2.20 11 | pymongo==3.11.0 12 | PySocks==1.7.1 13 | python-dotenv==0.14.0 14 | python-telegram-bot==12.8 15 | six==1.15.0 16 | tornado==6.0.4 17 | ujson==3.0.0 18 | pytz -------------------------------------------------------------------------------- /src/.gitignore: -------------------------------------------------------------------------------- 1 | db_actions/ 2 | tests/ 3 | .editorconfig -------------------------------------------------------------------------------- /src/bot.py: -------------------------------------------------------------------------------- 1 | # This Python file uses the following encoding: utf-8 2 | 3 | import os 4 | import logging 5 | import random 6 | import ranker 7 | import json 8 | import numpy as np 9 | from dbwrapper import Database 10 | from telegram import InlineKeyboardMarkup, InlineKeyboardButton 11 | from telegram.ext import Updater, Filters, CommandHandler, MessageHandler, ConversationHandler, CallbackQueryHandler 12 | from categories import CATEGORIES 13 | from utils import unique_list, make_buttons, correct_friends_order 14 | from timeit import default_timer as timer 15 | 16 | # ================================== ENV ======================================= 17 | 18 | 19 | if not os.getenv("IS_PRODUCTION"): 20 | from dotenv import load_dotenv 21 | load_dotenv() 22 | 23 | TELEGRAM_TOKEN = os.getenv("TELEGRAM_TOKEN") 24 | CONNECTION_STRING = os.getenv("CONNECTION_STRING") 25 | ADMINS = json.loads(os.getenv("ADMINS")) 26 | 27 | # Enable logging 28 | logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', 29 | level=logging.INFO) 30 | 31 | logger = logging.getLogger(__name__) 32 | 33 | # ================================== BD ======================================== 34 | 35 | 36 | is_production = False if os.getenv("IS_PRODUCTION") is None else True 37 | db = Database(CONNECTION_STRING, is_production=is_production) 38 | 39 | 40 | # ================================== BOT ======================================= 41 | 42 | 43 | # States 44 | LIMBO, REGISTER_NAME, REGISTER_BIO, CHOOSE_ACTION, CHOOSE_ANSWER_FOR_BUTTONS, GIVE_NEW_NAME, GIVE_NEW_BIO, SEND_NOTIFICATION = range( 45 | 8) 46 | # States for interests and friends conversations 47 | CHOOSE_INTERESTS, CHOOSE_PAGE = range(8, 10) 48 | 49 | 50 | # ================================= HELP ======================================= 51 | 52 | 53 | def help_command(update, context): 54 | ''' 55 | Mostra os comandos disponiveis 56 | ''' 57 | db.register_action('help_command', update.effective_user.id) 58 | 59 | text = "/prefs --> Retorna uma lista com todas as categorias de interesse. A partir dela, você poderá adicionar ou remover interesses.\n" 60 | text += "/show --> Mostra uma pessoa que tem interesses em comum.\n" 61 | text += "/random --> Mostra uma pessoa aleatória.\n" 62 | text += "/clear --> Permite que as pessoas que você respondeu com \"Agora não\" apareçam de novo nos dois comandos acima.\n" 63 | text += "/pending --> Mostra uma solicitação de conexão que você possui e ainda não respondeu.\n" 64 | text += "/friends --> Mostra o contato de todas as pessoas com que você já se conectou.\n" 65 | text += "/name --> Troca o seu nome.\n" 66 | text += "/desc --> Troca a sua descrição.\n" 67 | text += "/help --> Mostra novamente essa lista. Alternativamente, você pode digitar \"/\" e a lista de comandos também aparecerá!\n\n" 68 | text += "Caso tenha algum problema ou crítica/sugestão, chama um dos meus desenvolvedores (eles me disseram que não mordem) --> @vitorsanc @Lui_Tombo @arenasoy @Angra018 @OliveiraNelson" 69 | update.message.reply_text(text) 70 | 71 | return CHOOSE_ACTION 72 | 73 | 74 | # ================================= START ====================================== 75 | 76 | 77 | def ask_for_username(update, context): 78 | response = "Por favor, siga os passos dados no comando /start e defina o seu Nome de Usuário do Telegram para continuar." 79 | update.message.reply_text(response) 80 | return LIMBO 81 | 82 | 83 | def start_command(update, context): 84 | ''' 85 | start => Inicia o bot. Se a pessoa não estiver cadastrada na base de dados 86 | (dá pra ver pelo ID do Tele), pede para ela fornecer: 87 | um nome, uma pequena descrição pessoal e, por último para escolher seus interesses iniciais. 88 | ''' 89 | 90 | # facilita na hora de referenciar esse usuario 91 | myself = update.effective_user.id 92 | my_data = db.get_user_by_id(myself) 93 | 94 | db.register_action('start_command', myself) 95 | 96 | if my_data is not None: 97 | # Pega os dados dele do BD 98 | context.user_data['chat_id'] = my_data['chat_id'] # number 99 | context.user_data['username'] = my_data['username'] # string 100 | context.user_data['name'] = my_data['name'] # string 101 | context.user_data['bio'] = my_data['bio'] # string 102 | context.user_data['interests'] = my_data['interests'] 103 | context.user_data['rejects'] = my_data['rejects'] 104 | context.user_data['invited'] = my_data['invited'] 105 | context.user_data['pending'] = my_data['pending'] 106 | context.user_data['connections'] = my_data['connections'] 107 | 108 | # Manda a mensagem de "boas-vindas" 109 | message = "É muito bom ter você de volta! Bora começar a usar o Approxima :)\n" 110 | message += "Me diz: o que você quer fazer agora?\n\n" 111 | message += "Use /help para uma lista dos comandos disponíveis.\n" 112 | 113 | update.message.reply_text(message) 114 | 115 | return CHOOSE_ACTION 116 | 117 | # Se chegou aqui é novo usuario e deve se registrar 118 | 119 | if update.effective_user.name[0] != '@': 120 | # User does not have username 121 | message = "Parece que você não possui um Nome de Usuário do Telegram ainda :(\n" 122 | message += "Infelizmente, eu não posso completar o seu registro se você não tiver um, pois será a única forma dos outros usuários entrarem em contato com você.\n\n" 123 | message += "Caso queira criar um, basta seguir esses passos (é super simples):\n" 124 | message += "\t1: Vá na parte de Configurações (Settings) do Telegram;\n" 125 | message += "\t2: É só preencher o campo Nome de Usuário (Username);\n" 126 | message += "\t3: Assim que tiver com tudo certinho, me dê o comando /start.\n" 127 | 128 | update.message.reply_text(message) 129 | 130 | return LIMBO 131 | 132 | # Crio os campos necessarios para o user context 133 | context.user_data['chat_id'] = update.effective_chat.id 134 | context.user_data['username'] = update.effective_user.name 135 | context.user_data['name'] = '' # string 136 | context.user_data['bio'] = '' # string 137 | context.user_data['interests'] = [] 138 | context.user_data['rejects'] = [] 139 | context.user_data['invited'] = [] 140 | context.user_data['pending'] = [] 141 | context.user_data['connections'] = [] 142 | 143 | message = "Muito prazer! Vamos começar o seu registro no Approxima!" 144 | 145 | update.message.reply_text(message) 146 | 147 | message = "Primeiro, me forneça o seu nome.\n" 148 | message += "Ex: João Vitor dos Santos" 149 | 150 | update.message.reply_text(message) 151 | 152 | return REGISTER_NAME 153 | 154 | 155 | # ================================ REGISTER ==================================== 156 | 157 | 158 | def register_name(update, context): 159 | response = "Legal! Agora, me conte um pouco mais sobre seus gostos... faça uma pequena descrição de si mesmo.\n" 160 | response += "Ela será utilizada para apresentar você para os outros usuários do Approxima (não mostrarei o seu nome).\n\n" 161 | response += "OBS: Você poderá mudar essa descrição depois, mas lembre-se de que somente ela irá aparecer para os outros usuários quando formos te apresentar a eles!" 162 | 163 | context.user_data['name'] = update.message.text 164 | 165 | update.message.reply_text(response) 166 | 167 | return REGISTER_BIO 168 | 169 | 170 | def register_bio(update, context): 171 | 172 | # facilita na hora de referenciar esse usuario 173 | myself = update.effective_user.id 174 | 175 | response = "Boa! Agora só falta você adicionar alguns interesses para começar a usar o Approxima!\n" 176 | response += "Clique (ou toque) aqui --> /prefs\n\n" 177 | response += "Após finalizada a etapa acima você já poderá começar a usar os meus comandos!\n" 178 | response += "Caso se sinta perdido em algum momento, lembre-se que existe o comando /help para te ajudar ;)" 179 | 180 | context.user_data['bio'] = update.message.text 181 | 182 | # Joga as informacoes no BD 183 | db.insert_user(myself, context.user_data) 184 | 185 | # Loga que um novo usuario foi registrado 186 | logger.info( 187 | f"User {update.effective_user.name} has been registered in the database.") 188 | logger.info( 189 | f'{update.effective_user.name} (id: {update.effective_user.id}) data: {context.user_data}') 190 | 191 | update.message.reply_text(response) 192 | 193 | return CHOOSE_ACTION 194 | 195 | 196 | # ================================== EDIT ====================================== 197 | 198 | 199 | def edit_name_command(update, context): 200 | response = f"Seu nome atual é: {context.user_data['name']}\n\n" 201 | response += "Agora, manda pra mim o seu novo nome! Envie um ponto (.) caso tenha desistido de mudá-lo." 202 | update.message.reply_text(response) 203 | 204 | return GIVE_NEW_NAME 205 | 206 | 207 | def update_name(update, context): 208 | # facilita na hora de referenciar esse usuario 209 | myself = update.effective_user.id 210 | 211 | if update.message.text == ".": 212 | db.register_action('edit_name_command', myself, 213 | additional_data={'changed': False}) 214 | update.message.reply_text("Ok! Não vou alterar seu nome.") 215 | return CHOOSE_ACTION 216 | 217 | context.user_data['name'] = update.message.text 218 | db.update_user_by_id(myself, {'name': update.message.text}) 219 | db.register_action('edit_name_command', myself, 220 | additional_data={'changed': True, 'new_name': update.message.text}) 221 | 222 | update.message.reply_text("Seu nome foi alterado com sucesso!") 223 | 224 | return CHOOSE_ACTION 225 | 226 | 227 | def edit_bio_command(update, context): 228 | response = "Sua descrição atual é:\n\n" 229 | response += f"{context.user_data['bio']}\n\n" 230 | response += "Agora, manda pra mim a sua nova descrição! Envie um ponto (.) caso tenha desistido de mudá-la." 231 | update.message.reply_text(response) 232 | 233 | return GIVE_NEW_BIO 234 | 235 | 236 | def update_bio(update, context): 237 | # facilita na hora de referenciar esse usuario 238 | myself = update.effective_user.id 239 | 240 | if update.message.text == ".": 241 | db.register_action('edit_desc_command', myself, 242 | additional_data={'changed': False}) 243 | update.message.reply_text("Ok! Não vou alterar sua descrição.") 244 | return CHOOSE_ACTION 245 | 246 | context.user_data['bio'] = update.message.text 247 | db.update_user_by_id(myself, {'bio': update.message.text}) 248 | db.register_action('edit_desc_command', myself, 249 | additional_data={'changed': True, 'new_desc': update.message.text}) 250 | 251 | update.message.reply_text("Sua descrição foi alterada com sucesso!") 252 | 253 | return CHOOSE_ACTION 254 | 255 | 256 | # ================================== PREFS ===================================== 257 | 258 | def build_prefs_keyboard(my_cats, sub_menu=''): 259 | keyboard = [] 260 | 261 | if not sub_menu: 262 | categories_to_show = CATEGORIES 263 | else: 264 | categories_to_show = CATEGORIES[sub_menu][1] 265 | 266 | for category in categories_to_show: 267 | if sub_menu: 268 | category_id = str(CATEGORIES[sub_menu][0]) + \ 269 | "," + str(categories_to_show[category][0]) 270 | else: 271 | category_id = str(categories_to_show[category][0]) 272 | 273 | category_sub_menu_text = '|sub' + sub_menu if sub_menu else '' 274 | 275 | if not sub_menu and isinstance(CATEGORIES[category][1], dict) and len(CATEGORIES[category][1].keys()) > 0: 276 | category_text = category + " ⬊" 277 | callback_text = "open" + category 278 | elif category_id in my_cats: 279 | category_text = "✅ " + category 280 | callback_text = "toggle" + category_id + category_sub_menu_text 281 | else: 282 | category_text = category 283 | callback_text = "toggle" + category_id + category_sub_menu_text 284 | keyboard.append([ 285 | InlineKeyboardButton(category_text, callback_data=callback_text) 286 | ]) 287 | 288 | if sub_menu: 289 | keyboard.append( 290 | [InlineKeyboardButton("⬉ VOLTAR", callback_data="goback")] 291 | ) 292 | 293 | keyboard.append( 294 | [InlineKeyboardButton("❰ ENVIAR ❱", callback_data="finish")] 295 | ) 296 | 297 | return keyboard 298 | 299 | 300 | def prefs_command(update, context): 301 | ''' 302 | prefs => Retorna lista de interesses (caixa de seleção). A pessoa pode marcar 303 | ou desmarcar o que ela quiser. 304 | ''' 305 | 306 | response = "Escolha suas categorias de interesse.\n" 307 | response += "Utilizaremos elas para te recomendar pessoas que tenham gostos parecidos com os seus.\n" 308 | response += "O que você marcar aqui NÃO SERÁ VISÍVEL para nenhum outro usuário além de você mesmo!\n" 309 | 310 | my_cats = context.user_data['interests'] 311 | 312 | keyboard = build_prefs_keyboard(my_cats) 313 | 314 | update.message.reply_text( 315 | response, reply_markup=InlineKeyboardMarkup(keyboard)) 316 | 317 | return CHOOSE_INTERESTS 318 | 319 | 320 | def open_category_state(update, context): 321 | update.callback_query.answer() # await for answer 322 | 323 | my_cats = context.user_data['interests'] 324 | 325 | # Trata a resposta anterior 326 | category = update.callback_query.data[4:] 327 | 328 | # Constroi o novo teclado 329 | keyboard = build_prefs_keyboard(my_cats, category) 330 | 331 | update.callback_query.edit_message_reply_markup( 332 | reply_markup=InlineKeyboardMarkup(keyboard)) 333 | 334 | return CHOOSE_INTERESTS 335 | 336 | 337 | def back_to_all_categories_state(update, context): 338 | update.callback_query.answer() # await for answer 339 | 340 | my_cats = context.user_data['interests'] 341 | 342 | # Constroi o novo teclado 343 | keyboard = build_prefs_keyboard(my_cats) 344 | 345 | update.callback_query.edit_message_reply_markup( 346 | reply_markup=InlineKeyboardMarkup(keyboard)) 347 | 348 | return CHOOSE_INTERESTS 349 | 350 | 351 | def change_category_state(update, context): 352 | update.callback_query.answer() # await for answer 353 | 354 | my_cats = context.user_data['interests'] 355 | 356 | # Trata a resposta anterior 357 | sub_category = '' 358 | category_id = update.callback_query.data[6:] 359 | if '|sub' in category_id: 360 | sub_index = category_id.index('|sub') 361 | sub_category = category_id[sub_index + 4:] 362 | category_id = category_id[:sub_index] 363 | if category_id in my_cats: 364 | my_cats.remove(category_id) 365 | else: 366 | my_cats.append(category_id) 367 | 368 | # Constroi o novo teclado 369 | keyboard = build_prefs_keyboard(my_cats, sub_category) 370 | 371 | update.callback_query.edit_message_reply_markup( 372 | reply_markup=InlineKeyboardMarkup(keyboard)) 373 | 374 | return CHOOSE_INTERESTS 375 | 376 | 377 | def submit_selection(update, context): 378 | update.callback_query.answer() # await for answer 379 | 380 | # facilita na hora de referenciar esse usuario 381 | myself = update.effective_user.id 382 | 383 | # Guarda as informacoes no BD 384 | db.update_user_by_id( 385 | myself, {'interests': context.user_data['interests']}) 386 | 387 | db.register_action('edit_interests_command', myself) 388 | 389 | update.effective_message.reply_text('Seus interesses foram atualizados!') 390 | return ConversationHandler.END 391 | 392 | 393 | # ================================== SHOW ====================================== 394 | 395 | 396 | def show_person_command(update, context): 397 | ''' 398 | show => Mostra uma pessoa que tem interesses em comum (vai com base no ranking). 399 | Embaixo, um botão para enviar a solicitação de conexão deve existir, 400 | bem como um botão de "agora não". 401 | ''' 402 | 403 | # facilita na hora de referenciar esse usuario 404 | myself = update.effective_user.id 405 | 406 | my_data = db.get_user_by_id(myself) 407 | 408 | context.user_data['pending'] = my_data['pending'] 409 | context.user_data['connections'] = my_data['connections'] 410 | context.user_data['rejects'] = my_data['rejects'] 411 | context.user_data['invited'] = my_data['invited'] 412 | 413 | # get all users (IDs) from the DB 414 | all_users = np.array(db.list_user_ids(), dtype=np.uint32) 415 | 416 | not_allowed_users = np.hstack( 417 | ( 418 | [myself], 419 | context.user_data['pending'], 420 | context.user_data['invited'], 421 | context.user_data['connections'], 422 | context.user_data['rejects'] 423 | ) 424 | ) 425 | 426 | # Usuarios que podem aparecer para mim, de acordo com os dados do meu perfil 427 | allowed_users = np.setdiff1d( 428 | all_users, not_allowed_users, assume_unique=True 429 | ) 430 | 431 | if len(all_users) == len(not_allowed_users): 432 | update.message.reply_text( 433 | 'Não tenho ninguém novo para te mostrar no momento... que tal tentar amanhã? :)') 434 | return CHOOSE_ACTION 435 | 436 | # LEMBRAR QUE, A PARTIR DAQUI, TODOS OS USERS SÃO np.uint32 E NÃO int, 437 | # portanto o casting se faz necessario 438 | allowed_users = allowed_users.astype(int) 439 | 440 | # Pega as informacoes de todos os usuarios na DB 441 | allowed_users_data = db.get_users_in_list(allowed_users.tolist()) 442 | 443 | # Mapeia os usuarios aos seus interesses e posicao no vetor de allowed users 444 | map_users = {} 445 | for i, user_data in enumerate(allowed_users_data): 446 | if myself not in user_data['rejects']: 447 | map_users[user_data['_id']] = { 448 | "interests": user_data['interests'], 449 | "original_pos": i 450 | } 451 | 452 | target = ranker.rank( 453 | context.user_data['interests'].copy(), map_users) 454 | 455 | db.register_action('show_person_command', myself) 456 | 457 | if target is None: 458 | # Nao ha ninguem com as preferencias do usuario ainda 459 | response = "Parece que não há ninguém com os mesmos gostos que você no sistema ainda...\n\n" 460 | response += "Você pode tentar:\n" 461 | response += "- Marcar mais categorias de interesse\n" 462 | response += "- O comando /random (pessoa aleatória)" 463 | 464 | update.message.reply_text(response) 465 | 466 | return CHOOSE_ACTION 467 | 468 | # Daqui para frente, sabemos que uma pessoa similar existe 469 | target_index = map_users[target]['original_pos'] 470 | target_bio = allowed_users_data[target_index].get('bio') 471 | 472 | # Avisa no contexto que essa pessoa foi a ultima a ser exibida para o usuario (ajuda nas callback queries) 473 | context.user_data['lastShownId'] = target 474 | 475 | # MENSAGEM DO BOT 476 | 477 | keyboard = [[ 478 | InlineKeyboardButton('Conectar', callback_data='connect'), 479 | InlineKeyboardButton('Agora não', callback_data='dismiss') 480 | ]] 481 | 482 | text = f'\"{target_bio}\"' 483 | 484 | update.message.reply_text( 485 | text, reply_markup=InlineKeyboardMarkup(keyboard)) 486 | 487 | return CHOOSE_ANSWER_FOR_BUTTONS 488 | 489 | 490 | # ================================== RANDOM ==================================== 491 | 492 | 493 | def get_random_person_command(update, context): 494 | ''' 495 | random => Mostra uma pessoa aleatória. Embaixo, um botão para enviar a solicitação 496 | de conexão deve existir, bem como um botão de "agora não". 497 | ''' 498 | 499 | # facilita na hora de referenciar esse usuario 500 | myself = update.effective_user.id 501 | 502 | my_data = db.get_user_by_id(myself) 503 | 504 | context.user_data['pending'] = my_data['pending'] 505 | context.user_data['connections'] = my_data['connections'] 506 | context.user_data['rejects'] = my_data['rejects'] 507 | context.user_data['invited'] = my_data['invited'] 508 | 509 | # get all users (IDs) from the DB 510 | all_users = np.array(db.list_user_ids(), dtype=np.uint32) 511 | 512 | not_allowed_users = np.hstack( 513 | ( 514 | [myself], 515 | context.user_data['pending'], 516 | context.user_data['invited'], 517 | context.user_data['connections'], 518 | context.user_data['rejects'] 519 | ) 520 | ) 521 | 522 | # Usuarios que podem aparecer para mim, de acordo com os dados do meu perfil 523 | allowed_users = np.setdiff1d( 524 | all_users, not_allowed_users, assume_unique=True 525 | ) 526 | 527 | # LEMBRAR QUE, A PARTIR DAQUI, TODOS OS USERS SÃO np.uint32 E NÃO int, 528 | # portanto o casting se faz necessario 529 | allowed_users = allowed_users.astype(int) 530 | 531 | # Pega as informacoes de todos os usuarios na DB 532 | allowed_users_data = db.get_users_in_list(allowed_users.tolist()) 533 | 534 | # Preciso, ainda, tirar aqueles que me tem em sua lista de rejects 535 | remove_index = [] 536 | 537 | for i, user_data in enumerate(allowed_users_data): 538 | if myself in user_data.get('rejects'): 539 | remove_index.append(i) 540 | 541 | allowed_users = np.delete(allowed_users, remove_index) 542 | 543 | db.register_action('random_person_command', myself) 544 | 545 | if len(allowed_users) == 0: 546 | update.message.reply_text( 547 | 'Não tenho ninguém novo para te mostrar no momento... que tal tentar amanhã? :)') 548 | return CHOOSE_ACTION 549 | 550 | target = int(random.choice(allowed_users)) 551 | 552 | for user_data in allowed_users_data: 553 | if target == user_data['_id']: 554 | target_bio = user_data['bio'] 555 | break 556 | 557 | # Avisa no contexto que essa pessoa foi a ultima a ser exibida para o usuario (ajuda nas callback queries) 558 | context.user_data['lastShownId'] = target 559 | 560 | # MENSAGEM DO BOT 561 | 562 | keyboard = [[ 563 | InlineKeyboardButton('Conectar', callback_data='connect'), 564 | InlineKeyboardButton('Agora não', callback_data='dismiss') 565 | ]] 566 | 567 | text = f'\"{target_bio}\"' 568 | 569 | update.message.reply_text( 570 | text, reply_markup=InlineKeyboardMarkup(keyboard)) 571 | 572 | return CHOOSE_ANSWER_FOR_BUTTONS 573 | 574 | 575 | def handle_invite_answer(update, context): 576 | target_id = context.user_data['lastShownId'] 577 | del context.user_data['lastShownId'] 578 | 579 | # facilita na hora de referenciar esse usuario 580 | myself = update.effective_user.id 581 | 582 | update.callback_query.answer() # awaits for answer 583 | answer = update.callback_query.data 584 | 585 | db.register_action('answered_suggestion', myself, 586 | additional_data={'answer': answer}) 587 | 588 | if answer == 'dismiss': 589 | context.user_data['rejects'].append(target_id) 590 | 591 | # Saves in DB 592 | db.update_user_by_id( 593 | myself, {'rejects': context.user_data['rejects']}) 594 | 595 | context.bot.sendMessage(chat_id=context.user_data['chat_id'], 596 | text='Sugestão rejeitada.') 597 | 598 | return CHOOSE_ACTION 599 | 600 | # For now on, we know that the answer is "connect"! 601 | 602 | context.user_data['invited'].append(target_id) 603 | 604 | # Update my info on BD 605 | db.update_user_by_id( 606 | myself, {'invited': context.user_data['invited']}) 607 | 608 | # Now, let's update info from the target user 609 | target_data = db.get_user_by_id(target_id) 610 | 611 | target_data['pending'].append(myself) 612 | 613 | db.update_user_by_id( 614 | target_id, {'pending': target_data['pending']}) 615 | 616 | # Send messages confirming the action 617 | target_msg = "Você recebeu uma nova solicitação de conexão!\n" 618 | target_msg += "Utilize o comando /pending para vê-la." 619 | 620 | target_chat = target_data['chat_id'] 621 | context.bot.sendMessage(chat_id=target_chat, 622 | text=target_msg) 623 | 624 | context.bot.sendMessage(chat_id=context.user_data['chat_id'], 625 | text='Solicitação enviada.') 626 | 627 | return CHOOSE_ACTION 628 | 629 | 630 | # ================================= CLEAR ====================================== 631 | 632 | 633 | def clear_rejected_command(update, context): 634 | ''' 635 | Deleta o array de pessoas que o usuario já rejeitou, 636 | permitindo que elas apareçam novamente nas buscas 637 | ''' 638 | 639 | # facilita na hora de referenciar esse usuario 640 | myself = update.effective_user.id 641 | 642 | db.register_action('clear_rejects_command', myself) 643 | 644 | if len(context.user_data['rejects']) == 0: 645 | update.message.reply_text( 646 | 'Você não \"rejeitou\" ninguém por enquanto.') 647 | return CHOOSE_ACTION 648 | 649 | context.user_data['rejects'] = [] 650 | db.update_user_by_id(myself, {'rejects': []}) 651 | 652 | update.message.reply_text( 653 | 'Tudo certo! Sua lista de \"rejeitados\" foi limpa!') 654 | 655 | return CHOOSE_ACTION 656 | 657 | 658 | # ================================ PENDING ==================================== 659 | 660 | 661 | def pending_command(update, context): 662 | ''' 663 | pending => Mostra todas as solicitações de conexão que aquela pessoa possui e 664 | para as quais ela ainda não deu uma resposta. Mostra, para cada solicitação, 665 | a descrição da pessoa e dois botões: conectar ou descartar). 666 | ''' 667 | 668 | # facilita na hora de referenciar esse usuario 669 | myself = update.effective_user.id 670 | 671 | my_data = db.get_user_by_id(myself) 672 | 673 | context.user_data['pending'] = my_data['pending'] 674 | 675 | db.register_action('pending_command', myself) 676 | 677 | if len(context.user_data['pending']) == 0: 678 | update.message.reply_text( 679 | 'Você não possui novas solicitações de conexão.') 680 | return CHOOSE_ACTION 681 | 682 | # Pego o primeiro elemento na "fila" 683 | target = context.user_data['pending'].pop(0) 684 | 685 | target_data = db.get_user_by_id(target) 686 | target_bio = target_data.get('bio') 687 | 688 | # Avisa no contexto que essa pessoa foi a ultima a ser exibida para o usuario (ajuda nas callback queries) 689 | context.user_data['lastShownId'] = target 690 | 691 | # Salvo no BD o novo array de 'pending' 692 | db.update_user_by_id( 693 | myself, {'pending': context.user_data['pending']}) 694 | 695 | # Me retiro da lista de "invited" do outro usuario 696 | target_invited = target_data.get('invited') 697 | target_invited.remove(myself) 698 | db.update_user_by_id(target, {'invited': target_invited}) 699 | 700 | # MENSAGEM DO BOT 701 | 702 | keyboard = [[ 703 | InlineKeyboardButton('Aceitar', callback_data='accept'), 704 | InlineKeyboardButton('Rejeitar', callback_data='reject') 705 | ]] 706 | 707 | text = "A seguinte pessoa quer se conectar a você:\n\n" 708 | text += f'\"{target_bio}\"' 709 | 710 | update.message.reply_text( 711 | text, reply_markup=InlineKeyboardMarkup(keyboard)) 712 | 713 | return CHOOSE_ANSWER_FOR_BUTTONS 714 | 715 | 716 | def handle_pending_answer(update, context): 717 | target_id = context.user_data['lastShownId'] 718 | del context.user_data['lastShownId'] 719 | 720 | # facilita na hora de referenciar esse usuario 721 | myself = update.effective_user.id 722 | 723 | update.callback_query.answer() # awaits for answer 724 | answer = update.callback_query.data 725 | 726 | db.register_action('answered_pending', myself, 727 | additional_data={'answer': answer}) 728 | 729 | if answer == 'reject': 730 | context.user_data['rejects'].append(target_id) 731 | 732 | # Saves in DB 733 | db.update_user_by_id( 734 | myself, {'rejects': context.user_data['rejects']}) 735 | 736 | context.bot.sendMessage(chat_id=context.user_data['chat_id'], 737 | text='Pedido de conexão rejeitado.') 738 | 739 | return CHOOSE_ACTION 740 | 741 | # For now on, we know that the answer is "accept"! 742 | 743 | # Register the new connection 744 | context.user_data['connections'].append(target_id) 745 | context.user_data['pending'] 746 | 747 | # Update my info on BD 748 | db.update_user_by_id( 749 | myself, {'connections': context.user_data['connections']}) 750 | 751 | # Update their info on BD 752 | 753 | target_data = db.get_user_by_id(target_id) 754 | 755 | target_data['connections'].append(myself) 756 | 757 | db.update_user_by_id( 758 | target_id, {'connections': target_data['connections']}) 759 | 760 | # Send messages confirming the action 761 | 762 | target_chat = target_data['chat_id'] 763 | 764 | text_target = 'Uma pessoa acaba de aceitar seu pedido de conexão! Use o comando /friends para checar.\n' 765 | text_target += 'Ela estará no final da última página.' 766 | 767 | context.bot.sendMessage(chat_id=target_chat, 768 | text=text_target) 769 | 770 | my_text = 'Parabéns! Você acaba de ganhar uma nova conexão! Que tal dar um \"oi\" pra elu? :)\n' 771 | my_text += "Use o comando /friends para ver a sua nova conexão! Ela estará no final da última página." 772 | 773 | context.bot.sendMessage(chat_id=context.user_data['chat_id'], 774 | text=my_text) 775 | 776 | return CHOOSE_ACTION 777 | 778 | 779 | # ================================ FRIENDS ===================================== 780 | 781 | 782 | def friends_paginator(connections): 783 | resulting_pages = [] 784 | 785 | divider = "\n\n" 786 | divider += "=" * 32 787 | divider += "\n\n" 788 | 789 | msg_limit = 1300 # Limit beatifully crafted by hand 790 | 791 | cur_page_text = '' 792 | 793 | connections_info = db.get_users_in_list(connections) 794 | 795 | connections_info = correct_friends_order(connections_info, connections) 796 | 797 | # Adding friends info to the message 798 | for user_info in connections_info: 799 | # Format their info on a string 800 | user_info_txt = f"{user_info['name']}\n" 801 | user_info_txt += f"{user_info['username']}\n\n" 802 | user_info_txt += f"\"{user_info['bio']}\"" 803 | 804 | # If user_info_txt is greater than the limit (+ the divider), TRUNCATE IT! 805 | if len(user_info_txt) > msg_limit - len(divider): 806 | user_info_txt = user_info_txt[:msg_limit - 3] + '...' 807 | 808 | user_info_txt += divider 809 | 810 | # If adding one more user is gonna break the msg limit 811 | if len(cur_page_text) + len(user_info_txt) > msg_limit: 812 | resulting_pages.append(cur_page_text) 813 | cur_page_text = '' 814 | 815 | cur_page_text += user_info_txt 816 | 817 | resulting_pages.append(cur_page_text) 818 | 819 | return resulting_pages 820 | 821 | 822 | def friends_command(update, context): 823 | ''' 824 | friends => Mostra o contato (@ do Tele) de todas as pessoas com que o usuário 825 | já se conectou. 826 | ''' 827 | 828 | start_t = timer() # When want to store how long this function takes to complete 829 | 830 | # facilita na hora de referenciar esse usuario 831 | myself = update.effective_user.id 832 | 833 | my_data = db.get_user_by_id(myself) 834 | 835 | context.user_data['connections'] = my_data['connections'] 836 | 837 | if len(context.user_data['connections']) == 0: 838 | # Este usuario ainda nao tem conexoes 839 | response = "Você ainda não possui nenhuma conexão!\n" 840 | response += "Que tal usar o comando /show para conhecer alguém novo?" 841 | 842 | update.message.reply_text(response) 843 | 844 | return ConversationHandler.END 845 | 846 | # Se chegou ate aqui é porque ele tem conexoes 847 | 848 | connections_set = unique_list(context.user_data['connections']) 849 | 850 | # Corrige as suas conexoes caso hajam repetições 851 | if len(connections_set) < len(context.user_data['connections']): 852 | # Existem repeticoes no original 853 | context.user_data['connections'] = list(connections_set) 854 | db.update_user_by_id( 855 | myself, {'connections': context.user_data['connections']}) 856 | 857 | bottom_msg = "Utilize esses botões para navegar entre as páginas:\n\n" 858 | 859 | pages_text_list = friends_paginator(connections_set) 860 | context.user_data['friend_pages'] = pages_text_list 861 | 862 | button_pairs = make_buttons(0, len(pages_text_list) - 1) 863 | 864 | response = pages_text_list[0] 865 | 866 | if len(button_pairs) != 0: 867 | response += bottom_msg 868 | 869 | # Button pairs consist of (button_text, callback_text) 870 | keyboard = [[InlineKeyboardButton( 871 | text, callback_data=callback) for text, callback in button_pairs]] 872 | 873 | end_t = timer() 874 | 875 | ellapsed_t = end_t - start_t 876 | db.register_action('friends_command', myself, additional_data={ 877 | 'ellapsed_time': ellapsed_t}) 878 | 879 | update.message.reply_text( 880 | response, reply_markup=InlineKeyboardMarkup(keyboard)) 881 | 882 | return CHOOSE_PAGE 883 | 884 | 885 | def change_friends_page(update, context): 886 | update.callback_query.answer() # await for answer 887 | 888 | # Trata a resposta anterior 889 | cur_page = int(update.callback_query.data) 890 | 891 | bottom_msg = "Utilize esses botões para navegar entre as páginas:\n\n" 892 | 893 | pages = context.user_data['friend_pages'] 894 | 895 | button_pairs = make_buttons(cur_page, len(pages) - 1) 896 | 897 | response = pages[cur_page] 898 | 899 | if len(button_pairs) != 0: 900 | response += bottom_msg 901 | 902 | # Button pairs consist of (button_text, callback_text) 903 | keyboard = [[InlineKeyboardButton( 904 | text, callback_data=callback) for text, callback in button_pairs]] 905 | 906 | update.callback_query.edit_message_text(response) 907 | 908 | update.callback_query.edit_message_reply_markup( 909 | reply_markup=InlineKeyboardMarkup(keyboard)) 910 | 911 | return CHOOSE_PAGE 912 | 913 | 914 | # ============================== ERROR/UNKNOWN ================================= 915 | 916 | 917 | def handle_incorrect_choice(update, context): 918 | db.register_action('user_ignored_buttons', update.effective_user.id) 919 | 920 | context.bot.sendMessage(chat_id=context.user_data['chat_id'], 921 | text='Você deve decidir a sua ação acerca do usuário acima antes de prosseguir.') 922 | 923 | return CHOOSE_ANSWER_FOR_BUTTONS 924 | 925 | 926 | def prefs_unknown_message(update, context): 927 | ''' 928 | Mensagem ou comando desconhecido (dentro da conversa de selecionar interesses) 929 | ''' 930 | db.register_action('prefs_wrong_action', update.effective_user.id) 931 | 932 | response_message = "Por favor, clique em ENVIAR para terminar de atualizar as suas preferências." 933 | update.message.reply_text(response_message) 934 | 935 | 936 | def friends_unknown_message(update, context): 937 | ''' 938 | Mensagem ou comando desconhecido (dentro da conversa do comando friends) 939 | ''' 940 | return ConversationHandler.END 941 | 942 | 943 | def unknown_message(update, context): 944 | ''' 945 | Mensagem ou comando desconhecido 946 | ''' 947 | 948 | response_message = "Não entendi! Por favor, use um comando válido...\nUse /help se estiver com dificuldades." 949 | update.message.reply_text(response_message) 950 | 951 | 952 | # ================================= ADMIN ====================================== 953 | 954 | def notify_command(update, context): 955 | 956 | # facilita na hora de referenciar esse usuario 957 | myself = update.effective_user.id 958 | 959 | is_admin = False 960 | admin_name = "" 961 | 962 | for admin in ADMINS: 963 | if myself == admin['telegramId']: 964 | is_admin = True 965 | admin_name = admin['name'] 966 | break 967 | 968 | if not is_admin: 969 | response = "O que você está tentando fazer? Esse comando é só para admins." 970 | update.message.reply_text(response) 971 | 972 | return CHOOSE_ACTION 973 | 974 | response = f"Olá {admin_name}!\n" 975 | response += "Me informe a mensagem que deseja mandar para TODOS os usuários do Approxima.\n" 976 | response += "PS: Lembre-se de usar esse recurso com responsabilidade :)" 977 | 978 | db.register_action('notify_command', myself) 979 | 980 | update.message.reply_text(response) 981 | 982 | return SEND_NOTIFICATION 983 | 984 | 985 | def send_notification(update, context): 986 | 987 | all_chats = db.list_chat_ids() 988 | 989 | for chat in all_chats: 990 | try: 991 | context.bot.sendMessage(chat_id=chat, text=update.message.text) 992 | except Exception as e: 993 | logger.error(f"Erro ao interagir com o chat {chat}: {e}") 994 | 995 | # Avisa que esse admin mandou o broadcast 996 | logger.info( 997 | f"{context.user_data['name']} mandou uma notificação para todos os usuários: {update.message.text}") 998 | 999 | db.register_action('admin_notified', update.effective_user.id, 1000 | additional_data={'message': update.message.text}) 1001 | 1002 | return CHOOSE_ACTION 1003 | 1004 | # ================================= MAIN ======================================= 1005 | 1006 | 1007 | def main(): 1008 | updater = Updater(token=TELEGRAM_TOKEN, use_context=True) 1009 | 1010 | prefs_handler = ConversationHandler( 1011 | entry_points=[ 1012 | CommandHandler('prefs', prefs_command) 1013 | ], 1014 | 1015 | states={ 1016 | CHOOSE_INTERESTS: [ 1017 | CallbackQueryHandler( 1018 | change_category_state, pattern='^toggle' 1019 | ), 1020 | CallbackQueryHandler( 1021 | open_category_state, pattern='^open' 1022 | ), 1023 | CallbackQueryHandler(submit_selection, pattern='^finish$'), 1024 | CallbackQueryHandler( 1025 | back_to_all_categories_state, pattern='^goback$') 1026 | ], 1027 | }, 1028 | 1029 | fallbacks=[ 1030 | MessageHandler(Filters.all, prefs_unknown_message) 1031 | ], 1032 | 1033 | map_to_parent={ 1034 | ConversationHandler.END: CHOOSE_ACTION 1035 | } 1036 | ) 1037 | 1038 | friends_handler = ConversationHandler( 1039 | entry_points=[ 1040 | CommandHandler('friends', friends_command) 1041 | ], 1042 | 1043 | states={ 1044 | CHOOSE_PAGE: [ 1045 | CallbackQueryHandler(change_friends_page, pattern='^[\\d]+$'), 1046 | ], 1047 | }, 1048 | 1049 | fallbacks=[ 1050 | MessageHandler(Filters.all, friends_unknown_message) 1051 | ], 1052 | 1053 | map_to_parent={ 1054 | ConversationHandler.END: CHOOSE_ACTION 1055 | } 1056 | ) 1057 | 1058 | conv_handler = ConversationHandler( 1059 | entry_points=[ 1060 | CommandHandler('start', start_command) 1061 | ], 1062 | 1063 | states={ 1064 | # O estado abaixo só será utilizado se o usuário não possuir um username 1065 | # (@ do Telegram) 1066 | LIMBO: [ 1067 | MessageHandler(Filters.all & ( 1068 | ~Filters.command), ask_for_username) 1069 | ], 1070 | 1071 | # Os primeiros dois estados são para o CADASTRO 1072 | REGISTER_NAME: [ 1073 | MessageHandler(Filters.text, register_name) 1074 | ], 1075 | 1076 | REGISTER_BIO: [ 1077 | MessageHandler(Filters.text, register_bio) 1078 | ], 1079 | 1080 | CHOOSE_ACTION: [ 1081 | prefs_handler, 1082 | CommandHandler('show', show_person_command), 1083 | CommandHandler('random', get_random_person_command), 1084 | CommandHandler('clear', clear_rejected_command), 1085 | CommandHandler('pending', pending_command), 1086 | friends_handler, 1087 | CommandHandler('name', edit_name_command), 1088 | CommandHandler('desc', edit_bio_command), 1089 | CommandHandler('help', help_command), 1090 | CommandHandler('notify', notify_command), # SO PARA ADMINS 1091 | ], 1092 | 1093 | CHOOSE_ANSWER_FOR_BUTTONS: [ 1094 | CallbackQueryHandler( 1095 | handle_invite_answer, pattern='^(connect|dismiss)$'), 1096 | CallbackQueryHandler( 1097 | handle_pending_answer, pattern='^(accept|reject)$'), 1098 | MessageHandler(Filters.all, handle_incorrect_choice), 1099 | ], 1100 | 1101 | GIVE_NEW_NAME: [ 1102 | MessageHandler(Filters.text, update_name) 1103 | ], 1104 | 1105 | GIVE_NEW_BIO: [ 1106 | MessageHandler(Filters.text, update_bio) 1107 | ], 1108 | 1109 | SEND_NOTIFICATION: [ 1110 | MessageHandler(Filters.text, send_notification) 1111 | ], 1112 | }, 1113 | 1114 | fallbacks=[ 1115 | CommandHandler('start', start_command), 1116 | MessageHandler(Filters.all, unknown_message) 1117 | ] 1118 | ) 1119 | 1120 | updater.dispatcher.add_handler(conv_handler) 1121 | 1122 | updater.start_polling() 1123 | logging.info("=== Bot running! ===") 1124 | updater.idle() 1125 | logging.info("=== Bot shutting down! ===") 1126 | 1127 | 1128 | if __name__ == '__main__': 1129 | print("press CTRL + C to cancel.") 1130 | main() 1131 | -------------------------------------------------------------------------------- /src/categories.py: -------------------------------------------------------------------------------- 1 | CATEGORIES = { 2 | "Anime e Mangá": [ 3 | 0, 4 | {} 5 | ], 6 | "Artesanato": [ 7 | 1, 8 | {} 9 | ], 10 | "Automóveis e Veículos": [ 11 | 2, 12 | {} 13 | ], 14 | "Beleza e Fitness": [ 15 | 3, 16 | {} 17 | ], 18 | "Casa e Jardim": [ 19 | 4, 20 | {} 21 | ], 22 | "Causas (ambientais, feminismo, vegan...)": [ 23 | 5, 24 | {} 25 | ], 26 | "Ciclismo": [ 27 | 6, 28 | {} 29 | ], 30 | "Ciência e Ensino (tópicos acadêmicos)": [ 31 | 7, 32 | {} 33 | ], 34 | "Compras": [ 35 | 8, 36 | {} 37 | ], 38 | "Culinária": [ 39 | 9, 40 | {} 41 | ], 42 | "Dança": [ 43 | 10, 44 | {} 45 | ], 46 | "Empreendedorismo e Negócios": [ 47 | 11, 48 | {} 49 | ], 50 | "Esotérico e Holístico": [ 51 | 12, 52 | {} 53 | ], 54 | "Espiritualidade": [ 55 | 13, 56 | {} 57 | ], 58 | "Esportes": [ 59 | 14, 60 | {} 61 | ], 62 | "Fantasia (RPG, senhor dos anéis, etc.)": [ 63 | 15, 64 | {} 65 | ], 66 | "Ficção científica": [ 67 | 16, 68 | {} 69 | ], 70 | "Filmes": [ 71 | 17, 72 | { 73 | "Animação": [ 74 | 0 75 | ], 76 | "Aventura": [ 77 | 1 78 | ], 79 | "Ação": [ 80 | 2 81 | ], 82 | "Comédia": [ 83 | 3 84 | ], 85 | "Documentário": [ 86 | 4 87 | ], 88 | "Drama": [ 89 | 5 90 | ], 91 | "Fantasia": [ 92 | 6 93 | ], 94 | "Faroeste": [ 95 | 7 96 | ], 97 | "Ficção Científica": [ 98 | 8 99 | ], 100 | "Guerra": [ 101 | 9 102 | ], 103 | "Musical": [ 104 | 10 105 | ], 106 | "Romance": [ 107 | 11 108 | ], 109 | "Suspense": [ 110 | 12 111 | ], 112 | "Terror": [ 113 | 13 114 | ] 115 | } 116 | ], 117 | "Finanças": [ 118 | 18, 119 | {} 120 | ], 121 | "Fotografia": [ 122 | 19, 123 | {} 124 | ], 125 | "Hardware": [ 126 | 20, 127 | {} 128 | ], 129 | "História": [ 130 | 21, 131 | {} 132 | ], 133 | "Idiomas": [ 134 | 22, 135 | {} 136 | ], 137 | "Imobiliário": [ 138 | 23, 139 | {} 140 | ], 141 | "Intercâmbio": [ 142 | 24, 143 | {} 144 | ], 145 | "Jogos de cartas": [ 146 | 25, 147 | {} 148 | ], 149 | "Jogos de tabuleiro": [ 150 | 26, 151 | {} 152 | ], 153 | "Jogos eletrônicos": [ 154 | 27, 155 | {} 156 | ], 157 | "Livros e Literatura": [ 158 | 28, 159 | { 160 | "Biográfico": [ 161 | 0 162 | ], 163 | "Científico": [ 164 | 1 165 | ], 166 | "Histórico": [ 167 | 2 168 | ], 169 | "Romance": [ 170 | 3 171 | ], 172 | "Suspense": [ 173 | 4 174 | ], 175 | "Épico": [ 176 | 5 177 | ] 178 | } 179 | ], 180 | "Mitologia": [ 181 | 29, 182 | {} 183 | ], 184 | "Moda": [ 185 | 30, 186 | {} 187 | ], 188 | "Mão na massa (consertos, costura, tricô, etc.)": [ 189 | 31, 190 | {} 191 | ], 192 | "Música": [ 193 | 32, 194 | { 195 | "Axé": [ 196 | 0 197 | ], 198 | "Blues": [ 199 | 1 200 | ], 201 | "Country": [ 202 | 2 203 | ], 204 | "Eletrônica": [ 205 | 3 206 | ], 207 | "Forró": [ 208 | 4 209 | ], 210 | "Funk": [ 211 | 5 212 | ], 213 | "Gospel": [ 214 | 6 215 | ], 216 | "Hip Hop": [ 217 | 7 218 | ], 219 | "J-rock": [ 220 | 8 221 | ], 222 | "Jazz": [ 223 | 9 224 | ], 225 | "K-pop": [ 226 | 10 227 | ], 228 | "MPB": [ 229 | 11 230 | ], 231 | "Música Clássica": [ 232 | 12 233 | ], 234 | "Pagode": [ 235 | 13 236 | ], 237 | "Pop": [ 238 | 14 239 | ], 240 | "Rap": [ 241 | 15 242 | ], 243 | "Reggae": [ 244 | 16 245 | ], 246 | "Rock": [ 247 | 17 248 | ], 249 | "Samba": [ 250 | 18 251 | ], 252 | "Sertanejo": [ 253 | 19 254 | ] 255 | } 256 | ], 257 | "Pessoas e Sociedade": [ 258 | 33, 259 | {} 260 | ], 261 | "Pets": [ 262 | 34, 263 | {} 264 | ], 265 | "Pintura e Desenho": [ 266 | 35, 267 | {} 268 | ], 269 | "Política": [ 270 | 36, 271 | {} 272 | ], 273 | "Quadrinhos": [ 274 | 37, 275 | {} 276 | ], 277 | "Rolês universitários": [ 278 | 38, 279 | {} 280 | ], 281 | "Saúde": [ 282 | 39, 283 | {} 284 | ], 285 | "Shows": [ 286 | 40, 287 | {} 288 | ], 289 | "Software": [ 290 | 41, 291 | {} 292 | ], 293 | "Séries": [ 294 | 42, 295 | { 296 | "Aventura": [ 297 | 0 298 | ], 299 | "Ação": [ 300 | 1 301 | ], 302 | "Biográfica": [ 303 | 2 304 | ], 305 | "Comédia": [ 306 | 3 307 | ], 308 | "Desenho Animado": [ 309 | 4 310 | ], 311 | "Documentário": [ 312 | 5 313 | ], 314 | "Drama": [ 315 | 6 316 | ], 317 | "Esportes Radicais": [ 318 | 7 319 | ], 320 | "Fantasia": [ 321 | 8 322 | ], 323 | "Faroeste": [ 324 | 9 325 | ], 326 | "Ficção Científica": [ 327 | 10 328 | ], 329 | "Ficção Histórica": [ 330 | 11 331 | ], 332 | "Humor Negro": [ 333 | 12 334 | ], 335 | "Mistério": [ 336 | 13 337 | ], 338 | "Musical": [ 339 | 14 340 | ], 341 | "Médica": [ 342 | 15 343 | ], 344 | "Policial": [ 345 | 16 346 | ], 347 | "Política": [ 348 | 17 349 | ], 350 | "Pós-apocalíptica": [ 351 | 18 352 | ], 353 | "Romance": [ 354 | 19 355 | ], 356 | "Sitcom (ex: Friends, HIMYM, TBBT...)": [ 357 | 20 358 | ], 359 | "Suspense": [ 360 | 21 361 | ], 362 | "Terror/Horror": [ 363 | 22 364 | ] 365 | } 366 | ], 367 | "Teatro": [ 368 | 43, 369 | {} 370 | ], 371 | "Trabalho voluntário": [ 372 | 44, 373 | {} 374 | ], 375 | "Viagens e Turismo": [ 376 | 45, 377 | {} 378 | ] 379 | } 380 | -------------------------------------------------------------------------------- /src/dbwrapper.py: -------------------------------------------------------------------------------- 1 | import os 2 | import sys 3 | import pymongo 4 | from dotenv import load_dotenv 5 | from bson.codec_options import CodecOptions 6 | import pytz 7 | from datetime import datetime 8 | 9 | 10 | class Database: 11 | def __init__(self, connection_string, is_production=False): 12 | client = pymongo.MongoClient(connection_string) 13 | self.db = client['approxima'] 14 | 15 | self.users = self.db['production-users'] if is_production else self.db['users'] 16 | self.users = self.users.with_options(codec_options=CodecOptions( 17 | tz_aware=True, 18 | tzinfo=pytz.timezone('America/Sao_Paulo'))) 19 | 20 | self.stats = self.db['production-stats'] if is_production else self.db['test-stats'] 21 | self.stats = self.stats.with_options(codec_options=CodecOptions( 22 | tz_aware=True, 23 | tzinfo=pytz.timezone('America/Sao_Paulo'))) 24 | 25 | self.today_id = self.__create_today_id() 26 | self.__create_today_doc() 27 | self.users_doc_checked = {} # Map from a user_id to a boolean 28 | 29 | def __create_today_id(self): 30 | today_date = datetime.utcnow() 31 | today_id = f"{today_date.year}-{today_date.month}-{today_date.day}" 32 | return today_id 33 | 34 | def __create_today_user_doc(self, user_id): 35 | try: 36 | self.stats.update_one({'_id': self.today_id}, { 37 | "$push": {"active_users": {"_id": user_id}}}) 38 | except: 39 | print( 40 | f"An exception occurred in the database while creating the user doc for day {datetime.utcnow()}:\n") 41 | print(sys.exc_info()[0]) 42 | 43 | def __create_today_doc(self): 44 | try: 45 | self.stats.insert_one( 46 | {"_id": self.today_id, "active_users": []}) 47 | except: 48 | print( 49 | f"An exception occurred in the database while creating the {self.today_id} day doc:\n") 50 | print(sys.exc_info()[0]) 51 | 52 | def list_user_ids(self): 53 | ids = [] 54 | for user in self.users.find(): 55 | ids.append(user['_id']) 56 | return ids 57 | 58 | def list_chat_ids(self): 59 | chat_ids = [] 60 | for user in self.users.find(): 61 | chat_ids.append(user['chat_id']) 62 | return chat_ids 63 | 64 | def get_user_by_id(self, telegram_id): 65 | try: 66 | return self.users.find_one({'_id': telegram_id}) 67 | except: 68 | print("An exception occurred in the database while getting user by id:\n") 69 | print(sys.exc_info()[0]) 70 | 71 | def get_users_in_list(self, users_list): 72 | # Users list is a list of telegram ID's 73 | try: 74 | query = {"_id": {"$in": users_list}} 75 | response = self.users.find(query) 76 | return list(response) 77 | except: 78 | print( 79 | "An exception occurred in the database while getting users in a list:\n") 80 | print(sys.exc_info()[0]) 81 | 82 | def update_user_by_id(self, telegram_id, data): 83 | try: 84 | new_values = {"$set": data} 85 | new_values["$set"]["updated_at"] = datetime.utcnow() 86 | self.users.update_one({'_id': telegram_id}, new_values) 87 | except: 88 | print("An exception occurred in the database while updating user by id:\n") 89 | print(sys.exc_info()[0]) 90 | 91 | def insert_user(self, telegram_id, data): 92 | try: 93 | new_document = { 94 | "_id": telegram_id, 95 | "chat_id": data['chat_id'], 96 | "username": data['username'], 97 | "name": data['name'], 98 | "bio": data['bio'], 99 | "interests": data['interests'], 100 | "rejects": data['rejects'], 101 | "invited": data['invited'], 102 | "pending": data['pending'], 103 | "connections": data['connections'], 104 | "created_at": datetime.utcnow(), 105 | "updated_at": None, 106 | } 107 | self.users.insert_one(new_document) 108 | except: 109 | print("An exception occurred in the database while updating user by id:\n") 110 | print(sys.exc_info()[0]) 111 | 112 | def register_action(self, action_name, user_id, additional_data=None): 113 | if not user_id or not action_name: 114 | raise ValueError( 115 | "Both \"action_name\" and \"user_id\" are required.") 116 | 117 | # Additional data must be a dict if not none 118 | if additional_data is not None and not isinstance(additional_data, dict): 119 | raise ValueError("\"additional_data\" must be a dict.") 120 | 121 | # Make sure that everything is ok with the Database before proceeding 122 | tid = self.__create_today_id() 123 | if tid != self.today_id: 124 | # I am in another day! 125 | self.today_id = tid 126 | self.__create_today_doc() 127 | self.__create_today_user_doc(user_id) 128 | self.user_doc_checked = True 129 | else: 130 | # Se não existir a entrada para este usuario OU se o booleano for False... 131 | if not self.users_doc_checked.get(user_id): 132 | # I'm gonna check it! 133 | query = {'_id': self.today_id, "active_users._id": user_id} 134 | if not self.stats.find_one(query): 135 | self.__create_today_user_doc(user_id) 136 | self.users_doc_checked[user_id] = True 137 | 138 | action = {} 139 | action['timestamp'] = datetime.utcnow() 140 | 141 | if additional_data: 142 | action['data'] = additional_data 143 | 144 | try: 145 | # Push to the correct array 146 | query = {'_id': self.today_id, "active_users._id": user_id} 147 | update = {"$push": {f"active_users.$.{action_name}": action}} 148 | 149 | self.stats.update_one(query, update) 150 | 151 | except: 152 | print( 153 | f"An exception occurred in the database while adding an action to user {user_id}:\n") 154 | print(sys.exc_info()[0]) 155 | 156 | 157 | def test(): 158 | load_dotenv() 159 | CONNECTION_STRING = os.getenv("CONNECTION_STRING") 160 | db = Database(CONNECTION_STRING) 161 | 162 | 163 | if __name__ == '__main__': 164 | test() 165 | -------------------------------------------------------------------------------- /src/ranker.py: -------------------------------------------------------------------------------- 1 | # This Python file uses the following encoding: utf-8 2 | 3 | import numpy as np 4 | 5 | 6 | def rank(my_interests, other_users_id_interests, log=False): # interests are lists in this moment 7 | ''' 8 | O segundo argumento é um map, que tem como chave o ID do Telegram do usuário 9 | e como valor um objeto contendo os interesses dele (Python list) e a posição original 10 | no vetor de allowed_users. A posição original deve ser ignorada por essa função. 11 | Essa funcao ja vai receber apenas usuarios elegiveis para serem sugeridos. 12 | ''' 13 | if len(other_users_id_interests) == 0: 14 | return None 15 | 16 | my_interests = np.array(my_interests) 17 | 18 | scores = np.zeros((len(other_users_id_interests), 2), dtype=np.uint32) 19 | 20 | for i, user_id in enumerate(other_users_id_interests): 21 | their_interests = np.array( 22 | other_users_id_interests[user_id]['interests']) 23 | their_score = len(np.intersect1d(my_interests, their_interests)) 24 | 25 | scores[i][0] = user_id 26 | scores[i][1] = their_score 27 | 28 | if log: 29 | print('\nPositional scores:\n', scores) 30 | 31 | ranking = scores[ 32 | np.flip(np.argsort(scores[:, 1])) 33 | ] # sort by score (flip to get in decreasing order) 34 | 35 | if ranking[0][1] == 0: 36 | # O maior score que o usuario conseguiu foi 0... 37 | return None 38 | 39 | most_similar_user = ranking[0][0] 40 | # until this point it is a np.uint32 41 | most_similar_user = int(most_similar_user) 42 | 43 | if log: 44 | print('\nRanking:\n', ranking) 45 | # Most Similar User interests 46 | msu_interests = other_users_id_interests[most_similar_user]['interests'] 47 | print( 48 | f"\nMost similar: (userId: {most_similar_user}, interests: {msu_interests})" 49 | ) 50 | 51 | return most_similar_user 52 | 53 | 54 | def test(): 55 | print('===== TESTE =====') 56 | 57 | my_interests = ['0', '1', '3', '6,2', '7,3'] 58 | 59 | users_interests = { 60 | 1111: { 61 | "interests": ['0', '5', '6,0', '7,2'], 62 | "original_pos": -1 63 | }, 64 | 2222: { 65 | "interests": ['3', '6,1', '6,2'], 66 | "original_pos": -1 67 | }, 68 | 3333: { 69 | "interests": ['5', '7,3'], 70 | "original_pos": -1 71 | }, 72 | 4444: { 73 | "interests": ['3', '4', '7,3', '7,4'], 74 | "original_pos": -1 75 | }, 76 | 5555: { 77 | "interests": ['0', '1', '4', '7,4', '6,1'], 78 | "original_pos": -1 79 | }, 80 | 6666: { 81 | "interests": ['1', '2', '6,2', '7,3'], 82 | "original_pos": -1 83 | }, 84 | 7777: { 85 | "interests": ['1', '2', '3', '4', '6,3'], 86 | "original_pos": -1 87 | }, 88 | 8888: { 89 | "interests": ['1', '5'], 90 | "original_pos": -1 91 | }, 92 | } 93 | 94 | print(rank(my_interests, users_interests, log=True)) 95 | 96 | 97 | if __name__ == '__main__': 98 | test() 99 | -------------------------------------------------------------------------------- /src/utils.py: -------------------------------------------------------------------------------- 1 | import numpy as np 2 | 3 | # Tested and it's okay. Preserves the order of the input's elements 4 | def unique_list(sequence): 5 | seen = dict.fromkeys(sequence) 6 | return list(seen) 7 | 8 | def make_buttons(cur_page, final_page): 9 | # All pages are passed as 0-based. Then, when returning the buttons with 10 | # structure (text, callback_data), callback_data is 0-based and text is 1-based 11 | 12 | button_pairs = [] 13 | 14 | if final_page == 0: # there is only one page, button are not needed 15 | return button_pairs # empty 16 | 17 | if final_page <= 4: 18 | # Sei que nro de botoes é certinho o nro de paginas 19 | num_buttons = final_page + 1 20 | 21 | for page in np.arange(num_buttons): 22 | if page == cur_page: 23 | button_pairs.insert(page, (f'⦗{page + 1}⦘', f'{page}')) 24 | else: 25 | button_pairs.insert(page, (f'{page + 1}', f'{page}')) 26 | 27 | return button_pairs 28 | 29 | # For here on it is guaranteed that there are more than 5 pages and, thus, 30 | # there are always 5 buttons 31 | 32 | # Build the first page button 33 | if cur_page == 0: 34 | button_pairs.append(('⦗1⦘', '0')) 35 | elif cur_page < 3: 36 | button_pairs.append(('1', '0')) 37 | else: # going back to first page is a huge step 38 | button_pairs.append(('« 1 ', '0')) 39 | 40 | # Build the last page button 41 | if cur_page == final_page: 42 | button_pairs.append((f'⦗{final_page + 1}⦘', f'{final_page}')) 43 | elif cur_page > final_page - 3: 44 | button_pairs.append((f'{final_page + 1}', f'{final_page}')) 45 | else: # going to the last page is a huge step 46 | button_pairs.append((f'{final_page + 1} »', f'{final_page}')) 47 | 48 | # Middle buttons 49 | 50 | if cur_page < 3: 51 | index = 1 52 | for page in np.arange(1, 3): 53 | if page == cur_page: 54 | button_pairs.insert(index, (f'⦗{page + 1}⦘', f'{page}')) 55 | else: 56 | button_pairs.insert(index, (f'{page + 1}', f'{page}')) 57 | index += 1 58 | button_pairs.insert(3, ('4 ›', '3')) 59 | 60 | elif cur_page > final_page - 3: 61 | index = 1 62 | for page in np.arange(final_page - 2, final_page): 63 | if page == cur_page: 64 | button_pairs.insert(index, (f'⦗{page + 1}⦘', f'{page}')) 65 | else: 66 | button_pairs.insert(index, (f'{page + 1}', f'{page}')) 67 | index += 1 68 | button_pairs.insert(1, (f'‹ {final_page - 2}', f'{final_page - 3}')) 69 | 70 | else: 71 | button_pairs.insert(1, (f'‹ {cur_page}', f'{cur_page - 1}')) 72 | button_pairs.insert(2, (f'⦗{cur_page + 1}⦘', f'{cur_page}')) 73 | button_pairs.insert(3, (f'{cur_page + 2} ›', f'{cur_page + 1}')) 74 | 75 | return button_pairs 76 | 77 | def correct_friends_order(list_of_friends, correct_order): 78 | ''' 79 | list_of_friends: a list of dicts with all the friend's info (including their _id) 80 | correct_order: a list of _ids that must be used to correct the order of the friends list 81 | 82 | returns: 83 | a list containing all dicts with friends infos, now sorted in the right order 84 | ''' 85 | 86 | return [ friend_data for friend_id in correct_order for friend_data in list_of_friends if friend_data['_id'] == friend_id ] --------------------------------------------------------------------------------