[ WIP ] Developing a GUI for the populator

populate_oc/GUI/DatabaseHandler.py

@@ -0,0 +1,87 @@
+#!/usr/bin/env python
+
+from tinydb import TinyDB, where
+
+class DatabaseHandler():
+	def __init__(self):
+		self.db = TinyDB('database.json')
+		self.admin_table = self.db.table('admin')
+		self.users_table = self.db.table('users')
+		self.files_table = self.db.table('files')
+		self.folders_table = self.db.table('folders')
+		self.shares_table = self.db.table('shares')
+
+	def insert_admin_values(self, login, password, host):
+		if (len(self.admin_table) == 0):
+			self.admin_table.insert({'login': login, 'password': password, 'host': host})
+		else:
+			self.admin_table.update({'login': login, 'password': password, 'host': host}, eids=[1])
+
+	def insert_user(self, login, password):
+		self.users_table.insert({'login': login, 'password': password})
+
+	def get_user_password(self, login):
+		return self.users_table.search(where('login') == login)[0]['password']
+
+	def insert_users(self, elements):
+		for user,password in elements:
+			self.users_table.insert({'login': login, 'password': password})
+
+	def insert_file(self, user, path, destination):
+		self.files_table.insert({'user': user, 'path':path, 'destination':destination})
+
+	def insert_files(self, elements):
+		for user, path, destination in elements:
+			self.files_table.insert({'user': user, 'path':path, 'destination':destination})
+
+	def insert_folder(self, user, destination):
+		self.folders_table.insert({'user': user, 'destination': destination})
+
+	def insert_folders(self, elements):
+		for user, path, destination in elements:
+			self.folders_table.insert({'user': user, 'destination': destination})
+
+	def insert_share(self, sharer, sharee, path, share_type):
+		self.shares_table.insert({'sharer': sharer, 'sharee': sharee, 'path':path, 'share_type':share_type })
+
+	def delete_all_users(self):
+		self.users_table.purge()
+
+	def delete_all_files(self):
+		self.files_table.purge()
+
+	def delete_all_folders(self):
+		self.folders_table.purge()
+
+	def delete_all_shares(self):
+		self.shares_table.purge()
+
+	def get_users_values(self):
+		if (len(self.users_table) == 0):
+			return None
+		else:
+			return self.users_table.all()
+
+	def get_files_values(self):
+		if (len(self.files_table) == 0):
+			return None
+		else:
+			return self.files_table.all()
+
+	def get_folders_values(self):
+		if (len(self.folders_table) == 0):
+			return None
+		else:
+			return self.folders_table.all()
+
+	def get_shares_values(self):
+		if (len(self.shares_table) == 0):
+			return None
+		else:
+			return self.shares_table.all()
+
+	def get_admin_values(self):
+		if (len(self.admin_table) == 0):
+			return None
+		else:
+			return self.admin_table.all()[0]

populate_oc/GUI/Populator.py

