import os
import psutil
import refinitiv.data as rd

if __name__ == '__main__':

    process = psutil.Process(os.getpid())
    mem = process.memory_info().rss / (1024 * 1024)

    platform_session = rd.session.platform.Definition(
        app_key="USER",
        deployed_platform_host="HOST",
        deployed_platform_username="USER",
        dacs_application_id=str("256")
    ).get_session()

    rdp_session = platform_session
    rd.session.set_default(rdp_session)
    rdp_session.open()

    process = psutil.Process(os.getpid())
    mem = process.memory_info().rss / (1024 * 1024)
    print(f"Memory before loop = {mem}")

    counter = 0
    while counter < 20:
        print(f"Loop #{counter}")

        mem = process.memory_info().rss / (1024 * 1024)
        print(f"Memory before open stream = {mem}")

        stream = rd.open_pricing_stream(
            universe=["EUR=", "JPY="],
            fields=['BID', 'ASK'],
            service='IDN'
        )

        mem = process.memory_info().rss / (1024 * 1024)
        print(f"Memory after open stream = {mem}")

        snap = stream.get_snapshot()

        mem = process.memory_info().rss / (1024 * 1024)
        print(f"Memory after get_snapshot = {mem}")

        stream.close()

        mem = process.memory_info().rss / (1024 * 1024)
        print(f"Memory after close stream = {mem}")

        print(f"End loop #{counter}")

        counter += 1

    print("Done")