import os
import sys

import dotenv

if __name__ == "__main__":
    # We can't do read_dotenv('../environment') because that assumes that when
    # manage.py we are in its current directory, which isn't the case for cron
    # jobs.
    env_path = os.path.join(
        os.path.dirname(os.path.realpath(__file__)),
        '..', 'environment'
    )

    dotenv.read_dotenv(env_path, override=True)

    if len(sys.argv) > 1 and sys.argv[1] in ['test', 'pipeline_e2e_tests']:
        os.environ["DJANGO_SETTINGS_MODULE"] = "openprescribing.settings.test"

    from django.core.management import execute_from_command_line

    execute_from_command_line(sys.argv)

import os
import sys

import dotenv

if __name__ == "__main__":
    # We can't do read_dotenv('../environment') because that assumes that when
    # manage.py we are in its current directory, which isn't the case for cron
    # jobs.
    env_path = os.path.join(
        os.path.dirname(os.path.realpath(__file__)),
        '..', 'environment'
    )

    dotenv.read_dotenv(env_path, override=True)

    if len(sys.argv) > 1:
        if sys.argv[1] == 'test':
            os.environ["DJANGO_SETTINGS_MODULE"] = "openprescribing.settings.test"
        elif sys.argv[1] == 'pipeline_e2e_tests':
            os.environ["DJANGO_SETTINGS_MODULE"] = "openprescribing.settings.e2etest"

    from django.core.management import execute_from_command_line

    execute_from_command_line(sys.argv)

Set settings for e2e tests correctly
Set settings for e2e tests correctly

  import os
  import sys
  
  import dotenv
  
  if __name__ == "__main__":
      # We can't do read_dotenv('../environment') because that assumes that when
      # manage.py we are in its current directory, which isn't the case for cron
      # jobs.
      env_path = os.path.join(
          os.path.dirname(os.path.realpath(__file__)),
          '..', 'environment'
      )
  
      dotenv.read_dotenv(env_path, override=True)
  
-     if len(sys.argv) > 1 and sys.argv[1] in ['test', 'pipeline_e2e_tests']:
+     if len(sys.argv) > 1:
+         if sys.argv[1] == 'test':
-         os.environ["DJANGO_SETTINGS_MODULE"] = "openprescribing.settings.test"
+             os.environ["DJANGO_SETTINGS_MODULE"] = "openprescribing.settings.test"
+         elif sys.argv[1] == 'pipeline_e2e_tests':
+             os.environ["DJANGO_SETTINGS_MODULE"] = "openprescribing.settings.e2etest"
  
      from django.core.management import execute_from_command_line
  
      execute_from_command_line(sys.argv)
  
Set settings for e2e tests correctly
## Code Before:
import os
import sys

import dotenv

if __name__ == "__main__":
    # We can't do read_dotenv('../environment') because that assumes that when
    # manage.py we are in its current directory, which isn't the case for cron
    # jobs.
    env_path = os.path.join(
        os.path.dirname(os.path.realpath(__file__)),
        '..', 'environment'
    )

    dotenv.read_dotenv(env_path, override=True)

    if len(sys.argv) > 1 and sys.argv[1] in ['test', 'pipeline_e2e_tests']:
        os.environ["DJANGO_SETTINGS_MODULE"] = "openprescribing.settings.test"

    from django.core.management import execute_from_command_line

    execute_from_command_line(sys.argv)

## Instruction:
Set settings for e2e tests correctly
## Code After:
import os
import sys

import dotenv

if __name__ == "__main__":
    # We can't do read_dotenv('../environment') because that assumes that when
    # manage.py we are in its current directory, which isn't the case for cron
    # jobs.
    env_path = os.path.join(
        os.path.dirname(os.path.realpath(__file__)),
        '..', 'environment'
    )

    dotenv.read_dotenv(env_path, override=True)

    if len(sys.argv) > 1:
        if sys.argv[1] == 'test':
            os.environ["DJANGO_SETTINGS_MODULE"] = "openprescribing.settings.test"
        elif sys.argv[1] == 'pipeline_e2e_tests':
            os.environ["DJANGO_SETTINGS_MODULE"] = "openprescribing.settings.e2etest"

    from django.core.management import execute_from_command_line

    execute_from_command_line(sys.argv)
