python cli: add a flag to ignore verifying the SSL certificate

This commit is contained in:
Nicola Murino 2019-08-07 23:06:26 +02:00
parent 8f421b7d0f
commit 5ad222fc53
3 changed files with 27 additions and 20 deletions

View file

@ -174,9 +174,9 @@ If you want to use a private key that use an algorithm different from RSA or mor
] ]
``` ```
The configuration can be read from JSON, TOML, YAML, HCL, envfile and Java properties config files, if your config-file flag is set to `sftpgo` (default value) you need to create a configuration file called `sftpgo.json` or `sftpgo.yaml` and so on inside `config-dir`. The configuration can be read from JSON, TOML, YAML, HCL, envfile and Java properties config files, if your `config-file` flag is set to `sftpgo` (default value) you need to create a configuration file called `sftpgo.json` or `sftpgo.yaml` and so on inside `config-dir`.
You can also configure all the available options using environment variables, sftpgo will check for a environment variable with a name matching the key uppercased and prefixed with the `SFTPGO_`. You need to use `__` to traverse a struct. You can also configure all the available options using environment variables, sftpgo will check for environment variables with a name matching the key uppercased and prefixed with the `SFTPGO_`. You need to use `__` to traverse a struct.
Let's see some examples: Let's see some examples:
@ -234,7 +234,7 @@ and you can add authentication with something like this:
``` ```
<Location /api/v1> <Location /api/v1>
AuthType Digest AuthType Digest
AuthName "Private" AuthName "Private"
AuthBasicProvider file AuthBasicProvider file
AuthUserFile "/etc/httpd/conf/auth_digest" AuthUserFile "/etc/httpd/conf/auth_digest"

View file

@ -11,7 +11,11 @@ python sftpgo_api_cli.py --help
Basically there is a subcommand for each REST API and a two global arguments: Basically there is a subcommand for each REST API and a two global arguments:
- `debug`, default disabled, print useful debug info. - `debug`, default disabled, print useful debug info.
- `base_url`, default `http://127.0.0.1:8080`. Base URL for SFTPGo REST API - `base_url`, default `http://127.0.0.1:8080`. Base URL for SFTPGo REST API
- `auth_type` supported auth type are `basic` and `digest`. Default none
- `auth_user`
- `auth_password`
- `verify`, disable to ignore verifying the SSL certificate. Default enabled
For each subcommand `--help` shows the required arguments, try for example: For each subcommand `--help` shows the required arguments, try for example:
@ -206,8 +210,8 @@ Output:
```json ```json
{ {
"status": 201, "status": 201,
"message": "Scan started", "message": "Scan started",
"error": "" "error": ""
} }
``` ```

View file

