tilde
/
ssh-reg
3
0
Fork 2
Code Issues 4 Pull Requests Releases Wiki Activity

Merge remote-tracking branch 'origin/pr/3'

Browse Source
master
n1trux 4 years ago
parent d79e06a70e c4912502d7
commit 43a3dd81ac
3 changed files with 500 additions and 488 deletions
Show Stats Download Patch File Download Diff File

6
Dockerfile
Unescape Escape View File

@ -38,12 +38,10 @@ RUN mkdir -p /var/run/sshd
# expose SSH port # expose SSH port
EXPOSE 22 EXPOSE 22
ENV TILDE_CONF="/app/data/applicationsconfig.ini" ENV TILDE_CONF="/app/data/applicationsconfig.ini"
#COPY config/environment /app/user/.ssh/environment
RUN mkdir /app/user/.ssh
RUN echo TILDE_CONF=$TILDE_CONF > /app/user/.ssh/environment
RUN touch /app/data/applications.sqlite RUN touch /app/data/applications.sqlite
RUN touch /app/data/applications.log RUN touch /app/data/applications.log
# Doesnt work, @TODO why # Doesnt work, @TODO why
#RUN setfacl -R -m u:tilde:rwx /app/data/ #RUN setfacl -R -m u:tilde:rwx /app/data/
RUN chown -R tilde /app/data RUN chown -R tilde /app/data
CMD ["/usr/sbin/sshd", "-D"] RUN mkdir /app/user/.ssh
CMD ["sh", "-c", " echo TILDE_CONF=$TILDE_CONF > /app/user/.ssh/environment && /usr/sbin/sshd -D"]

275
private/administrate.py
Unescape Escape View File

