Update the examples to use the imported config object.

Fizzadar · Fizzadar · commit 87a0ea4e473d · 2021-11-01T09:56:45.000Z
diff --git a/examples/digitalocean/remove_10_ip.py b/examples/digitalocean/remove_10_ip.py
@@ -1,8 +1,8 @@
-from pyinfra import host
+from pyinfra import config, host
 from pyinfra.facts.hardware import Ipv4Addresses
 from pyinfra.operations import server
 
-SUDO = True
+config.SUDO = True
 
 ip = host.get_fact(Ipv4Addresses)['eth0']
 
diff --git a/examples/docker_ce.py b/examples/docker_ce.py
@@ -1,4 +1,4 @@
-from pyinfra import host
+from pyinfra import config, host
 from pyinfra.facts.server import LinuxName, LsbRelease
 from pyinfra.operations import apt, init, python
 
@@ -9,14 +9,17 @@
 # Please see https://github.com/Fizzadar/pyinfra-docker
 # for a more complete pyinfra docker installation.
 
-SUDO = True
+config.SUDO = True
 # If the remote system requires a password for sudo, un-comment the line below:
 # USE_SUDO_PASSWORD = True
 
 
 def check_docker_works(state, host):
     command = 'docker run hello-world'
-    status, stdout, stderr = host.run_shell_command(command=command, sudo=SUDO)
+    status, stdout, stderr = host.run_shell_command(
+        command=command,
+        sudo=config.SUDO,
+    )
     if not status or 'Hello from Docker!' not in stdout:
         raise Exception('`{}` did not work as expected'.format(command))
 
diff --git a/examples/files_download.py b/examples/files_download.py
@@ -1,8 +1,8 @@
-from pyinfra import host
+from pyinfra import config, host
 from pyinfra.facts.server import LinuxName
 from pyinfra.operations import apt, files
 
-SUDO = True
+config.SUDO = True
 
 if host.get_fact(LinuxName) == 'Ubuntu':
 
diff --git a/examples/files_line_with_quotes.py b/examples/files_line_with_quotes.py
@@ -1,6 +1,7 @@
+from pyinfra import config
 from pyinfra.operations import files
 
-SUDO = True
+config.SUDO = True
 
 # Run: pyinfra @docker/ubuntu files_line_with_quotes.py
 
diff --git a/examples/puppet/step0.py b/examples/puppet/step0.py
@@ -1,8 +1,8 @@
-from pyinfra import host
+from pyinfra import config, host
 from pyinfra.facts.hardware import Ipv4Addresses
 from pyinfra.operations import server
 
-SUDO = True
+config.SUDO = True
 
 ip = host.get_fact(Ipv4Addresses)['eth0']
 
diff --git a/examples/puppet/step1.py b/examples/puppet/step1.py
@@ -1,9 +1,9 @@
-from pyinfra import host, inventory
+from pyinfra import config, host, inventory
 from pyinfra.facts.hardware import Ipv4Addresses
 from pyinfra.facts.server import LinuxDistribution, LinuxName
 from pyinfra.operations import files, init, server, yum
 
-SUDO = True
+config.SUDO = True
 
 
 # update the /etc/hosts file
diff --git a/examples/puppet/step2.py b/examples/puppet/step2.py
@@ -1,8 +1,8 @@
-from pyinfra import host, inventory
+from pyinfra import config, host, inventory
 from pyinfra.operations import init, puppet, server
 
