Add metadata to database and UI

indexer.py

@@ -1,7 +1,9 @@
 #!/usr/bin/python3
 #/* vim:set ts=2 set noexpandtab */
 import json, uuid, hashlib, sqlite3, base64
-from flask import Flask, render_template, url_for, request
+import bencoder
+import requests
+from flask import Flask, render_template, url_for, request, send_file
 from werkzeug import secure_filename
 app = Flask(__name__)
 strings = None
@@ -27,6 +29,14 @@ def create():
 		else:
 			return render_template("create.html", language="english", categories=settings["categories"], strings=strings, errors=errors)
 
+@app.route("/download/<filename>")
+def download(filename):
+	connection = sqlite3.connect("torrentdb.sqlite")
+	c = connection.cursor()
+	c.execute("SELECT name FROM torrents WHERE fileid = :fileid", { 'fileid' : filename})
+	name = c.fetchone()[0]
+	return send_file("torrentFiles/" + filename, as_attachment=True, attachment_filename=name + ".torrent", conditional=True)
+
 @app.route("/search", methods=['GET'])
 def search():
 	global strings
@@ -46,19 +56,24 @@ def search():
 			if field is "q":
 				names = query.split(" ")
 				search_params += list(map(lambda x: "%" + x + "%", names))
-				search += " AND ".join(["name LIKE (?)"] * len(query.split(" ")))
+				search += " AND ".join(["torrents.name LIKE (?)"] * len(query.split(" ")))
 			elif field is "c":
 				search_params += query.split(" ")
-				search += " AND ".join(["category LIKE (?)"] * len(query.split(" ")))
+				search += " AND ".join(["torrents.category LIKE (?)"] * len(query.split(" ")))
 			elif field is "s":
 				search_params += query.split(" ")
-				search += " AND ".join(["subcategory LIKE (?)"] * len(query.split(" ")))
+				search += " AND ".join(["torrents.subcategory LIKE (?)"] * len(query.split(" ")))
 
 	print(search)
-	c.execute("SELECT fileid, name FROM torrents WHERE " + search, search_params)
+	c.execute("SELECT torrents.fileid, torrents.name, metadata.torrentsize FROM torrents LEFT JOIN metadata on metadata.fileid = torrents.fileid WHERE " + search, search_params)
 	results = c.fetchall()
 	return render_template("result.html", results=results, strings=strings, language="english", categories=settings["categories"])
 
+def scrapeAll():
+	TRACKER_URL = "http://tracker.lootbox.cf:6969/"
+	statedump = requests.get(TRACKER_URL + "stats" + "?mode=statedump")
+	return
+
 def init():
 	global strings
 	global settings
@@ -68,11 +83,13 @@ def init():
 	with open("settings.json") as settingsJson:
 		settings = json.load(settingsJson)
 	initDb()
+	scrapeAll()
 
 def initDb():
 	connection = sqlite3.connect("torrentdb.sqlite")
 	c = connection.cursor()
 	c.execute('CREATE TABLE IF NOT EXISTS torrents (fileid TEXT PRIMARY KEY NOT NULL, name TEXT NOT NULL, category TEXT NOT NULL, subcategory TEXT NOT NULL, description TEXT NOT NULL, audioquality_description TEXT NOT NULL, videoquality_description TEXT NOT NULL);')
+	c.execute('CREATE TABLE IF NOT EXISTS metadata (fileid TEXT PRIMARY KEY NOT NULL, created_by TEXT, creation_date TEXT, announce_url TEXT NOT NULL, source TEXT, torrentsize TEXT NOT NULL, name TEXT NOT NULL, private TEXT NOT NULL)')
 	connection.commit()
 	connection.close()
 
@@ -94,6 +111,13 @@ def createNewTorrent(reuqest):
 	h.update((str(uuid.uuid4()) + filename).encode())
 	safeFilename = h.hexdigest()
 	uploadfile.save("torrentFiles/" + safeFilename)
+	import bencoder
+	with open("torrentFiles/" + safeFilename, "rb") as content:
+		content = content.read()
+	
+	bcoded = bencoder.decode(content)
+	size = ((len(bcoded[b'info'][b'pieces']) / 20) * bcoded[b'info'][b'piece length']) / 1024 / 1024
+
 	print("=== CREATE NEW TORRENT FILE ===")
 	print( "Name:         " + request.form["name"] )
 	print( "Torrent file: " + safeFilename )
@@ -111,10 +135,33 @@ def createNewTorrent(reuqest):
 	newTFile = TorrentFile(safeFilename, name, category, subcategory, description, audioquality_description, videoquality_description)
 	connection = sqlite3.connect("torrentdb.sqlite")
 	newTFile.writeToDb(connection.cursor())
+	newTFile.metadata.writeToDb(connection.cursor())
 	connection.commit()
 	connection.close()
 	return ["Error1"]
 