@ -4,16 +4,20 @@ import configparser, logging, sqlite3, argparse, pwd
import os import os
import subprocess import subprocess
# Clear shell # Clear shell
def clear(): def clear():
os.system('cls' if os.name == 'nt' else 'clear') os.system('cls' if os.name == 'nt' else 'clear')
# create dictionary out of sqlite results # create dictionary out of sqlite results
def dict_factory(cursor, row): def dict_factory(cursor, row):
d = {} d = {}
for idx, col in enumerate(cursor.description): for idx, col in enumerate(cursor.description):
d[col[0]] = row[idx] d[col[0]] = row[idx]
return d return d
# prints command(but doesnt execute them) # prints command(but doesnt execute them)
# need this for work, just convenience # need this for work, just convenience
def debugExec(commands): def debugExec(commands):
@ -24,38 +28,38 @@ def debugExec(commands):
# @TODO hardcoded config? # @TODO hardcoded config?
cwd = os.environ.get('TILDE_CONF') cwd = os.environ.get('TILDE_CONF')
if cwd is None: if cwd is None:
cwd=os.getcwd()+"/applicationsconfig.ini" cwd = os.getcwd() + "/applicationsconfig.ini"
else: else:
if os.path.isfile(cwd) is False: if os.path.isfile(cwd) is False:
cwd=os.getcwd()+"/applicationsconfig.ini" cwd = os.getcwd() + "/applicationsconfig.ini"
# cwd is now either cwd/applicationsconfig or $TILDE_CONF # cwd is now either cwd/applicationsconfig or $TILDE_CONF
argparser = argparse.ArgumentParser(description = 'interactive registration formular for tilde platforms') argparser = argparse.ArgumentParser(description='interactive registration formular for tilde platforms')
argparser.add_argument('-c', '--config', default = cwd, argparser.add_argument('-c', '--config', default=cwd,
type = str, help = 'Path to configuration file', required = False) type=str, help='Path to configuration file', required=False)
args = argparser.parse_args() args = argparser.parse_args()
CONF_FILE = args.config CONF_FILE = args.config
config = configparser.ConfigParser() config = configparser.ConfigParser()
config.read(CONF_FILE) config.read(CONF_FILE)
logging.basicConfig(format="%(asctime)s: %(message)s", logging.basicConfig(format="%(asctime)s: %(message)s",
level = int(config['LOG_LEVEL']['log_level']) level=int(config['LOG_LEVEL']['log_level'])
) )
del(cwd) del cwd
REG_FILE = config['DEFAULT']['applications_db'] REG_FILE = config['DEFAULT']['applications_db']
# Does everything related to applicants, i.e. creating, manipulations... # Does everything related to applicants, i.e. creating, manipulations...
class applicants(): class Applicants:
# User identifier # User identifier
identifier = "username" identifier = "username"
# SQLite DB Path # SQLite DB Path
sourceDB = "" sourceDB = ""
# another sqlite to batch-recreate users # another sqlite to batch-recreate users
differentDB = "" differentDB = ""
def __init__(self, lident, sourceDB=REG_FILE):
def __init__(self, lident, sourcedb=REG_FILE):
self.identifier = lident self.identifier = lident
self.sourceDB = sourceDB self.sourceDB = sourcedb
self.__connectToDB__("source") self.__connectToDB__("source")
# all results shall be done with dict_factory! Makes everything so much simpler # all results shall be done with dict_factory! Makes everything so much simpler
self.sdbCursor.row_factory = dict_factory self.sdbCursor.row_factory = dict_factory
@ -75,58 +79,61 @@ class applicants():
self.ddbCursor = self.ddbConnection.cursor() self.ddbCursor = self.ddbConnection.cursor()
def __closeDB__(self, which): def __closeDB__(self, which):
if(which == "source"): if which == "source":
try: try:
self.sdbConnection.close() self.sdbConnection.close()
except sqlite3.Error as e: except sqlite3.Error as e:
logging.exception("Couldn't close database! Error: %s" % e) # @TODO: Dump full db with query or just the executed querys to file logging.exception(
"Couldn't close database! Error: %s" % e)
# @TODO: Dump full db with query or just the executed querys to file
else: else:
self.ddbConnection.close() # @TODO: Evaluate getting rid of ddb(differentDB)? self.ddbConnection.close() # @TODO: Evaluate getting rid of ddb(differentDB)?
# get List of all applications(not accepted yet) # get List of all applications(not accepted yet)
def getApplicationsList(self): def getapplicationslist(self):
query = "SELECT * FROM `applications` WHERE `status` = '0'" query = "SELECT * FROM `applications` WHERE `status` = '0'"
try: try:
self.sdbCursor.execute(query) self.sdbCursor.execute(query)
rows = self.sdbCursor.fetchall() rows = self.sdbCursor.fetchall()
except sqlite3.Error as e: except sqlite3.Error as e:
logging.exception("Database Error: %s" % e) logging.exception("Database Error: %s" % e)
rows=[] rows = []
return rows return rows
def getApprovedApplicantsList(self): def getapprovedapplicantslist(self):
query = "SELECT * From `applications` WHERE `status` = '1'" query = "SELECT * From `applications` WHERE `status` = '1'"
try: try:
self.sdbCursor.execute(query) self.sdbCursor.execute(query)
rows = self.sdbCursor.fetchall() rows = self.sdbCursor.fetchall()
except sqlite3.Error as e: except sqlite3.Error as e:
logging.exception("Database Error: %s" % e) logging.exception("Database Error: %s" % e)
rows=[] rows = []
return rows return rows
# edit aproved users # edit aproved users
def editApprovedApplicant(self, term, updaterow): def editapprovedapplicants(self, term):
try: try:
# the fuck did i try here?
self.sdbCursor.execute( self.sdbCursor.execute(
"UPDATE `applications` SET ? WHERE id=?", "UPDATE `applications` WHERE id=?", (str(term),)
( str(term), )
) )
self.sdbConnection.commit() self.sdbConnection.commit()
except sqlite3.Error as e: except sqlite3.Error as e:
logging.exception("Database Error: %s" % e) logging.exception("Database Error: %s" % e)
# set user to aproved # set user to aproved
def setApprovedApplication(self, selectterm): def setapprovedapplication(self, selectterm):
query = "SELECT `username` FROM `applications` WHERE `username` = `{0!s}`".format(selectterm) # query = "SELECT `username` FROM `applications` WHERE `username` = `{0!s}`".format(selectterm)
pass
# get applicants data # get applicants data
def getApplicantsData(self, term): def getapplicantsdata(self, term):
# @TODO: Use shorthand if for the correct query, directly into sqlite # @TODO: Use shorthand if for the correct query, directly into sqlite
if self.identifier == "id": if self.identifier == "id":
try: try:
self.sdbCursor.execute( self.sdbCursor.execute(
"SELECT * FROM `applications` WHERE id = ?", "SELECT * FROM `applications` WHERE id = ?",
( str(term), ) (str(term),)
) )
except sqlite3.Error as e: except sqlite3.Error as e:
logging.exception("Database Error: %s" % e) logging.exception("Database Error: %s" % e)
@ -134,26 +141,26 @@ class applicants():
else: else:
self.sdbCursor.execute( self.sdbCursor.execute(
"SELECT * FROM `applications` WHERE username = ?", "SELECT * FROM `applications` WHERE username = ?",
( str(term), ) (str(term),)
) )
result = self.sdbCursor.fetchone() result = self.sdbCursor.fetchone()
return result return result
# @TODO: migrade just approved users to some new/another sqlitedb # @TODO: migrade just approved users to some new/another sqlitedb
def migrateApprovedData(self, different_db): def migrateapproveddata(self, different_db):
pass pass
# @TODO: delete migrated data # @TODO: delete migrated data
def deleteMigratedDataSet(self, selectterm): def deletemigrateddata(self, selectterm):
pass pass
# Applicants whom doesnt got approved should get removed # Applicants whom doesnt got approved should get removed
def removeApplicant(self, term): def removeapplicant(self, term):
if self.identifier == "id": if self.identifier == "id":
try: try:
self.sdbCursor.execute( self.sdbCursor.execute(
"DELETE FROM `applications` WHERE id = ?", "DELETE FROM `applications` WHERE id = ?",
( str(term), ) (str(term),)
) )
self.sdbConnection.commit() self.sdbConnection.commit()
except sqlite3.Error as e: except sqlite3.Error as e:
@ -161,25 +168,26 @@ class applicants():
else: else:
self.sdbCursor.execute( self.sdbCursor.execute(
"DELETE FROM `applications` WHERE username = ?", 'DELETE FROM `applications` WHERE username = ?',
( str(term), ) (str(term),)
) )
self.sdbConnection.commit() self.sdbConnection.commit()
#@TODO: Possibility to work without passing users manually # @TODO: Possibility to work without passing users manually
def selectedUser(userid, username = False): def selecteduser(userid, username=False):
pass pass
# Print out a list of aprovable users # Print out a list of aprovable users
def printApprovableUsers(self, users): def printapprovableusers(self, users):
i=0 i = 0
for user in users: for user in users:
print("ID: {0!s}, Status: {0!s}, Name: {0!s}".format(i, user["status"], user["username"])) print("ID: {0!s}, Status: {0!s}, Name: {0!s}".format(i, user["status"], user["username"]))
i += 1 i += 1
return i return i
# Get List of users # Get List of users
def userPrint(self, fetched, userid): @staticmethod
def userprint(fetched, userid):
print("ID: {0!s}".format(fetched[int(userid)]["id"])) print("ID: {0!s}".format(fetched[int(userid)]["id"]))
print("Username: {0!s}".format(fetched[int(userid)]["username"])) print("Username: {0!s}".format(fetched[int(userid)]["username"]))
print("Mail: {0!s}".format(fetched[int(userid)]["email"])) print("Mail: {0!s}".format(fetched[int(userid)]["email"]))
@ -187,8 +195,8 @@ class applicants():
print("Registrated time: {0!s}".format(fetched[int(userid)]["timestamp"])) print("Registrated time: {0!s}".format(fetched[int(userid)]["timestamp"]))
# Approve an applicant. Handles everything related, like create home dir, set flags blabla # Approve an applicant. Handles everything related, like create home dir, set flags blabla
def approveApplicant(self, term): def approveapplicant(self, term):
user = self.getApplicantsData(term) user = self.getapplicantsdata(term)
ret = self.__execScript(user) ret = self.__execScript(user)
if ret[0] != 0: # @DEBUG: Change to == 0 if ret[0] != 0: # @DEBUG: Change to == 0
print("Something went wrong in the user creation! Exiting without deleting users record in database!") print("Something went wrong in the user creation! Exiting without deleting users record in database!")
@ -199,7 +207,7 @@ class applicants():
try: try:
self.sdbCursor.execute( self.sdbCursor.execute(
"UPDATE `applications` SET `status`=1 WHERE `id`=?", "UPDATE `applications` SET `status`=1 WHERE `id`=?",
( str(term), ) (str(term),)
) )
self.sdbConnection.commit() self.sdbConnection.commit()
except sqlite3.Error as e: except sqlite3.Error as e:
@ -207,172 +215,170 @@ class applicants():
else: else:
self.sdbCursor.execute( self.sdbCursor.execute(
"UPDATE `applications` SET `status`=1 WHERE `username`=?" "UPDATE `applications` SET `status`=1 WHERE `username`=?",
( str(term), ) (str(term), )
) )
self.sdbConnection.commit() self.sdbConnection.commit()
# Script execution, handles everything done with the shell/commands themselves # Script execution, handles everything done with the shell/commands themselves
def __execScript(self, user): @staticmethod
def __execScript(user):
# @TODO: omfg just write some wrapper-class/lib... sucks hard! # @TODO: omfg just write some wrapper-class/lib... sucks hard!
username=user["username"] username = user["username"]
homeDir="/home/"+username+"/" home_dir = "/home/" + username + "/"
sshDir=homeDir+".ssh/" ssh_dir = home_dir + ".ssh/"
executed=[] executed = []
executed.append(["useradd", "-m", username]) executed.append(["useradd", "-m", username])
rcode = subprocess.call(executed[0]) returncode = subprocess.call(executed[0])
if rcode != 0: if returncode != 0:
return [rcode,executed,] return [returncode, executed, ]
executed.append(["usermod", "--lock", username]) executed.append(["usermod", "--lock", username])
rcode = subprocess.call(executed[1]) #empty pw returncode = subprocess.call(executed[1]) # empty pw
if rcode != 0: if returncode != 0:
return [rcode,executed,] return [returncode, executed, ]
executed.append(["usermod", "-a", "-G", "tilde", username]) executed.append(["usermod", "-a", "-G", "tilde", username])
rcode = subprocess.call(executed[2]) # add to usergroup returncode = subprocess.call(executed[2]) # add to usergroup
if rcode != 0: if returncode != 0:
return [rcode,executed,] return [returncode, executed, ]
executed.append(["mkdir", sshDir]) executed.append(["mkdir", ssh_dir])
try: try:
# @TODO: use config variable(chmodPerms) # @TODO: use config variable(chmodPerms)
ret = os.mkdir(sshDir, 0o777) #create sshdir os.mkdir(ssh_dir, 0o777) # create sshdir
rcode = 0 returncode = 0
except OSError as e: except OSError as e:
logging.exception(e.strerror) logging.exception(e.strerror)
rcode = e.errno # False, couldn't create. returncode = e.errno # False, couldn't create.
return [rcode,executed,] return [returncode, executed, ]
executed.append(["write(sshkey) to", sshDir+"authorized_keys"]) executed.append(["write(sshkey) to", ssh_dir + "authorized_keys"])
with open(sshDir+"authorized_keys", "w") as f: with open(ssh_dir + "authorized_keys", "w") as f:
f.write(user["pubkey"]) f.write(user["pubkey"])
if f.closed != True: if not f.closed:
logging.exception("Could'nt write to authorized_keys!") logging.exception("Could'nt write to authorized_keys!")
return [rcode,executed,] return [returncode, executed, ]
executed.append(["chmod", "-Rv", "700", sshDir]) executed.append(["chmod", "-Rv", "700", ssh_dir])
try: try:
os.chmod(sshDir+"authorized_keys", 0o700) # directory is already 700 os.chmod(ssh_dir + "authorized_keys", 0o700) # directory is already 700
rcode = 0 returncode = 0
except OSError as e: except OSError as e:
logging.exception(e.strerror) logging.exception(e.strerror)
rcode = e.errno returncode = e.errno
return [rcode, executed,] return [returncode, executed, ]
try: try:
executed.append(["chown", "-Rv", username+":"+username, sshDir]) executed.append(["chown", "-Rv", username + ":" + username, ssh_dir])
os.chown(sshDir, pwd.getpwnam(username)[2], pwd.getpwnam(username)[3]) #2=>uid, 3=>gid os.chown(ssh_dir, pwd.getpwnam(username)[2], pwd.getpwnam(username)[3]) # 2=>uid, 3=>gid
executed.append(["chown", "-v", username+":"+username, sshDir+"authorized_keys"]) executed.append(["chown", "-v", username + ":" + username, ssh_dir + "authorized_keys"])
os.chown(sshDir+"authorized_keys", pwd.getpwnam(username)[2], pwd.getpwnam(username)[3]) os.chown(ssh_dir + "authorized_keys", pwd.getpwnam(username)[2], pwd.getpwnam(username)[3])
rcode = 0 returncode = 0
except OSError as e: except OSError as e:
logging.exception(e.strerror) # @TODO: maybe append strerror to executed instead of printing it logging.exception(e.strerror) # @TODO: maybe append strerror to executed instead of printing it
rcode = e.errno returncode = e.errno
return [rcode, executed,] return [returncode, executed, ]
return [rcode,executed,]
"""
{'id': 7, 'username': 'testuser47', 'email': '47test@testmail.com', 'name':
'test Name', 'pubkey': 'ssh-rsa [...]', 'timestamp': '2018-08-22 13:31:16', 'status': 0}
""" return [returncode, executed, ]
# {'id': 7, 'username': 'testuser47', 'email': '47test@testmail.com', 'name':
# 'test Name', 'pubkey': 'ssh-rsa [...]', 'timestamp': '2018-08-22 13:31:16', 'status': 0}
def main(): def main():
# how many times the Seperator/Delimiter? # how many times the Separator/Delimiter?
delcount = 40 delcount = 40
# The seperator for the menu # The separator for the menu
Seperator = "="*delcount separator = "=" * delcount
Menu = Seperator+"\n\t\t Main-Menu:\n\n" \ menu = separator + "\n\t\t Main-Menu:\n\n" \
"\t 1) list and edit pending users\n"\ "\t 1) list and edit pending users\n" \
"\t 2) list applicants\n"\ "\t 2) list applicants\n" \
"\t 3) edit applicant\n"\ "\t 3) edit applicant\n" \
"\t 4) quit\n"+Seperator+"\n" "\t 4) quit\n" + separator + "\n"
# Identify by ID # Identify by ID
applications = applicants(lident = "id") applications = Applicants(lident="id")
while 1 != 0: while 1 != 0:
print(Menu) print(menu)
command = input("Please select, what you want to do: \n -> ") command = input("Please select, what you want to do: \n -> ")
# User shouldnt be able to type something in that isnt a number # User shouldn't be able to type something in that isnt a number
if command.isalpha() or command == '': if command.isalpha() or command == '':
clear() clear()
print("!!! invalid input, please try again. !!!") print("!!! invalid input, please try again. !!!")
continue continue
# convert # convert
command=int(command) command = int(command)
if command == 4 or command == "q": if command == 4 or command == "q":
exit(0) exit(0)
# Edit and list pending users/applicants @TODO Wording: Users or applicants? # Edit and list pending users/applicants @TODO Wording: Users or applicants?
elif command == 1: elif command == 1:
users = applications.getApplicationsList() users = applications.getapplicationslist()
i=applications.printApprovableUsers(users) i = applications.printapprovableusers(users)
if i == 0 : if i == 0:
print("No pending users") print("No pending users")
# giving some time to aknowledge that something WRONG happened # giving some time to acknowledge that something WRONG happened
input("Continue with Keypress...") input("Continue with Keypress...")
clear() clear()
continue continue
usersel = 0 user_selection = 0
UserMax = i user_max = i
print("Menu:\n r=>return to main") print("Menu:\n r=>return to main")
# Edit Menue # Edit Menu
while 1 != 0 or usersel != "r": while 1 != 0 or user_selection != "r":
i = applications.printApprovableUsers(users) i = applications.printapprovableusers(users)
if usersel == "r": if user_selection == "r":
break # break when user presses r break # break when user presses r
usersel = input("Which user( ID ) do you want to change? ->") user_selection = input("Which user( ID ) do you want to change? ->")
if len(usersel) > 1 or usersel.isalpha(): if len(user_selection) > 1 or user_selection.isalpha():
usersel = "" user_selection = ""
# convert to int if input isnt an r # convert to int if input isnt an r
usersel = int(usersel) if usersel != '' and usersel != 'r' else 0 user_selection = int(user_selection) if user_selection != '' and user_selection != 'r' else 0
if usersel > UserMax - 1: if user_selection > user_max - 1:
print("User {0!s} doesn't exist!".format(usersel)) print("User {0!s} doesn't exist!".format(user_selection))
continue continue
# Show the user his chosen user and ask what to do # Show the user his chosen user and ask what to do
applications.userPrint(users, usersel) applications.userprint(users, user_selection)
print("You chosed ID No. {0!s}, what do you like to do?".format(usersel)) print("You chosed ID No. {0!s}, what do you like to do?".format(user_selection))
chosenUser = usersel chosen_user = user_selection
usersel = "" user_selection = ""
# Finally down the edit menue! # Finally down the edit menu!
while usersel != "e": while user_selection != "e":
usersel = input("User: {0!s}\n \t\t(A)ctivate \n\t\t(R)emove \n\t\tR(e)turn\n -> ".format(chosenUser)) user_selection = input(
if usersel == "A": "User: {0!s}\n \t\t(A)ctivate \n\t\t(R)emove \n\t\tR(e)turn\n -> ".format(chosen_user))
applications.approveApplicant(users[chosenUser]['id']) if user_selection == "A":
print("User {0!s} has been successfully approved!".format(users[chosenUser]['username'])) applications.approveapplicant(users[chosen_user]['id'])
print("User {0!s} has been successfully approved!".format(users[chosen_user]['username']))
input("waiting for input...") input("waiting for input...")
clear() clear()
usersel="e" # remove for being able to continue editing? user_selection = "e" # remove for being able to continue editing?
continue continue
elif usersel == "R": elif user_selection == "R":
applications.removeApplicant(users[chosenUser]['id']) applications.removeapplicant(users[chosen_user]['id'])
print("User {0!s} successfully deleted!".format(user[chosenUser]['username'])) print("User {0!s} successfully deleted!".format(user[chosen_user]['username']))
input("waiting for input...") input("waiting for input...")
clear() clear()
continue continue
elif usersel == "e": elif user_selection == "e":
clear() clear()
continue continue
elif int(command) == 2: elif int(command) == 2:
users = applications.getApprovedApplicantsList() users = applications.getapprovedapplicantslist()
if users == []: if not users:
print("no activate users yet!") print("no activate users yet!")
i=0 i = 0
for user in users: for user in users:
print("ID: {0!s}, Status: {1!s}, Name: {2!s}".format(user["id"], user["status"], user["username"])) print("ID: {0!s}, Status: {1!s}, Name: {2!s}".format(user["id"], user["status"], user["username"]))
continue continue
@ -381,11 +387,12 @@ def main():
else: else:
exit(0) exit(0)
if __name__ == "__main__": if __name__ == "__main__":
try: try:
main() main()
exit(0) exit(0)
except KeyboardInterrupt: except KeyboardInterrupt:
pass pass
#print("Exception occured. View log file for details.") # print("Exception occured. View log file for details.")
#logging.exception("Some exception occured") # logging.exception("Some exception occured")

