Ran the function during project execution using rocketry

Question

I have a code similar to this, and I want the function "do_something" to be executed when I run the project. Its job is to prepare the database, etc.

I don't know how do it. I would it if you could guide me. I have read the rocketry documentation, but I couldn't find what I need.

from pyrogram import Client
from rocketry import Rocketry
import asyncio

api = Client("api")
cli = Client("cli")
rocktry = Rocketry()

async def do_something():
    await api.send_message(123456, "loading ....")
    # code
    await api.send_message(123456, "bot is ready!")

async def main():
    await asyncio.gather(api.start(), cli.start(), rocketry.serve())

api.run(main())

The idea that came to my mind is to set the value of a variable to false and then check in my code if the variable's value is false, perform this action, and afterwards set the value of that variable to true.

nischal sharma · Answer 1 · 2023-07-24T08:19:43.353

1

async def main():
    await asyncio.gather(api.start(), cli.start(), rocktry.serve())
    await do_something()

api.loop.run_until_complete(main())

This will run the do_something() function after the clients have started and the Rocketry server has started.

edited Jul 24 '23 at 08:19

answered Jul 23 '23 at 19:18

nischal sharma

479
1
14

This code does not work correctly because there is an await keyword on the previous line, and the next line will not be executed until that operation is completed. – sobhani Jul 24 '23 at 07:45
@sobhani ooh.. yes you are correct .... i have edited my answer accordingly ... sorry for the confusion – nischal sharma Jul 24 '23 at 08:19
1

thank you for your help. not worked again. I write my answer for this question – sobhani Jul 25 '23 at 11:52
1

your answer not working for me :/ – sobhani Jul 27 '23 at 12:27

score 1 · Answer 2 · answered Jul 25 '23 at 11:50

I change my code to this , and it's worked correctly:

from pyrogram import Client
from rocketry import Rocketry
import asyncio

api = Client("api")
cli = Client("cli")
rocktry = Rocketry()

async def do_something():
    while not api_app.is_connected:
        await asyncio.sleep(1)

    await api.send_message(123456, "loading ....")
    # code
    await api.send_message(123456, "bot is ready!")

async def main():
    await asyncio.gather(
        api_app.start(),
        do_something(),
        cli_app.start(),
        skd_app.serve()
    )

api.run(main())

I don't know if it has good performance or not, but it's exactly what I was looking for.

Ran the function during project execution using rocketry

2 Answers2