tilde
/
ssh-reg
3
0
Fork 2
Code Issues 4 Pull Requests Releases Wiki Activity
You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
master
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from '2ab27aa019'
${ noResults }
ssh-reg/private/Import.py

98 lines
4.8 KiB
Python
Raw Normal View History Unescape Escape

Move import related stuff to the Import.py script. and delete occurences in Import It is much tidier to look at and doesn't clutter the Script. Is even a whole seperate task so it does make no sense to call it in Backup when i think about it.
5 years ago
import lib.CFG as CFG
import csv
import os
import lib.UserExceptions
import lib.validator
def ImportFromFile(fname: str = CFG.args.file, db: str = CFG.REG_FILE, userids: tuple = tuple([])):
if not os.path.isfile(fname):
print(f"File {fname} don't exist")
return None
if not os.path.isfile(db):
print(f"The database file {db} don't exist")
return None
if userids:
pass # empty tuple means everything
# noinspection PyBroadException
try:
with open(fname, 'r', newline='') as f:
import lib.sqlitedb
import lib.System
sysctl = lib.System.System()
sql = lib.sqlitedb.SQLitedb(CFG.REG_FILE)
reader = csv.DictReader(f) # @TODO csv.Sniffer to compare? When yes, give force-accept option
for row in reader:
Validate Timestamps and don't not insert into DB when error Validates now the timestamp in the import.py, doesn't insert users into the sqlite-database before creating the systems account and checks now for existence in the database too(unapproved users, who comes first, gets his name first..)
5 years ago
db_insert = False
Move import related stuff to the Import.py script. and delete occurences in Import It is much tidier to look at and doesn't clutter the Script. Is even a whole seperate task so it does make no sense to call it in Backup when i think about it.
5 years ago
# if any of this fails move on to the next user, just print a relatively helpful message lel
if not lib.validator.checkUsernameCharacters(row["username"]):
print(f"The username contains unsupported characters or starts with a number: "
Validate Timestamps and don't not insert into DB when error Validates now the timestamp in the import.py, doesn't insert users into the sqlite-database before creating the systems account and checks now for existence in the database too(unapproved users, who comes first, gets his name first..)
5 years ago
f"{row['username']}. Skipping.")
Move import related stuff to the Import.py script. and delete occurences in Import It is much tidier to look at and doesn't clutter the Script. Is even a whole seperate task so it does make no sense to call it in Backup when i think about it.
5 years ago
continue
if not lib.validator.checkUsernameLength(row["username"]):
Validate Timestamps and don't not insert into DB when error Validates now the timestamp in the import.py, doesn't insert users into the sqlite-database before creating the systems account and checks now for existence in the database too(unapproved users, who comes first, gets his name first..)
5 years ago
print(f"The username {row['username']} is either too long(>16) or short(<3). Skipping.")
Move import related stuff to the Import.py script. and delete occurences in Import It is much tidier to look at and doesn't clutter the Script. Is even a whole seperate task so it does make no sense to call it in Backup when i think about it.
5 years ago
continue
if not lib.validator.checkSSHKey(row["pubkey"]):
Validate Timestamps and don't not insert into DB when error Validates now the timestamp in the import.py, doesn't insert users into the sqlite-database before creating the systems account and checks now for existence in the database too(unapproved users, who comes first, gets his name first..)
5 years ago
print(f"Following SSH-Key isn't valid: {row['pubkey']}. Skipping.")
Move import related stuff to the Import.py script. and delete occurences in Import It is much tidier to look at and doesn't clutter the Script. Is even a whole seperate task so it does make no sense to call it in Backup when i think about it.
5 years ago
continue
if not lib.validator.checkEmail(row["email"]):
Validate Timestamps and don't not insert into DB when error Validates now the timestamp in the import.py, doesn't insert users into the sqlite-database before creating the systems account and checks now for existence in the database too(unapproved users, who comes first, gets his name first..)
5 years ago
print(f"The E-Mail address {row['email']} is not valid. Skipping")
Move import related stuff to the Import.py script. and delete occurences in Import It is much tidier to look at and doesn't clutter the Script. Is even a whole seperate task so it does make no sense to call it in Backup when i think about it.
5 years ago
continue
Validate Timestamps and don't not insert into DB when error Validates now the timestamp in the import.py, doesn't insert users into the sqlite-database before creating the systems account and checks now for existence in the database too(unapproved users, who comes first, gets his name first..)
5 years ago
if not lib.validator.checkUserExists(row["username"]):
print(f"The user '{row['username']}' already exists. Skipping.")
continue
if not lib.validator.checkDatetimeFormat(row["timestamp"]):
print(f"The timestamp '{row['timestamp']}' from user '{row['username']}' is invalid. Skipping.")
Move import related stuff to the Import.py script. and delete occurences in Import It is much tidier to look at and doesn't clutter the Script. Is even a whole seperate task so it does make no sense to call it in Backup when i think about it.
5 years ago
continue
if row["status"] == "1":
try:
sysctl.register(row["username"])
sysctl.lock_user_pw(row["username"])
sysctl.add_to_usergroup(row["username"])
sysctl.make_ssh_usable(row["username"], row["pubkey"])
print(row['username'], "====> Registered.")
except lib.UserExceptions.UserExistsAlready as UEA:
pass # @TODO User was determined to exists already, shouldn't happen but is possible
except lib.UserExceptions.UnknownReturnCode as URC:
pass # @TODO Unknown Return Codes. Can happen in various function
except lib.UserExceptions.SSHDirUncreatable as SDU:
pass # @TODO SSH Directory doesn't exist AND couldn't be created. Inherently wrong design!
except lib.UserExceptions.ModifyFilesystem as MFS:
pass # @TODO Same as SSH Dir but more general, same problem: Wrong Permissions,
# Missing Dirs etc
except Exception as E: # @TODO well less broad is hard to achieve Kappa
print(E)
continue
Validate Timestamps and don't not insert into DB when error Validates now the timestamp in the import.py, doesn't insert users into the sqlite-database before creating the systems account and checks now for existence in the database too(unapproved users, who comes first, gets his name first..)
5 years ago
db_insert = True
Move import related stuff to the Import.py script. and delete occurences in Import It is much tidier to look at and doesn't clutter the Script. Is even a whole seperate task so it does make no sense to call it in Backup when i think about it.
5 years ago
elif row["status"] == "0":
print(row['username'] + " not approved, therefore not registered.")
Validate Timestamps and don't not insert into DB when error Validates now the timestamp in the import.py, doesn't insert users into the sqlite-database before creating the systems account and checks now for existence in the database too(unapproved users, who comes first, gets his name first..)
5 years ago
db_insert = True
Move import related stuff to the Import.py script. and delete occurences in Import It is much tidier to look at and doesn't clutter the Script. Is even a whole seperate task so it does make no sense to call it in Backup when i think about it.
5 years ago
try:
Validate Timestamps and don't not insert into DB when error Validates now the timestamp in the import.py, doesn't insert users into the sqlite-database before creating the systems account and checks now for existence in the database too(unapproved users, who comes first, gets his name first..)
5 years ago
if not db_insert:
continue
Move import related stuff to the Import.py script. and delete occurences in Import It is much tidier to look at and doesn't clutter the Script. Is even a whole seperate task so it does make no sense to call it in Backup when i think about it.
5 years ago
sql.safequery(
"INSERT INTO `applications` (username, name, timestamp, email, pubkey, status) "
"VALUES (?,?,?,?,?,?)", tuple([row["username"], row["name"], row["timestamp"],
row["email"], row["pubkey"], row["status"]]))
except OSError as E:
pass
print(f"UUFFF, something went WRONG with the file {fname}: {E}")
except Exception as didntCatch:
Validate Timestamps and don't not insert into DB when error Validates now the timestamp in the import.py, doesn't insert users into the sqlite-database before creating the systems account and checks now for existence in the database too(unapproved users, who comes first, gets his name first..)
5 years ago
print(f"Exception! UNCATCHED! {type(didntCatch)}: {didntCatch}")
Move import related stuff to the Import.py script. and delete occurences in Import It is much tidier to look at and doesn't clutter the Script. Is even a whole seperate task so it does make no sense to call it in Backup when i think about it.
5 years ago
return True
if __name__ == "__main__":
try:
if not CFG.args.Import:
print("Error, need the import flag")
if not CFG.args.file:
print("Error, need the import file")
if not CFG.args.file:
print("You MUST set a CSV-file with the -f/--file flag that already exist")
exit(1)
Validate Timestamps and don't not insert into DB when error Validates now the timestamp in the import.py, doesn't insert users into the sqlite-database before creating the systems account and checks now for existence in the database too(unapproved users, who comes first, gets his name first..)
5 years ago
ImportFromFile()
Move import related stuff to the Import.py script. and delete occurences in Import It is much tidier to look at and doesn't clutter the Script. Is even a whole seperate task so it does make no sense to call it in Backup when i think about it.
5 years ago
exit(0)
except KeyboardInterrupt as e:
pass