code: def atexit_register(func):
      
      try:
          import uwsgi
  
          orig = getattr(uwsgi, "atexit", None)
  
          def uwsgi_atexit():
              if callable(orig):
                  orig()
              func()
  
          uwsgi.atexit = uwsgi_atexit
      except ImportError:
          atexit.register(func)
docs: Uses either uwsgi's atexit mechanism, or atexit from the stdlib.
  
      When running under uwsgi, using their atexit handler is more reliable,
      especially when using gevent
      :param func: the function to call at exit