92 lines
2.4 KiB
Python
92 lines
2.4 KiB
Python
|
#! /bin/python3
|
||
|
|
|
||
|
|
import socket
|
||
|
|
import os
|
||
|
|
|
||
|
|
import sys
|
||
|
|
import time
|
||
|
|
|
||
|
|
import apt_pkg
|
||
|
|
|
||
|
|
import apt
|
||
|
|
import apt.progress.base
|
||
|
|
|
||
|
|
def get_change(current, total):
|
||
|
|
if current == total:
|
||
|
|
return 100.0
|
||
|
|
try:
|
||
|
|
return float("{:.1f}".format(((current * 100) / total)))
|
||
|
|
except ZeroDivisionError:
|
||
|
|
return 0.0
|
||
|
|
|
||
|
|
class UpdateProgressSocket(apt.progress.base.AcquireProgress):
|
||
|
|
# Init
|
||
|
|
def __init__(self):
|
||
|
|
pass
|
||
|
|
|
||
|
|
# Start
|
||
|
|
def start(self):
|
||
|
|
self.current_bytes = 0
|
||
|
|
self.total_bytes = 0
|
||
|
|
print("Starting APT Cache Update.")
|
||
|
|
return super().start()
|
||
|
|
|
||
|
|
# Stop
|
||
|
|
def stop(self):
|
||
|
|
print("\nAPT Cache Update Complete!")
|
||
|
|
return super().stop()
|
||
|
|
|
||
|
|
# Progrss pulse
|
||
|
|
def pulse(self, owner):
|
||
|
|
# Calculate current progress percentage
|
||
|
|
progress_percent = get_change(self.current_bytes, self.total_bytes)
|
||
|
|
|
||
|
|
# apt_update_progress ipc sock
|
||
|
|
socket_path = "/tmp/pika_apt_update.sock"
|
||
|
|
|
||
|
|
# Create a Unix domain socket
|
||
|
|
with socket.socket(socket.AF_UNIX, socket.SOCK_STREAM) as client:
|
||
|
|
client.connect(socket_path)
|
||
|
|
# Send percentage to socket as UTF-8
|
||
|
|
client.sendall(str(progress_percent).encode('utf-8'))
|
||
|
|
|
||
|
|
#response = client.recv(1024)
|
||
|
|
#print(f"Received: {response.decode('utf-8')}")
|
||
|
|
|
||
|
|
return True
|
||
|
|
|
||
|
|
|
||
|
|
def fail(self, item):
|
||
|
|
print("Failure at: %s %s" % (item.uri, item.shortdesc))
|
||
|
|
|
||
|
|
def fetch(self, item):
|
||
|
|
print("Fetch: %s %s" % (item.uri, item.shortdesc))
|
||
|
|
|
||
|
|
def ims_hit(self, item):
|
||
|
|
print("Download: %s %s" % (item.uri, item.shortdesc))
|
||
|
|
|
||
|
|
def media_change(self, medium, drive):
|
||
|
|
print(f"Please insert medium {medium} in drive {drive}")
|
||
|
|
sys.stdin.readline()
|
||
|
|
# return False
|
||
|
|
|
||
|
|
def update_cache():
|
||
|
|
# First of all, open the cache
|
||
|
|
cache = apt.Cache()
|
||
|
|
# Now, lets update the package list
|
||
|
|
cache.update(UpdateProgressSocket())
|
||
|
|
# We need to re-open the cache because it needs to read the package list
|
||
|
|
cache.open(None)
|
||
|
|
# We need to re-open the cache because it needs to read the package list
|
||
|
|
for pkg in cache:
|
||
|
|
if pkg.is_upgradable:
|
||
|
|
print(f"{pkg.name} ({pkg.installed.version} -> {pkg.candidate.version})")
|
||
|
|
|
||
|
|
def process(data):
|
||
|
|
# Echo the input data
|
||
|
|
return data
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
|
||
|
|
update_cache()
|