gtfs_df 0.1.1

data/lib/gtfs_df/writer.rb

@@ -0,0 +1,26 @@
+# frozen_string_literal: true
+
+module GtfsDf
+  class Writer
+    # Exports a Feed to a GTFS zip file
+    def self.write_to_zip(feed, zip_path)
+      require "stringio"
+      require "zlib"
+      FileUtils.mkdir_p(File.dirname(zip_path))
+      Zip::File.open(zip_path, create: true) do |zipfile|
+        GtfsDf::Feed::GTFS_FILES.each do |file|
+          df = feed.send(file)
+          next unless df.is_a?(Polars::DataFrame)
+
+          # Write CSV to StringIO
+          csv_io = StringIO.new
+          df.write_csv(csv_io)
+          csv_io.rewind
+
+          # Write StringIO directly to zip
+          zipfile.get_output_stream("#{file}.txt") { |f| f.write(csv_io.read) }
+        end
+      end
+    end
+  end
+end

data/lib/gtfs_df.rb

@@ -0,0 +1,49 @@
+# frozen_string_literal: true
+
+require "polars-df"
+require "networkx"
+require "zip"
+require_relative "gtfs_df/base_gtfs_table"
+require_relative "gtfs_df/utils"
+require_relative "gtfs_df/version"
+require_relative "gtfs_df/schema/enum_values"
+require_relative "gtfs_df/schema/agency"
+require_relative "gtfs_df/schema/calendar"
+require_relative "gtfs_df/schema/calendar_dates"
+require_relative "gtfs_df/feed"
+require_relative "gtfs_df/graph"
+require_relative "gtfs_df/reader"
+require_relative "gtfs_df/schema/routes"
+require_relative "gtfs_df/schema_validator"
+require_relative "gtfs_df/schema/stop_times"
+require_relative "gtfs_df/schema/stops"
+require_relative "gtfs_df/schema/trips"
+require_relative "gtfs_df/schema/pathways"
+require_relative "gtfs_df/schema/levels"
+require_relative "gtfs_df/schema/feed_info"
+require_relative "gtfs_df/schema/shapes"
+require_relative "gtfs_df/schema/frequencies"
+require_relative "gtfs_df/schema/transfers"
+require_relative "gtfs_df/schema/fare_attributes"
+require_relative "gtfs_df/schema/fare_rules"
+require_relative "gtfs_df/schema/attributions"
+require_relative "gtfs_df/schema/translations"
+require_relative "gtfs_df/schema/stop_areas"
+require_relative "gtfs_df/schema/stop_attributes"
+require_relative "gtfs_df/schema/rider_categories"
+require_relative "gtfs_df/schema/fare_media"
+require_relative "gtfs_df/schema/fare_products"
+require_relative "gtfs_df/schema/fare_leg_rules"
+require_relative "gtfs_df/schema/fare_leg_join_rules"
+require_relative "gtfs_df/schema/fare_transfer_rules"
+require_relative "gtfs_df/schema/areas"
+require_relative "gtfs_df/schema/networks"
+require_relative "gtfs_df/schema/route_networks"
+require_relative "gtfs_df/schema/location_groups"
+require_relative "gtfs_df/schema/location_group_stops"
+require_relative "gtfs_df/schema/booking_rules"
+require_relative "gtfs_df/writer"
+
+module GtfsDf
+  class Error < StandardError; end
+end

data/sig/gtfs-df.rbs

@@ -0,0 +1,4 @@
+module GtfsDf
+  VERSION: String
+  # See the writing guide of rbs: https://github.com/ruby/rbs#guides
+end

metadata