+class Metadata():
+	def __init__(self, fileid):
+		with open("torrentFiles/" + fileid, "rb") as f:
+			torrent = f.read()
+		self.fileid = fileid
+		self.bcoded = bencoder.decode(torrent)
+		self.created_by = self.bcoded.get(b'created by', "")
+		self.creation_date = self.bcoded.get(b'creation date', "")
+		self.announce_url = self.bcoded.get(b'info', dict()).get(b'', "")
+		self.source = self.bcoded.get(b'info', dict()).get(b'source', "")
+		self.torrentsize = ((len(self.bcoded.get(b'info', dict()).get(b'pieces', "")) / 20) * self.bcoded.get(b'info', dict()).get(b'piece length'))
+		self.name = self.bcoded.get(b'info', dict()).get(b'name', "")
+		self.private = self.bcoded.get(b'info', dict()).get(b'private', "")
+
+	def writeToDb(self, cursor):
+		c = cursor
+		b64created_by = base64.b64encode(self.created_by)
+		b64announce_url = base64.b64encode(self.announce_url.encode()) if self.announce_url else ""
+		b64source = base64.b64encode(self.source.encode()) if self.source else ""
+		b64name = base64.b64encode(self.name)
+		c.execute("INSERT INTO metadata(fileid, created_by, creation_date, announce_url, source, torrentsize, name, private) VALUES(:fileid, :created_by, :creation_date, :announce_url, :source, :torrentsize, :name, :private)", { 'fileid' : self.fileid, 'created_by' : b64created_by, 'creation_date' : self.creation_date, 'announce_url' : b64announce_url, 'source' : b64source , 'torrentsize' : self.torrentsize, 'name' : b64name, 'private' : self.private})
+
 class TorrentFile():
 	fileid = None
 	name = None
@@ -131,6 +178,7 @@ class TorrentFile():
 		self.description = description
 		self.audioquality_description = audioquality_description
 		self.videoquality_description = videoquality_description
+		self.metadata = Metadata(fileid)
 
 	def writeToDb(self, cursor):
 		c = cursor
@@ -144,4 +192,4 @@ if __name__ == "__main__":
 	app.jinja_env.globals.update(getLocalString=getLocalString)
 	app.jinja_env.globals.update(json=json)
 	app.jinja_env.globals.update(sorted=sorted)
-	app.run(debug=True, host="127.0.0.1")
+	app.run(debug=True, host="0.0.0.0")

static/css/categories.css

@@ -0,0 +1,10 @@
+.inner.cover {
+	display: flex;
+	align-items: baseline;
+	flex-wrap: wrap;
+}
+
+.inner.cover > span {
+	width: 33%;
+	margin-bottom: 10px;
+}

static/js/create.js

@@ -80,19 +80,6 @@ function fillDetectedFilelist(file) {
 	reader.readAsArrayBuffer(file)
 }
 
-function getNextUnit(bSize) {
-	units = ["B", "kB", "MB", "GB", "TB", "PB", "To damn high"]
-	iters = 0
-	size = bSize
-	do {
-		size /= 1024
-		iters++
-	} while(size >= 1)
-	size *= 1024
-	if(iters > units.length) { iters = units.length }
-	return [size, units[iters - 1]]
-}
-
 // Fill a defined dropdown with values.
 // These values will be generated out of the categories.json
 function fillDropdownByValue(value, dropdownToFill) {

static/js/main.js

@@ -14,3 +14,17 @@ function getDescriptorByLocalString(language, localString){
 	}
 	return localString
 }
+
+function getNextUnit(bSize) {
+	units = ["B", "kB", "MB", "GB", "TB", "PB", "To damn high"]
+	iters = 0
+	size = bSize
+	do {
+		size /= 1024
+		iters++
+	} while(size >= 1)
+	size *= 1024
+	if(iters > units.length) { iters = units.length }
+	return [size, units[iters - 1]]
+}
+

templates/categories.html

@@ -5,9 +5,10 @@ vim: ts=2 noexpandtab
 {% block title %}{{ super() }} - Categories{% endblock%}
 {% set active_page = "categories" %}
 {% block content %}
-<p>
+<link href="{{ url_for("static", filename="css/categories.css") }}" rel="stylesheet">
-	<ul>
 	{% for category in categories %}
+	<span>
+		<ul>
 			<li>
 				<a href="search?c={{ category }}">{{ category }}</a>
 				<ul>
@@ -18,7 +19,7 @@ vim: ts=2 noexpandtab
 				{% endfor %}
 				</ul>
 			</li>
-		{% endfor %}
 		</ul>
-</p>
+	</span>
+	{% endfor %}
 {% endblock content%}

templates/result.html

@@ -6,6 +6,7 @@ vim: ts=2 noexpandtab
 {% set active_page = "search" %}
 {% block content %}
 <link href="{{ url_for("static", filename="css/result.css") }}" rel="stylesheet">
+<script src="{{ url_for("static", filename="js/main.js") }}"></script>
 	<table>
 		<tr>
 			<th>Name</th>
@@ -16,8 +17,8 @@ vim: ts=2 noexpandtab
 		</tr>
 		{% for result in results %}
 		<tr>
-			<td>{{ result[1] }}</td>
+			<td><a href="/download/{{ result[0] }}">{{ result[1] }}</a></td>
-				<td>test</td>
+			<td>{{ result[2]|int / 1000 / 1000 }}</td>
 			<td>N/A</td>
 			<td>N/A</td>
 			<td>N/A</td>