@@ -0,0 +1,180 @@
+#!/usr/bin/env python
+
+from termcolor import colored
+import sys
+sys.path.append('../pyocclient')
+import owncloud
+import datetime
+import random
+import shutil
+import os
+import logging
+
+class Populator:
+    """It populates owncloud"""
+
+    def __init__(self, host, oc_user, oc_password, number_users, number_groups):
+        logging.basicConfig(level=logging.ERROR)
+        self.logger = logging.getLogger(__name__)
+        self.oc = owncloud.Client(host)
+        self.users = {}
+        self.groups = []
+        self.users_groups = {}
+        self.oc_user = oc_user
+        self.oc_password = oc_password
+        #self.fill_users_array(number_users)
+        #self.fill_groups_array(number_groups)
+
+    '''
+    def check_connection(self):
+        try:
+            self.oc.login(self.oc_user, self.oc_password)
+            self.oc.mkdir('testdir')
+            self.oc.put_file('testdir/remotefile.txt', 'localfile.txt')
+            link_info = self.oc.share_file_with_link('testdir/remotefile.txt')
+            print (colored("Here is your link: " + link_info.get_link(), 'green'))
+            self.oc.delete('testdir')
+            self.oc.logout()
+        except:
+            print (colored("Connection failed", 'red'))
+    '''
+
+    def check_connection(self):
+        try:
+            self.oc.login(self.oc_user, self.oc_password)
+        except HTTPResponseError as e:
+            raise e
+
+    def fill_users_array(self, number):
+        for i in xrange(number):
+            current_user = 'user' + str(i)
+            self.users[current_user] = current_user
+
+    def fill_groups_array(self, number):
+        for i in xrange(number):
+            current_group = 'group' + str(i)
+            self.groups.append(current_group)
+
+    def pop_create_user(self, login, password):
+        self.oc.login(self.oc_user, self.oc_password)
+        self.oc.create_user(login, password)
+        self.logger.info('Creating user ' + login)
+        self.oc.logout()
+
+    def pop_delete_user(self, login):
+        self.oc.login(self.oc_user, self.oc_password)
+        self.oc.delete_user(login)
+        self.logger.info('Deleting user ' + login)
+        self.oc.logout()
+
+    def pop_upload_file(self, login, password, path, destination):
+        self.oc.login(login, password)
+        self.oc.put_file(destination, path)
+        self.oc.logout()
+
+    def pop_create_folder(self, login, password, destination):
+        self.oc.login(login, password)
+        self.oc.mkdir(destination)
+        self.oc.logout()
+
+    def pop_share_file(self, login, password, sharee, path, share_type):
+        self.oc.login(login, password)
+        if (share_type == '1'):
+            self.oc.share_file_with_user(path, sharee)
+        elif (share_type == '3'):
+            self.oc.share_file_with_link(path)
+        self.oc.logout()
+
+    def create_users(self):
+        try:
+            self.oc.login(self.oc_user, self.oc_password)
+            for user in self.users:
+                self.oc.create_user(user, self.users[user])
+                self.logger.info('Creating user ' + user)
+            self.oc.logout()
+            print (colored("Created users", 'green'))
+        except:
+            print (colored("Users couln't be created", 'red'))
+
+
+    def remove_users(self):
+        try:
+            self.oc.login(self.oc_user, self.oc_password)
+            for user in self.users:
+                self.oc.delete_user(user)
+            self.oc.logout()
+            print (colored("Users removed", 'green'))
+        except:
+            print (colored("Users couln't be removed", 'red'))
+
+    def remove_groups(self):
+        try:
+            self.oc.login(self.oc_user, self.oc_password)
+            for group in self.groups:
+                self.oc.delete_group(group)
+            self.oc.logout()
+            print (colored("Groups removed", 'green'))
+        except:
+            print (colored("Groups couln't be removed", 'red'))
+
+    def create_groups(self):
+        try:
+            self.oc.login(self.oc_user, self.oc_password)
+            for group in self.groups:
+                self.oc.create_group(group)
+                self.logger.info('Creating group ' + group)
+            self.oc.logout()
+            print (colored("Groups created", 'green'))
+        except:
+            print (colored("Groups couln't be created", 'red'))
+
+    def assign_group_to_users(self):
+        try:
+            self.oc.login(self.oc_user, self.oc_password)
+            for user in self.users:
+                self.users_groups[user] = random.choice(self.groups) 
+                self.oc.add_user_to_group(user, self.users_groups[user])
+            self.oc.logout()
+            print (colored("Every user assigned to a group", 'green'))
+        except:
+            print (colored("Groups couln't be assigned to users", 'red'))
+
+    def name_generator(self):
+        #r = random.Random(500)
+        one = ['AA', 'BB', 'CC', 'DD']
+        two = ['ZZ', 'YY', 'XX', 'WW']
+        return str(random.choice(one)) + str(random.choice(one)) +str(random.randint(1,999999))
+
+
+    def create_deep_structure(self, user, password, depth):
+        self.logger.info('Creating deep structure for ' + user)
+        self.oc.login(user, password)
+        current_path = ''
+        for i in xrange(depth):
+            current_path = current_path + '/' + self.name_generator() 
+            self.oc.mkdir(current_path)
+            self.oc.put_file(current_path + '/' + REMOTEFILE, LOCALFILE)
+        self.oc.logout()
+
+    def create_modified_textfile(self):
+        new_name = self.name_generator() + '.txt'
+        shutil.copy2(LOCALFILE, new_name)
+        modified_localfile = open (new_name, 'w')
+        with open(new_name, "a") as modified_localfile:
+            modified_localfile.write(new_name)
+        return new_name
+
+    def remove_modified_textfile(self, filename):
+        if os.path.isfile(filename):
+            os.remove(filename)
+        else:    
+            print("Error: %s file not found" % filename)
+
+    def upload_many_textfiles(self, user, password, number):
+        self.oc.login(user, password)
+        self.oc.mkdir(FOLDER)
+        for i in xrange(number):
+            current_filename = self.create_modified_textfile()
+            self.oc.put_file(FOLDER + '/' + current_filename, current_filename)
+            self.remove_modified_textfile(current_filename)
+        self.oc.logout()

populate_oc/GUI/Readme.md

@@ -0,0 +1,25 @@
+
+This requires python3, pyqt5 and Qt creator to modify the interface.
+
+Install steps required (OS X):
+
+xcode-select --install
+brew install python3
+pip3 install -r requirements.txt
+
+
+**Development**
+To compile .ui files do
+
+```
+pyuic5 -o ui_oc_populator.py oc_populator.ui
+```
+
+**Use**
+
+```
+python3 oc_populator.py &
+```
+
+
+