Plugin to manage optional files
This commit is contained in:
shortcutme
parent
b3ba79a0fd
commit
790dc4e3d4
4 changed files with 754 additions and 0 deletions
257
plugins/OptionalManager/UiWebsocketPlugin.py
Normal file
257
plugins/OptionalManager/UiWebsocketPlugin.py
Normal file
|
|
@ -0,0 +1,257 @@
|
|||
import re
|
||||
import time
|
||||
import cgi
|
||||
|
||||
import gevent
|
||||
|
||||
from Plugin import PluginManager
|
||||
from Config import config
|
||||
from util import helper
|
||||
|
||||
|
||||
@PluginManager.registerTo("UiWebsocket")
|
||||
class UiWebsocketPlugin(object):
|
||||
def __init__(self, *args, **kwargs):
|
||||
self.time_peer_numbers_updated = 0
|
||||
super(UiWebsocketPlugin, self).__init__(*args, **kwargs)
|
||||
|
||||
def actionFileWrite(self, to, inner_path, *args, **kwargs):
|
||||
super(UiWebsocketPlugin, self).actionFileWrite(to, inner_path, *args, **kwargs)
|
||||
|
||||
# Add file to content.db and set it as pinned
|
||||
content_db = self.site.content_manager.contents.db
|
||||
content_db.my_optional_files[self.site.address + "/" + inner_path] = time.time()
|
||||
if len(content_db.my_optional_files) > 50: # Keep only last 50
|
||||
oldest_key = min(
|
||||
content_db.my_optional_files.iterkeys(),
|
||||
key=(lambda key: content_db.my_optional_files[key])
|
||||
)
|
||||
del content_db.my_optional_files[oldest_key]
|
||||
|
||||
def updatePeerNumbers(self):
|
||||
content_db = self.site.content_manager.contents.db
|
||||
content_db.updatePeerNumbers()
|
||||
self.site.updateWebsocket(peernumber_updated=True)
|
||||
|
||||
# Optional file functions
|
||||
|
||||
def actionOptionalFileList(self, to, address=None, orderby="time_downloaded DESC", limit=10):
|
||||
if not address:
|
||||
address = self.site.address
|
||||
|
||||
# Update peer numbers if necessary
|
||||
content_db = self.site.content_manager.contents.db
|
||||
if time.time() - content_db.time_peer_numbers_updated > 60 * 1 and time.time() - self.time_peer_numbers_updated > 60 * 5:
|
||||
# Start in new thread to avoid blocking
|
||||
self.time_peer_numbers_updated = time.time()
|
||||
gevent.spawn(self.updatePeerNumbers)
|
||||
|
||||
if address != self.site.address and "ADMIN" not in self.site.settings["permissions"]:
|
||||
return self.response(to, "optionalSiteInfo not allowed on this site")
|
||||
|
||||
if not all([re.match("^[a-z_*/+-]+( DESC| ASC|)$", part.strip()) for part in orderby.split(",")]):
|
||||
return self.response(to, "Invalid order_by")
|
||||
|
||||
if type(limit) != int:
|
||||
return self.response(to, "Invalid limit")
|
||||
|
||||
back = []
|
||||
content_db = self.site.content_manager.contents.db
|
||||
site_id = content_db.site_ids[address]
|
||||
query = "SELECT * FROM file_optional WHERE site_id = %s AND is_downloaded = 1 ORDER BY %s LIMIT %s" % (site_id, orderby, limit)
|
||||
for row in content_db.execute(query):
|
||||
back.append(dict(row))
|
||||
self.response(to, back)
|
||||
|
||||
def actionOptionalFileInfo(self, to, inner_path):
|
||||
content_db = self.site.content_manager.contents.db
|
||||
site_id = content_db.site_ids[self.site.address]
|
||||
|
||||
# Update peer numbers if necessary
|
||||
if time.time() - content_db.time_peer_numbers_updated > 60 * 1 and time.time() - self.time_peer_numbers_updated > 60 * 5:
|
||||
# Start in new thread to avoid blocking
|
||||
self.time_peer_numbers_updated = time.time()
|
||||
gevent.spawn(self.updatePeerNumbers)
|
||||
|
||||
query = "SELECT * FROM file_optional WHERE site_id = :site_id AND inner_path = :inner_path LIMIT 1"
|
||||
res = content_db.execute(query, {"site_id": site_id, "inner_path": inner_path})
|
||||
row = next(res, None)
|
||||
if row:
|
||||
self.response(to, dict(row))
|
||||
else:
|
||||
self.response(to, None)
|
||||
|
||||
def setPin(self, inner_path, is_pinned, address=None):
|
||||
if not address:
|
||||
address = self.site.address
|
||||
|
||||
if not self.hasSitePermission(address):
|
||||
return {"error": "Forbidden"}
|
||||
|
||||
site = self.server.sites[address]
|
||||
|
||||
content_db = site.content_manager.contents.db
|
||||
site_id = content_db.site_ids[site.address]
|
||||
content_db.execute("UPDATE file_optional SET is_pinned = %s WHERE ?" % is_pinned, {"site_id": site_id, "inner_path": inner_path})
|
||||
|
||||
return "ok"
|
||||
|
||||
def actionOptionalFilePin(self, to, inner_path, address=None):
|
||||
back = self.setPin(inner_path, 1, address)
|
||||
if back == "ok":
|
||||
self.cmd("notification", ["done", "Pinned %s files" % len(inner_path) if type(inner_path) is list else 1, 5000])
|
||||
self.response(to, back)
|
||||
|
||||
def actionOptionalFileUnpin(self, to, inner_path, address=None):
|
||||
back = self.setPin(inner_path, 0, address)
|
||||
if back == "ok":
|
||||
self.cmd("notification", ["done", "Removed pin from %s files" % len(inner_path) if type(inner_path) is list else 1, 5000])
|
||||
self.response(to, back)
|
||||
|
||||
def actionOptionalFileDelete(self, to, inner_path, address=None):
|
||||
if not address:
|
||||
address = self.site.address
|
||||
|
||||
if not self.hasSitePermission(address):
|
||||
return self.response(to, {"error": "Forbidden"})
|
||||
|
||||
site = self.server.sites[address]
|
||||
|
||||
content_db = site.content_manager.contents.db
|
||||
site_id = content_db.site_ids[site.address]
|
||||
|
||||
res = content_db.execute("SELECT * FROM file_optional WHERE ? LIMIT 1", {"site_id": site_id, "inner_path": inner_path})
|
||||
row = next(res, None)
|
||||
|
||||
if not row:
|
||||
return self.response(to, {"error": "Not found in content.db"})
|
||||
|
||||
removed = site.content_manager.optionalRemove(inner_path, row["hash_id"], row["size"])
|
||||
# if not removed:
|
||||
# return self.response(to, {"error": "Not found in hash_id: %s" % row["hash_id"]})
|
||||
|
||||
content_db.execute("UPDATE file_optional SET is_downloaded = 0, is_pinned = 0, peer = peer - 1 WHERE ?", {"site_id": site_id, "inner_path": inner_path})
|
||||
|
||||
try:
|
||||
site.storage.delete(inner_path)
|
||||
except Exception, err:
|
||||
return self.response(to, {"error": "File delete error: %s" % err})
|
||||
|
||||
self.response(to, "ok")
|
||||
|
||||
|
||||
# Limit functions
|
||||
|
||||
def actionOptionalLimitStats(self, to):
|
||||
if "ADMIN" not in self.site.settings["permissions"]:
|
||||
return self.response(to, "Forbidden")
|
||||
|
||||
back = {}
|
||||
back["limit"] = config.optional_limit
|
||||
back["used"] = self.site.content_manager.contents.db.execute(
|
||||
"SELECT SUM(size) FROM file_optional WHERE is_downloaded = 1 AND is_pinned = 0"
|
||||
).fetchone()[0]
|
||||
back["free"] = helper.getFreeSpace()
|
||||
|
||||
self.response(to, back)
|
||||
|
||||
def actionOptionalLimitSet(self, to, limit):
|
||||
if "ADMIN" not in self.site.settings["permissions"]:
|
||||
return self.response(to, {"error": "Forbidden"})
|
||||
config.optional_limit = re.sub("\.0+$", "", limit) # Remove unnecessary digits from end
|
||||
config.saveValue("optional_limit", limit)
|
||||
self.response(to, "ok")
|
||||
|
||||
# Distribute help functions
|
||||
|
||||
def actionOptionalHelpList(self, to, address=None):
|
||||
if not address:
|
||||
address = self.site.address
|
||||
|
||||
if not self.hasSitePermission(address):
|
||||
return self.response(to, {"error": "Forbidden"})
|
||||
|
||||
site = self.server.sites[address]
|
||||
|
||||
self.response(to, site.settings.get("optional_help", {}))
|
||||
|
||||
def actionOptionalHelp(self, to, directory, title, address=None):
|
||||
if not address:
|
||||
address = self.site.address
|
||||
|
||||
if not self.hasSitePermission(address):
|
||||
return self.response(to, {"error": "Forbidden"})
|
||||
|
||||
site = self.server.sites[address]
|
||||
content_db = site.content_manager.contents.db
|
||||
site_id = content_db.site_ids[address]
|
||||
|
||||
if "optional_help" not in site.settings:
|
||||
site.settings["optional_help"] = {}
|
||||
|
||||
stats = content_db.execute(
|
||||
"SELECT COUNT(*) AS num, SUM(size) AS size FROM file_optional WHERE site_id = :site_id AND inner_path LIKE :inner_path",
|
||||
{"site_id": site_id, "inner_path": directory + "%"}
|
||||
).fetchone()
|
||||
stats = dict(stats)
|
||||
|
||||
if not stats["size"]:
|
||||
stats["size"] = 0
|
||||
if not stats["num"]:
|
||||
stats["num"] = 0
|
||||
|
||||
self.cmd("notification", [
|
||||
"done",
|
||||
"You started to help distribute <b>%s</b>.<br><small>Directory: %s</small>" %
|
||||
(cgi.escape(title), cgi.escape(directory)),
|
||||
10000
|
||||
])
|
||||
|
||||
site.settings["optional_help"][directory] = title
|
||||
|
||||
self.response(to, dict(stats))
|
||||
|
||||
def actionOptionalHelpRemove(self, to, directory, address=None):
|
||||
if not address:
|
||||
address = self.site.address
|
||||
|
||||
if not self.hasSitePermission(address):
|
||||
return self.response(to, {"error": "Forbidden"})
|
||||
|
||||
site = self.server.sites[address]
|
||||
|
||||
try:
|
||||
del site.settings["optional_help"][directory]
|
||||
self.response(to, "ok")
|
||||
except Exception:
|
||||
self.response(to, {"error": "Not found"})
|
||||
|
||||
def cbOptionalHelpAll(self, to, site, value):
|
||||
site.settings["autodownloadoptional"] = value
|
||||
self.response(to, value)
|
||||
|
||||
def actionOptionalHelpAll(self, to, value, address=None):
|
||||
if not address:
|
||||
address = self.site.address
|
||||
|
||||
if not self.hasSitePermission(address):
|
||||
return self.response(to, {"error": "Forbidden"})
|
||||
|
||||
site = self.server.sites[address]
|
||||
|
||||
if value:
|
||||
if "ADMIN" in self.site.settings["permissions"]:
|
||||
self.cbOptionalHelpAll(to, site, True)
|
||||
else:
|
||||
site_title = site.content_manager.contents["content.json"].get("title", address)
|
||||
self.cmd(
|
||||
"confirm",
|
||||
[
|
||||
"Help distribute all new optional files on site <b>%s</b>" % cgi.escape(site_title),
|
||||
"Yes, I want to help!"
|
||||
],
|
||||
lambda (res): self.cbOptionalHelpAll(to, site, True)
|
||||
)
|
||||
else:
|
||||
site.settings["autodownloadoptional"] = False
|
||||
self.response(to, False)
|
||||
Loading…
Add table
Add a link
Reference in a new issue