Teamcity timeout event handling

Hello,

I'm running python scripts by TeamCity, now timeout event is stopping script immediately, but I'd like to perform some steps after set time is elapsed. Is it any possibility to catch the trigger in Python? I tried with signal library:

class TeamCityTimeoutHander:
  def __init__(self):
    signal.signal(signal.SIGINT, self.stopExecution)
    signal.signal(signal.SIGTERM, self.stopExecution)

  def stopExecution(self):
    raise Exception("Test execution stopped by TeamCity Timeout")

but exception is not raised.

Do you have any ideas how to solve this issue?

Thanks,

Pawel