-SUDO = True
-USE_SUDO_LOGIN = True
+config.SUDO = True
+config.USE_SUDO_LOGIN = True
 
 if host in inventory.get_group('master_servers'):
     server.script_template(
diff --git a/examples/puppet/step3.py b/examples/puppet/step3.py
@@ -1,8 +1,8 @@
-from pyinfra import host, inventory
+from pyinfra import config, host, inventory
 from pyinfra.operations import files, puppet
 
-SUDO = True
-USE_SUDO_LOGIN = True
+config.SUDO = True
+config.USE_SUDO_LOGIN = True
 
 if host in inventory.get_group('master_servers'):
     files.template(
diff --git a/examples/pxe/pxe_infra.py b/examples/pxe/pxe_infra.py
@@ -1,8 +1,8 @@
-from pyinfra import host
+from pyinfra import config, host
 from pyinfra.facts.server import LinuxName
 from pyinfra.operations import apt, files, init, server
 
-SUDO = True
+config.SUDO = True
 
 # If you change pxe_server value below then check/change Vagrantfile
 pxe_server = '192.168.0.240'
diff --git a/examples/pxe_with_nfs/pxe_with_nfs_infra.py b/examples/pxe_with_nfs/pxe_with_nfs_infra.py
@@ -1,9 +1,9 @@
-from pyinfra import host
+from pyinfra import config, host
 from pyinfra.facts.files import File
 from pyinfra.facts.server import LinuxName
 from pyinfra.operations import apt, files, init, server
 
-SUDO = True
+config.SUDO = True
 
 # If you change pxe_server value below then check/change Vagrantfile
 pxe_server = '192.168.0.240'
diff --git a/examples/python_webserver/myweb.py b/examples/python_webserver/myweb.py
@@ -1,8 +1,8 @@
-from pyinfra import host
+from pyinfra import config, host
 from pyinfra.facts.server import LinuxName
 from pyinfra.operations import files, init, server
 
-SUDO = True
+config.SUDO = True
 
 if host.get_fact(LinuxName) in ['Ubuntu']:
 
diff --git a/examples/server_outputs.py b/examples/server_outputs.py
@@ -1,6 +1,7 @@
+from pyinfra import config
 from pyinfra.operations import server
 
-SUDO = True
+config.SUDO = True
 
 # Various different outputs
 
diff --git a/examples/ssh_demo/ssh_demo1.py b/examples/ssh_demo/ssh_demo1.py
@@ -1,7 +1,7 @@
-from pyinfra import host, inventory
+from pyinfra import config, host, inventory
 from pyinfra.operations import files, server
 
-SUDO = True
+config.SUDO = True
 
 
 # update the /etc/hosts file
diff --git a/examples/ssh_demo/ssh_demo2.py b/examples/ssh_demo/ssh_demo2.py
@@ -1,7 +1,7 @@
-from pyinfra import host
+from pyinfra import config, host
 from pyinfra.operations import server
 
-SUDO = True
+config.SUDO = True
 
 if host.name == '@vagrant/two':
 
diff --git a/examples/user_sudo_su.py b/examples/user_sudo_su.py
@@ -1,7 +1,8 @@
+from pyinfra import config
 from pyinfra.operations import files, server
 
-SUDO = True
-FAIL_PERCENT = 0
+config.SUDO = True
+config.FAIL_PERCENT = 0
 
 
 server.user(
diff --git a/examples/vagrant/vagrant.py b/examples/vagrant/vagrant.py
@@ -1,13 +1,13 @@
-from pyinfra import host
+from pyinfra import config, host
 from pyinfra.facts.server import LinuxName
 from pyinfra.operations import apt, files, python, server
 
-SUDO = True
+config.SUDO = True
 
 
 def verify_vagrant(state, host):
     command = 'vagrant --version'
-    status, stdout, stderr = host.run_shell_command(state, command=command, sudo=SUDO)
+    status, stdout, stderr = host.run_shell_command(state, command=command, sudo=config.SUDO)
     assert status is True  # ensure the command executed OK
     if 'Vagrant ' not in str(stdout):
         raise Exception('`{}` did not work as expected.stdout:{} stderr:{}'.format(
diff --git a/examples/virtualbox/virtualbox.py b/examples/virtualbox/virtualbox.py
@@ -1,15 +1,15 @@
-from pyinfra import host
+from pyinfra import config, host
 from pyinfra.facts.server import LinuxDistribution, LinuxName, OsVersion
 from pyinfra.operations import apt, python, server
 
-SUDO = True
+config.SUDO = True
 
 virtualbox_version = '6.1'
 
 
 def verify_virtualbox_version(state, host, version):
     command = '/usr/bin/virtualbox --help'
-    status, stdout, stderr = host.run_shell_command(state, command=command, sudo=SUDO)
+    status, stdout, stderr = host.run_shell_command(state, command=command, sudo=config.SUDO)
     assert status is True  # ensure the command executed OK
     if version not in str(stdout):
         raise Exception('`{}` did not work as expected.stdout:{} stderr:{}'.format(
