collections

1 files changed, 157 insertions, 0 deletions

diff --git a/modules/language/python/module/collections.scm b/modules/language/python/module/collections.scm
new file mode 100644
index 0000000..bfb75fa
--- /dev/null
+++ b/modules/language/python/module/collections.scm
@@ -0,0 +1,157 @@
+(define-module (language python module collections)
+  #:use-module (oop pf-objects)
+  #:use-module (language python module collections abc)
+  #:export (ChainMap))
+
+(define (u self)
+  (let ((s (set)))
+    (apply (ref s 'union) (ref self 'maps)))
+  s)
+
+(define-python-class ChainMap (MutableMapping)
+  (define __init__
+    (lambda (self . l)
+      (set self 'maps
+	   (if (null? l)
+	       (list (dict))
+	       l))))
+
+  (define __getitem__
+    (lambda (self key)
+      (let/ec ret
+	 (let lp ((l (ref self 'maps)))
+	   (if (pair? l)
+	       (let ((m (car l)))
+		 (if (in key m)
+		     (pylist-ref m key)
+		     (lp (cdr l))))
+	       (raise (KeyError key)))))))
+  
+
+  (define get
+    (lambda* (self key #:key (default None))
+      (if (in key self)
+	  (pylist-ref self key)
+	  default)))
+
+  (define __len__
+    (lambda (self)
+      (len (u self))))
+
+
+  (define __iter__
+    (lambda (self)
+      (iter (u self))))
+
+  (define __contains__
+    (lambda (self, key)
+      (for ((m : (ref self 'maps)))
+	   (if (in key m)
+	       (break #t))
+	   #:final #f)))
+  
+  (define __bool__
+    (lambda (self)
+      (any (ref self 'maps))))
+
+  (define __repr__
+    (lambda (self)
+      (format #f "~a(~a,~{,~a~})"
+	      (ref (ref self '__class__) '__name__)
+	      (map (lambda (x) (repr x)) (ref self 'maps)))))
+
+  (define fromkeys
+    (class-method
+     (lambda (cls iter . args)
+       (apply cls (for ((x : iter)) ((l '()))
+		       (cons (apply py-fromkeys x args) l)
+		       
+		       #:final
+		       (reverse l))))))
+
+  (define copy
+    (lambda (self)
+      (let ((maps (ref self 'maps)))
+	(apply (ref self '__class__) (copy (car maps)) (cdr maps)))))
+
+  (define __copy__
+    (lambda (self)
+      ((ref self 'copy))))
+
+  (define new_child
+    (lambda* (self #:optional (r #f))
+	     (apply (ref self '__class__) (if r r (set)) (ref self 'maps))))
+
+  (define parents
+    (property
+     (lambda (self)
+       (apply (ref self '__class__) (cdr (ref self 'maps))))))
+  
+  (define __setitem__
+    (lambda (self key value)
+      (pylist-set! (car (ref self 'maps)) key value)))
+    
+
+  (define __delitem__
+    (lambda (self, key)
+      (try
+       (lambda () (pylist-del! (car (ref self 'maps))))
+       #:except KeyError =>
+       (lambda x
+	 (raise KeyError
+		(format #f "Key not found in the first mapping: ~a" key))))))
+
+  (define popitem
+    (lambda (self)
+      (try
+       (lambda () (popitem (car (ref self 'maps))))	 
+       #:except KeyError =>
+       (lambda x
+	 (raise KeyError "No keys found in the first mapping")))))
+
+  (define pop
+    (lambda (self key . args)
+      (try
+       (lambda () (apply py-pop (car (ref self 'maps)) args))
+       #:except KeyError =>
+       (lambda ()
+	 (raise KeyError
+		(format #f "Key not found in the first mapping: ~ a" key))))))
+
+  (define clear
+    (lambda (self)
+      (py-clear (car (ref self 'maps))))))
+
+(define-python-class Counter (dict)
+  (define __init__
+    (lam (self (* args) (** kwds))
+        (if (> (length args) 1)
+            (raise TypeError
+		   (format
+		    #f
+		    "expected at most 1 arguments, got ~ a"
+		    (length  args))))
+        ((ref (super Counter self) '__init__))
+        (py-apply (ref self 'update) (* args) (** kwds))))
+
+  (define __missing__
+    (lambda (self key) 0))
+
+  (define most_common
+    (lambda* (self #:key (n None)):
+       (if (eq? n None)
+            (sorted ((ref self 'items) #:key (_itemgetter 1) #:reverse #t)
+	    _heapq.nlargest(n ((ref self 'items)) #:key (_itemgetter 1))
+
+
+	       
+
+
+  
+
+(define* (namedtuple typename field-names #key (verbose #f) (rename=#f)
+(module None))
+  
+  
+
+