Linux scripting with python

How to Execute Linux Commands in Python

Linux is one of the most popular operating systems used by software developers and system administrators. It is open-source, free, customizable, very robust, and adaptable. Making it an ideal choice for servers, virtual machines (VMs), and many other use cases.

Therefore, it is essential for anyone working in the tech industry to know how to work with Linux because it is used almost everywhere. In this tutorial, we are going to look at how we can automate and run Linux commands in Python.

Table of contents

Prerequisites

Introduction

Python has a rich set of libraries that allow us to execute shell commands.

A naive approach would be to use the os library:

import os cmd = 'ls -l' os.system(cmd) 

The os.system() function allows users to execute commands in Python. The program above lists all the files inside a directory. However, we can’t read and parse the output of the command.

In some commands, it is imperative to read the output and analyze it. The subprocess library provides a better, safer, and faster approach for this and allows us to view and parse the output of the commands.

Читайте также:  Измерения массива в java
OS subprocess
os.system function has been deprecated. In other words, this function has been replaced. The subprocess module serves as a replacement to this and Python officially recommends using subprocess for shell commands.
os.system directly executes shell commands and is susceptible to vulnerabilities. The subprocess module overcomes these vulnerabilities and is more secure.
The os.system function simply runs the shell command and only returns the status code of that command. The subprocess module returns an object that can be used to get more information on the output of the command and kill or terminate the command if necessary. This cannot be done in the os module.

Although you can execute commands using the OS module, the subprocess library provides a better and newer approach and is officially recommended. Therefore, we are going to use subprocess in this tutorial. This documentation explores the motivation behind creating this module.

Building an application to ping servers

Let’s use the subprocess library to write a script that pings multiple servers to see whether they are reachable or not. This would be a good use case when you have multiple hosts, servers, or VMs(AWS ec2 instances) and want to check if they are up and running without any problems.

A simple solution is to just ping these servers and see if they respond to the request. However, when you have a considerable amount of machines, it will be extremely tedious and time-consuming to manually ping them. A better approach is to use Python to automate this process.

Code

According to the official documentation, the subprocess module allows you to spawn new processes, connect to their input/output/error pipes, and obtain their return codes.

This module intends to replace several older modules and functions. The subprocess library has a class called Popen() that allows us to execute shell commands and get the output of the command.

Create a Python file and add the following code. We also need to create a file called “servers.txt”, where we can add a list of all the servers we need to ping. The Python script will read from this file and ping each server listed in it.

Servers

I have added 4 servers, out of which two exist and the other two do not. Only the servers that exist can be “pinged”.

import subprocess  def ping(servers):   # The command you want to execute  cmd = 'ping'   # send one packet of data to the host  # this is specified by '-c 1' in the argument list  outputlist = []  # Iterate over all the servers in the list and ping each server  for server in servers:  temp = subprocess.Popen([cmd, '-c 1', server], stdout = subprocess.PIPE)  # get the output as a string  output = str(temp.communicate())  # store the output in the list  outputlist.append(output)  return outputlist  if __name__ == '__main__':   # Get the list of servers from the text file  servers = list(open('servers.txt'))  # Iterate over all the servers that we read from the text file  # and remove all the extra lines. This is just a preprocessing step  # to make sure there aren't any unnecessary lines.  for i in range(len(servers)):  servers[i] = servers[i].strip('\n')  outputlist = ping(servers)   # Uncomment the following lines to print the output of successful servers  # print(outputlist) 

Output

As you can see in the output, we get the message “name or service not known” for the two servers that did not exist.

In the program above, the ping() function takes a list of servers and returns the output of each running ping command on each server. If a server is unreachable, it displays an output saying “ping: somethingthatdoesntexist: Name or service not known”.

The Popen() is a constructor method of the Popen class and takes in the following arguments:

  • A list of commands and any additional options these commands might require. For example, the ls command can be used with ‘-l’ option. To execute the ls -l command, the argument list would look like this: [‘ls’, ‘-l’] . The commands are specified as strings. In the example above, we use the ping command with the option -c 1 so that it only sends one packet of data, and the server replies with a single packet . Without this limit, the command would run forever until an external process stops it.
  • The stdout argument is optional and can be used to set where you want the subprocess to display the output. By default, the output is sent to the terminal. However, if you don’t want to dump a large output onto the terminal, you can use subprocess.PIPE to send the output of one command to the next. This corresponds to the | option in Linux.
  • The stderr argument is also optional and is used to set where you want the errors to be displayed. By default, it sends the errors to the terminal. Since we need to get a list of servers that cannot be reached, we don’t need to change this. The servers that cannot be reached (error) will be displayed to us on the terminal.