@ -3,7 +3,6 @@ import argparse
import json import json
import requests import requests
from requests.auth import HTTPBasicAuth, HTTPDigestAuth
try: try:
import urllib.parse as urlparse import urllib.parse as urlparse
@ -13,17 +12,18 @@ except ImportError:
class SFTPGoApiRequests: class SFTPGoApiRequests:
def __init__(self, debug, baseUrl, authType, authUser, authPassword): def __init__(self, debug, baseUrl, authType, authUser, authPassword, verify):
self.userPath = urlparse.urljoin(baseUrl, "/api/v1/user") self.userPath = urlparse.urljoin(baseUrl, "/api/v1/user")
self.quotaScanPath = urlparse.urljoin(baseUrl, "/api/v1/quota_scan") self.quotaScanPath = urlparse.urljoin(baseUrl, "/api/v1/quota_scan")
self.activeConnectionsPath = urlparse.urljoin(baseUrl, "/api/v1/sftp_connection") self.activeConnectionsPath = urlparse.urljoin(baseUrl, "/api/v1/sftp_connection")
self.debug = debug self.debug = debug
if authType == "basic": if authType == "basic":
self.auth = HTTPBasicAuth(authUser, authPassword) self.auth = requests.auth.HTTPBasicAuth(authUser, authPassword)
elif authType == "digest": elif authType == "digest":
self.auth = HTTPDigestAuth(authUser, authPassword) self.auth = requests.auth.HTTPDigestAuth(authUser, authPassword)
else: else:
self.auth = None self.auth = None
self.verify = verify
def formatAsJSON(self, text): def formatAsJSON(self, text):
if not text: if not text:
@ -55,18 +55,18 @@ class SFTPGoApiRequests:
def getUsers(self, limit=100, offset=0, order="ASC", username=""): def getUsers(self, limit=100, offset=0, order="ASC", username=""):
r = requests.get(self.userPath, params={"limit":limit, "offset":offset, "order":order, r = requests.get(self.userPath, params={"limit":limit, "offset":offset, "order":order,
"username":username}, auth=self.auth) "username":username}, auth=self.auth, verify=self.verify)
self.printResponse(r) self.printResponse(r)
def getUserByID(self, user_id): def getUserByID(self, user_id):
r = requests.get(urlparse.urljoin(self.userPath, "user/" + str(user_id)), auth=self.auth) r = requests.get(urlparse.urljoin(self.userPath, "user/" + str(user_id)), auth=self.auth, verify=self.verify)
self.printResponse(r) self.printResponse(r)
def addUser(self, username="", password="", public_key="", home_dir="", uid=0, gid=0, max_sessions=0, def addUser(self, username="", password="", public_key="", home_dir="", uid=0, gid=0, max_sessions=0,
quota_size=0, quota_files=0, permissions=[], upload_bandwidth=0, download_bandwidth=0): quota_size=0, quota_files=0, permissions=[], upload_bandwidth=0, download_bandwidth=0):
u = self.buildUserObject(0, username, password, public_key, home_dir, uid, gid, max_sessions, u = self.buildUserObject(0, username, password, public_key, home_dir, uid, gid, max_sessions,
quota_size, quota_files, permissions, upload_bandwidth, download_bandwidth) quota_size, quota_files, permissions, upload_bandwidth, download_bandwidth)
r = requests.post(self.userPath, json=u, auth=self.auth) r = requests.post(self.userPath, json=u, auth=self.auth, verify=self.verify)
self.printResponse(r) self.printResponse(r)
def updateUser(self, user_id, username="", password="", public_key="", home_dir="", uid=0, gid=0, def updateUser(self, user_id, username="", password="", public_key="", home_dir="", uid=0, gid=0,
@ -74,15 +74,15 @@ class SFTPGoApiRequests:
download_bandwidth=0): download_bandwidth=0):
u = self.buildUserObject(user_id, username, password, public_key, home_dir, uid, gid, max_sessions, u = self.buildUserObject(user_id, username, password, public_key, home_dir, uid, gid, max_sessions,
quota_size, quota_files, permissions, upload_bandwidth, download_bandwidth) quota_size, quota_files, permissions, upload_bandwidth, download_bandwidth)
r = requests.put(urlparse.urljoin(self.userPath, "user/" + str(user_id)), json=u, auth=self.auth) r = requests.put(urlparse.urljoin(self.userPath, "user/" + str(user_id)), json=u, auth=self.auth, verify=self.verify)
self.printResponse(r) self.printResponse(r)
def deleteUser(self, user_id): def deleteUser(self, user_id):
r = requests.delete(urlparse.urljoin(self.userPath, "user/" + str(user_id)), auth=self.auth) r = requests.delete(urlparse.urljoin(self.userPath, "user/" + str(user_id)), auth=self.auth, verify=self.verify)
self.printResponse(r) self.printResponse(r)
def getSFTPConnections(self): def getSFTPConnections(self):
r = requests.get(self.activeConnectionsPath, auth=self.auth) r = requests.get(self.activeConnectionsPath, auth=self.auth, verify=self.verify)
self.printResponse(r) self.printResponse(r)
def closeSFTPConnection(self, connectionID): def closeSFTPConnection(self, connectionID):
@ -90,12 +90,12 @@ class SFTPGoApiRequests:
self.printResponse(r) self.printResponse(r)
def getQuotaScans(self): def getQuotaScans(self):
r = requests.get(self.quotaScanPath, auth=self.auth) r = requests.get(self.quotaScanPath, auth=self.auth, verify=self.verify)
self.printResponse(r) self.printResponse(r)
def startQuotaScan(self, username): def startQuotaScan(self, username):
u = self.buildUserObject(0, username) u = self.buildUserObject(0, username)
r = requests.post(self.quotaScanPath, json=u, auth=self.auth) r = requests.post(self.quotaScanPath, json=u, auth=self.auth, verify=self.verify)
self.printResponse(r) self.printResponse(r)
@ -120,7 +120,7 @@ if __name__ == '__main__':
parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter) parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
parser.add_argument("--base_url", type=str, default="http://127.0.0.1:8080", parser.add_argument("--base_url", type=str, default="http://127.0.0.1:8080",
help="Base URL for SFTPGo REST API. Default: %(default)s") help="Base URL for SFTPGo REST API. Default: %(default)s")
parser.add_argument("--auth_type", type=str, default="", choices=["", "basic", "digest"], parser.add_argument("--auth_type", type=str, default=None, choices=["basic", "digest"],
help="Authentication type to use. Default: %(default)s") help="Authentication type to use. Default: %(default)s")
parser.add_argument("--auth_user", type=str, default="", parser.add_argument("--auth_user", type=str, default="",
help="User to use for authentication. Default: %(default)s") help="User to use for authentication. Default: %(default)s")
@ -128,6 +128,9 @@ if __name__ == '__main__':
help="Password to use for authentication. Default: %(default)s") help="Password to use for authentication. Default: %(default)s")
parser.add_argument("--debug", dest='debug', action='store_true') parser.add_argument("--debug", dest='debug', action='store_true')
parser.set_defaults(debug=False) parser.set_defaults(debug=False)
parser.add_argument("--verify", dest='verify', action='store_true',
help="Set to false to ignore verifying the SSL certificate")
parser.set_defaults(verify=True)
subparsers = parser.add_subparsers(dest="command", help='sub-command --help') subparsers = parser.add_subparsers(dest="command", help='sub-command --help')
subparsers.required = True subparsers.required = True
@ -164,7 +167,7 @@ if __name__ == '__main__':
args = parser.parse_args() args = parser.parse_args()
api = SFTPGoApiRequests(args.debug, args.base_url, args.auth_type, args.auth_user, args.auth_password) api = SFTPGoApiRequests(args.debug, args.base_url, args.auth_type, args.auth_user, args.auth_password, args.verify)
if args.command == "add_user": if args.command == "add_user":
api.addUser(args.username, args.password, args.public_key, args.home_dir, api.addUser(args.username, args.password, args.public_key, args.home_dir,