77
public/userapplication.py
Unescape Escape View File

@ -14,62 +14,69 @@ try:
else: else:
if ospath.isfile(cwd) is False: if ospath.isfile(cwd) is False:
cwd=getcwd()+"/applicationsconfig.ini" cwd=getcwd()+"/applicationsconfig.ini"
# cwd is now either cwd/applicationsconfig or $TILDE_CONF # cwd is now either cwd/applicationsconfig or $TILDE_CONF
argparser = argparse.ArgumentParser(description='interactive registration formular for tilde platforms') argparser = argparse.ArgumentParser(description='interactive registration formular for tilde platforms')
argparser.add_argument('-c', '--config', default=cwd, type=str, help='Config file', required=False) argparser.add_argument('-c', '--config', default=cwd, type=str, help='Config file', required=False)
args = argparser.parse_args() args = argparser.parse_args()
CONF_FILE=args.config CONF_FILE = args.config
except: except:
# intended broad, @TODO check them all for errors instead of everything in one
logging.exception("Argumentparser-Exception: ") logging.exception("Argumentparser-Exception: ")
exit(0)
try: try:
config = configparser.ConfigParser() config = configparser.ConfigParser()
config.read(CONF_FILE) config.read(CONF_FILE)
logging.basicConfig(format="%(asctime)s: %(message)s", filename=config['DEFAULT']['log_file'],level=int(config['LOG_LEVEL']['log_level'])) logging.basicConfig(format="%(asctime)s: %(message)s", filename=config['DEFAULT']['log_file'],
del(cwd) level=int(config['LOG_LEVEL']['log_level']))
REG_FILE=config['DEFAULT']['applications_db'] REG_FILE = config['DEFAULT']['applications_db']
except: except:
# intended broad, @TODO check them all for errors instead of everything in one
logging.exception("logging or configparser-Exception: ") logging.exception("logging or configparser-Exception: ")
exit(0)
VALID_SSH=False VALID_SSH = False
VALID_USER=False VALID_USER = False
def __createTable(cursor, connection): def __createTable(cursor, connection):
try: try:
cursor.execute( cursor.execute(
"CREATE TABLE IF NOT EXISTS applications(" \ "CREATE TABLE IF NOT EXISTS applications("
"id INTEGER PRIMARY KEY AUTOINCREMENT,"\ "id INTEGER PRIMARY KEY AUTOINCREMENT,"
"username TEXT NOT NULL, email TEXT NOT NULL,"\ "username TEXT NOT NULL, email TEXT NOT NULL,"
"name TEXT NOT NULL, pubkey TEXT NOT NULL,"\ "name TEXT NOT NULL, pubkey TEXT NOT NULL,"
"timestamp DATETIME DEFAULT CURRENT_TIMESTAMP, status INTEGER NOT NULL DEFAULT 0);") "timestamp DATETIME DEFAULT CURRENT_TIMESTAMP, status INTEGER NOT NULL DEFAULT 0);")
connection.commit() connection.commit()
except: except sqlite3.Error as e:
logging.exception("Couldn't create needed SQLite Table!") logging.exception("Couldn't create needed SQLite Table! Exception: %s" % e)
def addtotable(cursor, connection, username, name, email, pubkey): def addtotable(cursor, connection, username, name, email, pubkey):
try: try:
cursor.execute("INSERT INTO 'applications'(username, name, email, pubkey)VALUES("\ cursor.execute("INSERT INTO 'applications'(username, name, email, pubkey)VALUES("
"?,?,?,?)", [username, name, email, pubkey]) "?,?,?,?)", [username, name, email, pubkey])
connection.commit() connection.commit()
except: except sqlite3.Error as e:
logging.exception("Couldn't insert user into the db") logging.exception("Couldn't insert user into the db: %s" % e)
# check if sqlite file does exists or already and has our structure # check if sqlite file does exists or already and has our structure
def __checkSQLite(cursor, connection): def __checkSQLite(cursor, connection):
#SELECT name FROM sqlite_master WHERE type='table' AND name='{table_name}'; # SELECT name FROM sqlite_master WHERE type='table' AND name='{table_name}';
cursor.execute("SELECT name FROM sqlite_master WHERE type='table' AND name='applications'") cursor.execute("SELECT name FROM sqlite_master WHERE type='table' AND name='applications'")
res=cursor.fetchall() res = cursor.fetchall()
if res== []: if not res:
try: try:
__createTable(cursor, connection) __createTable(cursor, connection)
except: except sqlite3.Error as e:
logging.exception("couldn't create table on given database. Exception: ") logging.exception("couldn't create table on given database. Exception: %s" % e)
else: else:
pass pass
return True return True
def check_username(value): def check_username(value):
global VALID_USER global VALID_USER
if len(value) < 3: if len(value) < 3:
@ -78,9 +85,10 @@ def check_username(value):
try: try:
from pwd import getpwnam from pwd import getpwnam
getpwnam(value) getpwnam(value)
VALID_USER=False VALID_USER = False
except: # intended broad
VALID_USER=True except Exception:
VALID_USER = True
return True return True
return False return False
@ -111,38 +119,34 @@ def validate_pubkey(value):
return True return True
def main(): def main():
print(" ▗▀▖ \n▗▖▖ ▐ ▌ ▌▛▀▖\n▘▝▗▖▜▀ ▌ ▌▌ ▌\n ▝▘▐ ▝▀▘▘ ▘") print(" ▗▀▖ \n▗▖▖ ▐ ▌ ▌▛▀▖\n▘▝▗▖▜▀ ▌ ▌▌ ▌\n ▝▘▐ ▝▀▘▘ ▘")
username = input("Welcome to the ~.fun user application form!\n\nWhat is your desired username? [a-z0-9] allowed:\n") username = input("Welcome to the ~.fun user application form!\n\nWhat is your desired username? [a-z0-9] allowed:\n")
while (not re.match("[a-z]+[a-z0-9]", username)) or (not check_username(username)): while (not re.match("[a-z]+[a-z0-9]", username)) or (not check_username(username)):
username = input("Invalid Username, maybe it exists already?\nValid characters are only a-z and 0-9.\nMake sure your username starts with a character and not a number." \ username = input("Invalid Username, maybe it exists already?\nValid characters are only a-z and 0-9."
"\nMake sure your username starts with a character and not a number."
"\nWhat is your desired username? [a-z0-9] allowed:\n") "\nWhat is your desired username? [a-z0-9] allowed:\n")
fullname = input("\nPlease enter your full name:\n") fullname = input("\nPlease enter your full name:\n")
while not re.match("\w+\s*\w*", username): while not re.match("\w+\s*\w*", username):
fullname = input("\nThat is not your real name.\nPlease enter your full name:\n") fullname = input("\nThat is not your real name.\nPlease enter your full name:\n")
email = input("\nPlease enter your email address:\n") email = input("\nPlease enter your email address:\n")
while not re.match("(^[a-zA-Z0-9_.+-]+@[a-zA-Z0-9-]+\.[a-zA-Z0-9-.]+$)", email): while not re.match("(^[a-zA-Z0-9_.+-]+@[a-zA-Z0-9-]+\.[a-zA-Z0-9-.]+$)", email):
email = input("\nThat is not a valid mail address.\nPlease enter your email address:\n") email = input("\nThat is not a valid mail address.\nPlease enter your email address:\n")
pubkey = input("\nPlease paste your ssh public key:\n") pubkey = input("\nPlease paste your ssh public key:\n")
while (not re.match("ssh-(\w)+\s(\w+)(\s*)([a-zA-Z0-9@]*)", pubkey)) or (not validate_pubkey(pubkey)): while (not re.match("ssh-(\w)+\s(\w+)(\s*)([a-zA-Z0-9@]*)", pubkey)) or (not validate_pubkey(pubkey)):
pubkey = input("\nPlease enter a valid public key. You can show it with ssh-keygen -f .ssh/id_rsa -y on your local machine.\nPlease enter your pubkey:\n") pubkey = input("\nPlease enter a valid public key. You can show it with ssh-keygen -f .ssh/id_rsa -y on your "
"local machine.\nPlease enter your pubkey:\n")
validate_pubkey(pubkey) validate_pubkey(pubkey)
print("\nUsername: {0!s}".format(username)) print("\nUsername: {0!s}".format(username))
print("Full Name: {0!s}".format(fullname)) print("Full Name: {0!s}".format(fullname))
print("Email: {0!s}".format(email)) print("Email: {0!s}".format(email))
print("Public {0!s}".format(pubkey)) print("Public {0!s}".format(pubkey))
validation = input("\nIs this information correct? [y/N]") validation = input("\nIs this information correct? [y/N]")
while not re.match("[yYnN\n]", validation): while not re.match("[yYnN\n]", validation):
print("Please answer y for yes or n for no") print("Please answer y for yes or n for no")
@ -156,13 +160,16 @@ def main():
addtotable(cursor, connection, username, fullname, email, pubkey) addtotable(cursor, connection, username, fullname, email, pubkey)
connection.commit() connection.commit()
connection.close() connection.close()
except: except sqlite3.Error as e:
logging.exception("Database {0!s} couldnt be accessed or created. Exception:".format(config['DEFAULT']['applications_db'])) logging.exception("Database {0!s} couldnt be accessed or created. Exception: {0!s}".
format(config['DEFAULT']['applications_db'], e))
if connection:
connection.close() connection.close()
exit(1) exit(1)
pass
return 0 return 0
if __name__ == "__main__": if __name__ == "__main__":
try: try:
main() main()

Write Preview
Loading…
Cancel
Save