@@ -0,0 +1,148 @@
+--- !ruby/object:Gem::Specification
+name: gtfs_df
+version: !ruby/object:Gem::Version
+  version: 0.1.1
+platform: ruby
+authors:
+- David Mejorado
+bindir: exe
+cert_chain: []
+date: 1980-01-01 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: networkx
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.4'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.4'
+- !ruby/object:Gem::Dependency
+  name: polars-df
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.22'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.22'
+- !ruby/object:Gem::Dependency
+  name: rubyzip
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.3'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.3'
+description: 'A Ruby gem to load, filter, and manipulate GTFS (General Transit Feed
+  Specification) feeds using DataFrames powered by Polars. Supports cascading filters
+  that maintain referential integrity across related tables. NOTE: This gem is not
+  ready for production use.'
+email:
+- david.mejorado@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".conform.yaml"
+- ".envrc"
+- ".rspec"
+- ".rubocop.yml"
+- ".solargraph.yml"
+- ".standard.yml"
+- CHANGELOG.md
+- LICENSE.txt
+- README.md
+- Rakefile
+- devenv.lock
+- devenv.nix
+- devenv.yaml
+- examples/split-by-agency/.gitignore
+- examples/split-by-agency/Gemfile
+- examples/split-by-agency/Gemfile.lock
+- examples/split-by-agency/README.md
+- examples/split-by-agency/split_by_agency.rb
+- lib/gtfs_df.rb
+- lib/gtfs_df/base_gtfs_table.rb
+- lib/gtfs_df/feed.rb
+- lib/gtfs_df/graph.rb
+- lib/gtfs_df/reader.rb
+- lib/gtfs_df/schema/agency.rb
+- lib/gtfs_df/schema/areas.rb
+- lib/gtfs_df/schema/attributions.rb
+- lib/gtfs_df/schema/booking_rules.rb
+- lib/gtfs_df/schema/calendar.rb
+- lib/gtfs_df/schema/calendar_dates.rb
+- lib/gtfs_df/schema/enum_values.rb
+- lib/gtfs_df/schema/fare_attributes.rb
+- lib/gtfs_df/schema/fare_leg_join_rules.rb
+- lib/gtfs_df/schema/fare_leg_rules.rb
+- lib/gtfs_df/schema/fare_media.rb
+- lib/gtfs_df/schema/fare_products.rb
+- lib/gtfs_df/schema/fare_rules.rb
+- lib/gtfs_df/schema/fare_transfer_rules.rb
+- lib/gtfs_df/schema/feed_info.rb
+- lib/gtfs_df/schema/frequencies.rb
+- lib/gtfs_df/schema/levels.rb
+- lib/gtfs_df/schema/location_group_stops.rb
+- lib/gtfs_df/schema/location_groups.rb
+- lib/gtfs_df/schema/networks.rb
+- lib/gtfs_df/schema/pathways.rb
+- lib/gtfs_df/schema/rider_categories.rb
+- lib/gtfs_df/schema/route_networks.rb
+- lib/gtfs_df/schema/routes.rb
+- lib/gtfs_df/schema/shapes.rb
+- lib/gtfs_df/schema/stop_areas.rb
+- lib/gtfs_df/schema/stop_attributes.rb
+- lib/gtfs_df/schema/stop_times.rb
+- lib/gtfs_df/schema/stops.rb
+- lib/gtfs_df/schema/transfers.rb
+- lib/gtfs_df/schema/translations.rb
+- lib/gtfs_df/schema/trips.rb
+- lib/gtfs_df/schema_validator.rb
+- lib/gtfs_df/utils.rb
+- lib/gtfs_df/version.rb
+- lib/gtfs_df/writer.rb
+- sig/gtfs-df.rbs
+homepage: https://github.com/davidmh/ruby-gtfs-df
+licenses:
+- MIT
+metadata:
+  homepage_uri: https://github.com/davidmh/ruby-gtfs-df
+  source_code_uri: https://github.com/davidmh/ruby-gtfs-df
+  changelog_uri: https://github.com/davidmh/ruby-gtfs-df/blob/main/CHANGELOG.md
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 3.1.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.6.9
+specification_version: 4
+summary: Manipulate GTFS feeds using DataFrames with Polars
+test_files: []