The output of the command is stored in a variable called temp . The communicate() function allows us to read the output and the str function can be used to convert it to a string. Once we get the output, we can parse it to extract only the essential details or just display it as it is. In this example, I am storing the output in a list for future use.

Conclusion

In conclusion, automation is one of the hottest topics in the industry, and almost every company is investing huge amounts of money to automate various manual tasks. In this tutorial, we explored the process of automatically running and analyzing Linux commands on multiple hosts using Python.

An old way of doing this is by using shell scripts. However, using Python gives developers more power and control over the execution and output of the commands. Now that you have understood the basics of executing Linux commands, you can go ahead and experiment with different commands and build more complex and robust applications.

Источник

How to Run Python Scripts in Linux

Python is one of the most popular programming languages of all. It’s an interpreted, object-oriented, high-level programming language that features dynamic semantics. If you’re using Linux, then you’ll come across Python scripts quite frequently.

One of the most basic and crucial things to learn is running a Python script when learning or working with Python. Because Python is an interpreted language, it requires the Python interpreter to execute any Python code. Depending on the type of script, there are a couple of ways you can execute it.

This guide will showcase executing a sample Python script.

Python scripts

Any script is a text file containing the code. The file can then be run using an interpreter. The same goes for any Python script.

Generally, a Python script will have the file extension PY. However, there’s another way of writing a Python script: embedding Python codes into a bash script.

Either way, you need to have the Python package installed in your system. Because it’s a popular programming language, all Linux distros offer pre-built Python binaries directly from the official package servers. Distros like Ubuntu, Linux Mint, Pop! OS etc., comes with Python pre-installed. The package name should be “python” or “python3″ for any other distros”.

Working with a Python script

Creating a sample Python script

For demonstration, let’s make a quick Python script. Open up the terminal and create a file named sample-script.py.

To be able to run the script, it must be marked as an executable file. Mark the file as an executable.

Check the file permission to verify if it worked.

Writing a sample Python code

Now, we’re going to put some code in the script. Open up the file in any text editor. For demonstration, I’m going to be using the nano text editor.

We’ll place a simple program that prints “hello world” on the console screen.

Save the file and close the editor.

Running the Python script

Finally, we can run the script. Call the Python interpreter and pass the location of the file.

Bash-style Python script

So far, we’ve seen the default way of running a Python script. However, there’s an unconventional way of writing and running a Python script as a shell script.

Generally, a shell script contains a list of commands that are interpreted and executed by a shell (bash, zsh, fish shell, etc.). A typical shell script uses shebang to declare the desired interpreter for the script.

We can take this structure to our advantage. We’ll declare the Python interpreter as the desired interpreter for our code. The body of the script will contain the desired Python scripts. Any modern shell will execute the script with the Python interpreter.

The structure will look something like this.

Location of Python interpreter

The shebang requires the path of the interpreter. It will tell the shell where to look for the interpreter. Generally, a Python interpreter is available as the command “python” or “python3”. Python 2 is deprecated, so it’s not recommended to use it anymore (except in very specific situations).

To find the location of the Python interpreter, use the which command. It finds the location of the binary of a command.

Creating a shell script

Similar to how we created the Python script, let’s create an empty shell script.

Mark the script as an executable file.

Writing a sample shell script

Open the script file in a text editor.

First, introduce the shebang with the location of the interpreter.

We’ll write a simple Python program that prints “hello world” on the next line.

Save the file and close the editor.

Running the script

Run the script as you’d run a shell script.

Final thought

It needs to be passed on to the interpreter to run a Python code. Using this principle, we can use various types of scripts to run our Python code. This guide demonstrated running Python scripts directly (filename.py scripts) or indirectly (filename.sh).

In Linux, scripts are generally used to automate certain tasks. If the task needs to be repeated regularly, it can also be automated with the help of crontab. Learn more about using crontab to automate various tasks.

About the author

Sidratul Muntaha

Student of CSE. I love Linux and playing with tech and gadgets. I use both Ubuntu and Linux Mint.

Источник

Оцените статью