popcorn-client

   1 #!/usr/bin/python
   2
   3 # Copyright (c) 2009 Pavol Rusnak <stick@gk2.sk>
   4 #
   5 # Permission is hereby granted, free of charge, to any person
   6 # obtaining a copy of this software and associated documentation
   7 # files (the "Software"), to deal in the Software without
   8 # restriction, including without limitation the rights to use,
   9 # copy, modify, merge, publish, distribute, sublicense, and/or sell
  10 # copies of the Software, and to permit persons to whom the
  11 # Software is furnished to do so, subject to the following
  12 # conditions:
  13 #
  14 # The above copyright notice and this permission notice shall be
  15 # included in all copies or substantial portions of the Software.
  16 #
  17 # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  18 # EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
  19 # OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  20 # NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
  21 # HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
  22 # WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  23 # FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
  24 # OTHER DEALINGS IN THE SOFTWARE.
  25
  26 import os
  27 import re
  28 import rpm
  29 import platform
  30 import httplib
  31 import time
  32
  33 version        = '0.1'
  34 popcorn_server = 'stats.opensuse.org'
  35
  36 days_recent = 30
  37 days_voted  = 30
  38
  39 pathre = '^/(bin|boot|lib|lib64|opt/gnome/bin|opt/gnome/sbin|opt/kde3/bin|opt/kde3/sbin|sbin|usr/bin|usr/games|usr/include|usr/lib|usr/lib64|usr/libexec|usr/sbin|usr/share)/'
  40
  41 class Statistics:
  42     arch = platform.machine()
  43     packages = []
  44
  45     def fill_packages(self):
  46         pathreprg = re.compile(pathre)
  47         now = int(time.time())
  48         ts = rpm.TransactionSet()
  49         mi = ts.dbMatch()
  50         for h in mi:
  51             name = h['name']
  52             if name == 'gpg-pubkey': continue
  53             installed = h[rpm.RPMTAG_INSTALLTIME]
  54             (accessed, atime) = (0, 0)
  55             for file in h[rpm.RPMTAG_FILENAMES]:
  56                 if not pathreprg.match(file): continue
  57                 try:
  58                     (atime, ) = os.stat(file)[7:8]
  59                     if atime > accessed:
  60                         accessed = atime
  61                 except:
  62                     pass
  63             if accessed == 0:
  64                 cat = 'n' # no-files
  65             else:
  66                 if now - installed < days_recent * 86400:
  67                     cat = 'r' # recent
  68                 else:
  69                     if now - accessed < days_voted * 86400:
  70                         cat = 'v' # voted
  71                     else:
  72                         cat ='o' # old
  73             self.packages.append( (cat, name) )
  74
  75     def serialize(self):
  76         ret  = ''
  77         ret += 'POPCORN %s %s\n' % (version, self.arch)
  78         for pkg in self.packages:
  79             ret += '%s %s\n' % pkg
  80         return ret
  81
  82 class Client:
  83     def submit(self, body, compress = False):
  84         conn = httplib.HTTPConnection(popcorn_server)
  85         conn.putrequest('POST', '/popcorn')
  86         conn.putheader('content-type', 'application/octet-stream')
  87         conn.putheader('content-length', str(len(body)))
  88         conn.endheaders()
  89         conn.send(body)
  90
  91 stats = Statistics()
  92 stats.fill_packages()
  93
  94 # client = Client()
  95 # client.submit( stats.serialize() )
  96
  97 print stats.serialize(),